This commit is contained in:
Ricky
2024-01-31 11:58:07 +08:00
committed by GitHub
parent 9e37702d24
commit 2660fbaa20
58 changed files with 312 additions and 312 deletions

View File

@@ -12,7 +12,7 @@ from core.tools.provider.tool_provider import ToolProviderController
from core.tools.provider.api_tool_provider import ApiBasedToolProviderController
from core.tools.utils.parser import ApiBasedToolSchemaParser
from core.tools.utils.encoder import serialize_base_model_array, serialize_base_model_dict
from core.tools.utils.configration import ToolConfiguration
from core.tools.utils.configuration import ToolConfiguration
from core.tools.errors import ToolProviderCredentialValidationError, ToolProviderNotFoundError, ToolNotFoundError
from extensions.ext_database import db
@@ -24,26 +24,26 @@ import json
class ToolManageService:
@staticmethod
def list_tool_providers(user_id: str, tanent_id: str):
def list_tool_providers(user_id: str, tenant_id: str):
"""
list tool providers
:return: the list of tool providers
"""
result = [provider.to_dict() for provider in ToolManager.user_list_providers(
user_id, tanent_id
user_id, tenant_id
)]
# add icon url prefix
for provider in result:
ToolManageService.repacket_provider(provider)
ToolManageService.repack_provider(provider)
return result
@staticmethod
def repacket_provider(provider: dict):
def repack_provider(provider: dict):
"""
repacket provider
repack provider
:param provider: the provider dict
"""
@@ -286,7 +286,7 @@ class ToolManageService:
).first()
if provider is None:
raise ValueError(f'yout have not added provider {provider}')
raise ValueError(f'you have not added provider {provider}')
return json.loads(
serialize_base_model_array([
@@ -424,7 +424,7 @@ class ToolManageService:
).first()
if provider is None:
raise ValueError(f'yout have not added provider {provider}')
raise ValueError(f'you have not added provider {provider}')
db.session.delete(provider)
db.session.commit()
@@ -457,7 +457,7 @@ class ToolManageService:
).first()
if provider is None:
raise ValueError(f'yout have not added provider {provider}')
raise ValueError(f'you have not added provider {provider}')
db.session.delete(provider)
db.session.commit()