kudzues/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-05 19:32:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: model_load_balancing_service and api_tools_manage_service (#34434)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Renzo
2026-04-02 06:38:35 +02:00
committed by GitHub
parent f9d9ad7a38
commit 399d3f8da5
3 changed files with 61 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
api/services/tools/api_tools_manage_service.py
View File

@@ -124,13 +124,13 @@ class ApiToolManageService:
provider_name = provider_name.strip()
# check if the provider exists
provider = (
db.session.query(ApiToolProvider)
provider = db.session.scalar(
select(ApiToolProvider)
.where(
ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == provider_name,
)
.first()
.limit(1)
)
if provider is not None:
@@ -215,13 +215,13 @@ class ApiToolManageService:
"""
list api tool provider tools
"""
provider: ApiToolProvider | None = (
db.session.query(ApiToolProvider)
provider: ApiToolProvider | None = db.session.scalar(
select(ApiToolProvider)
.where(
ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == provider_name,
)
.first()
.limit(1)
)
if provider is None:
@@ -259,13 +259,13 @@ class ApiToolManageService:
provider_name = provider_name.strip()
# check if the provider exists
provider = (
db.session.query(ApiToolProvider)
provider = db.session.scalar(
select(ApiToolProvider)
.where(
ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == original_provider,
)
.first()
.limit(1)
)
if provider is None:
@@ -328,13 +328,13 @@ class ApiToolManageService:
"""
delete tool provider
"""
provider = (
db.session.query(ApiToolProvider)
provider = db.session.scalar(
select(ApiToolProvider)
.where(
ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == provider_name,
)
.first()
.limit(1)
)
if provider is None:
@@ -378,13 +378,13 @@ class ApiToolManageService:
if tool_bundle is None:
raise ValueError(f"invalid tool name {tool_name}")
db_provider = (
db.session.query(ApiToolProvider)
db_provider = db.session.scalar(
select(ApiToolProvider)
.where(
ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == provider_name,
)
.first()
.limit(1)
)
if not db_provider: