Unverified Commit 2660fbaa authored by Ricky's avatar Ricky Committed by GitHub

Fix/typos (#2308)

parent 9e37702d
...@@ -170,7 +170,7 @@ class AssistantApplicationRunner(AppRunner): ...@@ -170,7 +170,7 @@ class AssistantApplicationRunner(AppRunner):
# load tool variables # load tool variables
tool_conversation_variables = self._load_tool_variables(conversation_id=conversation.id, tool_conversation_variables = self._load_tool_variables(conversation_id=conversation.id,
user_id=application_generate_entity.user_id, user_id=application_generate_entity.user_id,
tanent_id=application_generate_entity.tenant_id) tenant_id=application_generate_entity.tenant_id)
# convert db variables to tool variables # convert db variables to tool variables
tool_variables = self._convert_db_variables_to_tool_variables(tool_conversation_variables) tool_variables = self._convert_db_variables_to_tool_variables(tool_conversation_variables)
...@@ -254,13 +254,13 @@ class AssistantApplicationRunner(AppRunner): ...@@ -254,13 +254,13 @@ class AssistantApplicationRunner(AppRunner):
agent=True agent=True
) )
def _load_tool_variables(self, conversation_id: str, user_id: str, tanent_id: str) -> ToolConversationVariables: def _load_tool_variables(self, conversation_id: str, user_id: str, tenant_id: str) -> ToolConversationVariables:
""" """
load tool variables from database load tool variables from database
""" """
tool_variables: ToolConversationVariables = db.session.query(ToolConversationVariables).filter( tool_variables: ToolConversationVariables = db.session.query(ToolConversationVariables).filter(
ToolConversationVariables.conversation_id == conversation_id, ToolConversationVariables.conversation_id == conversation_id,
ToolConversationVariables.tenant_id == tanent_id ToolConversationVariables.tenant_id == tenant_id
).first() ).first()
if tool_variables: if tool_variables:
...@@ -271,7 +271,7 @@ class AssistantApplicationRunner(AppRunner): ...@@ -271,7 +271,7 @@ class AssistantApplicationRunner(AppRunner):
tool_variables = ToolConversationVariables( tool_variables = ToolConversationVariables(
conversation_id=conversation_id, conversation_id=conversation_id,
user_id=user_id, user_id=user_id,
tenant_id=tanent_id, tenant_id=tenant_id,
variables_str='[]', variables_str='[]',
) )
db.session.add(tool_variables) db.session.add(tool_variables)
......
...@@ -12,7 +12,7 @@ from models.model import MessageAgentThought, Message, MessageFile ...@@ -12,7 +12,7 @@ from models.model import MessageAgentThought, Message, MessageFile
from models.tools import ToolConversationVariables from models.tools import ToolConversationVariables
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolInvokeMessageBinary, \ from core.tools.entities.tool_entities import ToolInvokeMessage, ToolInvokeMessageBinary, \
ToolRuntimeVariablePool, ToolParamter ToolRuntimeVariablePool, ToolParameter
from core.tools.tool.tool import Tool from core.tools.tool.tool import Tool
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
from core.tools.tool_file_manager import ToolFileManager from core.tools.tool_file_manager import ToolFileManager
...@@ -108,9 +108,9 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -108,9 +108,9 @@ class BaseAssistantApplicationRunner(AppRunner):
else: else:
self.stream_tool_call = False self.stream_tool_call = False
def _repacket_app_orchestration_config(self, app_orchestration_config: AppOrchestrationConfigEntity) -> AppOrchestrationConfigEntity: def _repack_app_orchestration_config(self, app_orchestration_config: AppOrchestrationConfigEntity) -> AppOrchestrationConfigEntity:
""" """
Repacket app orchestration config Repack app orchestration config
""" """
if app_orchestration_config.prompt_template.simple_prompt_template is None: if app_orchestration_config.prompt_template.simple_prompt_template is None:
app_orchestration_config.prompt_template.simple_prompt_template = '' app_orchestration_config.prompt_template.simple_prompt_template = ''
...@@ -126,7 +126,7 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -126,7 +126,7 @@ class BaseAssistantApplicationRunner(AppRunner):
if response.type == ToolInvokeMessage.MessageType.TEXT: if response.type == ToolInvokeMessage.MessageType.TEXT:
result += response.message result += response.message
elif response.type == ToolInvokeMessage.MessageType.LINK: elif response.type == ToolInvokeMessage.MessageType.LINK:
result += f"result link: {response.message}. please dirct user to check it." result += f"result link: {response.message}. please tell user to check it."
elif response.type == ToolInvokeMessage.MessageType.IMAGE_LINK or \ elif response.type == ToolInvokeMessage.MessageType.IMAGE_LINK or \
response.type == ToolInvokeMessage.MessageType.IMAGE: response.type == ToolInvokeMessage.MessageType.IMAGE:
result += f"image has been created and sent to user already, you should tell user to check it now." result += f"image has been created and sent to user already, you should tell user to check it now."
...@@ -141,7 +141,7 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -141,7 +141,7 @@ class BaseAssistantApplicationRunner(AppRunner):
""" """
tool_entity = ToolManager.get_tool_runtime( tool_entity = ToolManager.get_tool_runtime(
provider_type=tool.provider_type, provider_name=tool.provider_id, tool_name=tool.tool_name, provider_type=tool.provider_type, provider_name=tool.provider_id, tool_name=tool.tool_name,
tanent_id=self.application_generate_entity.tenant_id, tenant_id=self.application_generate_entity.tenant_id,
agent_callback=self.agent_callback agent_callback=self.agent_callback
) )
tool_entity.load_variables(self.variables_pool) tool_entity.load_variables(self.variables_pool)
...@@ -185,20 +185,20 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -185,20 +185,20 @@ class BaseAssistantApplicationRunner(AppRunner):
for parameter in parameters: for parameter in parameters:
parameter_type = 'string' parameter_type = 'string'
enum = [] enum = []
if parameter.type == ToolParamter.ToolParameterType.STRING: if parameter.type == ToolParameter.ToolParameterType.STRING:
parameter_type = 'string' parameter_type = 'string'
elif parameter.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter.type == ToolParameter.ToolParameterType.BOOLEAN:
parameter_type = 'boolean' parameter_type = 'boolean'
elif parameter.type == ToolParamter.ToolParameterType.NUMBER: elif parameter.type == ToolParameter.ToolParameterType.NUMBER:
parameter_type = 'number' parameter_type = 'number'
elif parameter.type == ToolParamter.ToolParameterType.SELECT: elif parameter.type == ToolParameter.ToolParameterType.SELECT:
for option in parameter.options: for option in parameter.options:
enum.append(option.value) enum.append(option.value)
parameter_type = 'string' parameter_type = 'string'
else: else:
raise ValueError(f"parameter type {parameter.type} is not supported") raise ValueError(f"parameter type {parameter.type} is not supported")
if parameter.form == ToolParamter.ToolParameterForm.FORM: if parameter.form == ToolParameter.ToolParameterForm.FORM:
# get tool parameter from form # get tool parameter from form
tool_parameter_config = tool.tool_parameters.get(parameter.name) tool_parameter_config = tool.tool_parameters.get(parameter.name)
if not tool_parameter_config: if not tool_parameter_config:
...@@ -207,7 +207,7 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -207,7 +207,7 @@ class BaseAssistantApplicationRunner(AppRunner):
if not tool_parameter_config and parameter.required: if not tool_parameter_config and parameter.required:
raise ValueError(f"tool parameter {parameter.name} not found in tool config") raise ValueError(f"tool parameter {parameter.name} not found in tool config")
if parameter.type == ToolParamter.ToolParameterType.SELECT: if parameter.type == ToolParameter.ToolParameterType.SELECT:
# check if tool_parameter_config in options # check if tool_parameter_config in options
options = list(map(lambda x: x.value, parameter.options)) options = list(map(lambda x: x.value, parameter.options))
if tool_parameter_config not in options: if tool_parameter_config not in options:
...@@ -215,7 +215,7 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -215,7 +215,7 @@ class BaseAssistantApplicationRunner(AppRunner):
# convert tool parameter config to correct type # convert tool parameter config to correct type
try: try:
if parameter.type == ToolParamter.ToolParameterType.NUMBER: if parameter.type == ToolParameter.ToolParameterType.NUMBER:
# check if tool parameter is integer # check if tool parameter is integer
if isinstance(tool_parameter_config, int): if isinstance(tool_parameter_config, int):
tool_parameter_config = tool_parameter_config tool_parameter_config = tool_parameter_config
...@@ -226,11 +226,11 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -226,11 +226,11 @@ class BaseAssistantApplicationRunner(AppRunner):
tool_parameter_config = float(tool_parameter_config) tool_parameter_config = float(tool_parameter_config)
else: else:
tool_parameter_config = int(tool_parameter_config) tool_parameter_config = int(tool_parameter_config)
elif parameter.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter.type == ToolParameter.ToolParameterType.BOOLEAN:
tool_parameter_config = bool(tool_parameter_config) tool_parameter_config = bool(tool_parameter_config)
elif parameter.type not in [ToolParamter.ToolParameterType.SELECT, ToolParamter.ToolParameterType.STRING]: elif parameter.type not in [ToolParameter.ToolParameterType.SELECT, ToolParameter.ToolParameterType.STRING]:
tool_parameter_config = str(tool_parameter_config) tool_parameter_config = str(tool_parameter_config)
elif parameter.type == ToolParamter.ToolParameterType: elif parameter.type == ToolParameter.ToolParameterType:
tool_parameter_config = str(tool_parameter_config) tool_parameter_config = str(tool_parameter_config)
except Exception as e: except Exception as e:
raise ValueError(f"tool parameter {parameter.name} value {tool_parameter_config} is not correct type") raise ValueError(f"tool parameter {parameter.name} value {tool_parameter_config} is not correct type")
...@@ -238,7 +238,7 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -238,7 +238,7 @@ class BaseAssistantApplicationRunner(AppRunner):
# save tool parameter to tool entity memory # save tool parameter to tool entity memory
runtime_parameters[parameter.name] = tool_parameter_config runtime_parameters[parameter.name] = tool_parameter_config
elif parameter.form == ToolParamter.ToolParameterForm.LLM: elif parameter.form == ToolParameter.ToolParameterForm.LLM:
message_tool.parameters['properties'][parameter.name] = { message_tool.parameters['properties'][parameter.name] = {
"type": parameter_type, "type": parameter_type,
"description": parameter.llm_description or '', "description": parameter.llm_description or '',
...@@ -292,20 +292,20 @@ class BaseAssistantApplicationRunner(AppRunner): ...@@ -292,20 +292,20 @@ class BaseAssistantApplicationRunner(AppRunner):
for parameter in tool_runtime_parameters: for parameter in tool_runtime_parameters:
parameter_type = 'string' parameter_type = 'string'
enum = [] enum = []
if parameter.type == ToolParamter.ToolParameterType.STRING: if parameter.type == ToolParameter.ToolParameterType.STRING:
parameter_type = 'string' parameter_type = 'string'
elif parameter.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter.type == ToolParameter.ToolParameterType.BOOLEAN:
parameter_type = 'boolean' parameter_type = 'boolean'
elif parameter.type == ToolParamter.ToolParameterType.NUMBER: elif parameter.type == ToolParameter.ToolParameterType.NUMBER:
parameter_type = 'number' parameter_type = 'number'
elif parameter.type == ToolParamter.ToolParameterType.SELECT: elif parameter.type == ToolParameter.ToolParameterType.SELECT:
for option in parameter.options: for option in parameter.options:
enum.append(option.value) enum.append(option.value)
parameter_type = 'string' parameter_type = 'string'
else: else:
raise ValueError(f"parameter type {parameter.type} is not supported") raise ValueError(f"parameter type {parameter.type} is not supported")
if parameter.form == ToolParamter.ToolParameterForm.LLM: if parameter.form == ToolParameter.ToolParameterForm.LLM:
prompt_tool.parameters['properties'][parameter.name] = { prompt_tool.parameters['properties'][parameter.name] = {
"type": parameter_type, "type": parameter_type,
"description": parameter.llm_description or '', "description": parameter.llm_description or '',
......
...@@ -12,7 +12,7 @@ from core.model_runtime.entities.llm_entities import LLMResult, LLMUsage, LLMRes ...@@ -12,7 +12,7 @@ from core.model_runtime.entities.llm_entities import LLMResult, LLMUsage, LLMRes
from core.model_manager import ModelInstance from core.model_manager import ModelInstance
from core.tools.errors import ToolInvokeError, ToolNotFoundError, \ from core.tools.errors import ToolInvokeError, ToolNotFoundError, \
ToolNotSupportedError, ToolProviderNotFoundError, ToolParamterValidationError, \ ToolNotSupportedError, ToolProviderNotFoundError, ToolParameterValidationError, \
ToolProviderCredentialValidationError ToolProviderCredentialValidationError
from core.features.assistant_base_runner import BaseAssistantApplicationRunner from core.features.assistant_base_runner import BaseAssistantApplicationRunner
...@@ -28,7 +28,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -28,7 +28,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
Run Cot agent application Run Cot agent application
""" """
app_orchestration_config = self.app_orchestration_config app_orchestration_config = self.app_orchestration_config
self._repacket_app_orchestration_config(app_orchestration_config) self._repack_app_orchestration_config(app_orchestration_config)
agent_scratchpad: List[AgentScratchpadUnit] = [] agent_scratchpad: List[AgentScratchpadUnit] = []
...@@ -71,7 +71,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -71,7 +71,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
} }
final_answer = '' final_answer = ''
def increse_usage(final_llm_usage_dict: Dict[str, LLMUsage], usage: LLMUsage): def increase_usage(final_llm_usage_dict: Dict[str, LLMUsage], usage: LLMUsage):
if not final_llm_usage_dict['usage']: if not final_llm_usage_dict['usage']:
final_llm_usage_dict['usage'] = usage final_llm_usage_dict['usage'] = usage
else: else:
...@@ -105,7 +105,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -105,7 +105,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
self.queue_manager.publish_agent_thought(agent_thought, PublishFrom.APPLICATION_MANAGER) self.queue_manager.publish_agent_thought(agent_thought, PublishFrom.APPLICATION_MANAGER)
# update prompt messages # update prompt messages
prompt_messages = self._originze_cot_prompt_messages( prompt_messages = self._organize_cot_prompt_messages(
mode=app_orchestration_config.model_config.mode, mode=app_orchestration_config.model_config.mode,
prompt_messages=prompt_messages, prompt_messages=prompt_messages,
tools=prompt_messages_tools, tools=prompt_messages_tools,
...@@ -138,7 +138,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -138,7 +138,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
# get llm usage # get llm usage
if llm_result.usage: if llm_result.usage:
increse_usage(llm_usage, llm_result.usage) increase_usage(llm_usage, llm_result.usage)
# publish agent thought if it's first iteration # publish agent thought if it's first iteration
if iteration_step == 1: if iteration_step == 1:
...@@ -208,7 +208,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -208,7 +208,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
try: try:
tool_response = tool_instance.invoke( tool_response = tool_instance.invoke(
user_id=self.user_id, user_id=self.user_id,
tool_paramters=tool_call_args if isinstance(tool_call_args, dict) else json.loads(tool_call_args) tool_parameters=tool_call_args if isinstance(tool_call_args, dict) else json.loads(tool_call_args)
) )
# transform tool response to llm friendly response # transform tool response to llm friendly response
tool_response = self.transform_tool_invoke_messages(tool_response) tool_response = self.transform_tool_invoke_messages(tool_response)
...@@ -226,15 +226,15 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -226,15 +226,15 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
message_file_ids = [message_file.id for message_file, _ in message_files] message_file_ids = [message_file.id for message_file, _ in message_files]
except ToolProviderCredentialValidationError as e: except ToolProviderCredentialValidationError as e:
error_response = f"Plese check your tool provider credentials" error_response = f"Please check your tool provider credentials"
except ( except (
ToolNotFoundError, ToolNotSupportedError, ToolProviderNotFoundError ToolNotFoundError, ToolNotSupportedError, ToolProviderNotFoundError
) as e: ) as e:
error_response = f"there is not a tool named {tool_call_name}" error_response = f"there is not a tool named {tool_call_name}"
except ( except (
ToolParamterValidationError ToolParameterValidationError
) as e: ) as e:
error_response = f"tool paramters validation error: {e}, please check your tool paramters" error_response = f"tool parameters validation error: {e}, please check your tool parameters"
except ToolInvokeError as e: except ToolInvokeError as e:
error_response = f"tool invoke error: {e}" error_response = f"tool invoke error: {e}"
except Exception as e: except Exception as e:
...@@ -469,7 +469,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -469,7 +469,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
if not next_iteration.find("{{observation}}") >= 0: if not next_iteration.find("{{observation}}") >= 0:
raise ValueError("{{observation}} is required in next_iteration") raise ValueError("{{observation}} is required in next_iteration")
def _convert_strachpad_list_to_str(self, agent_scratchpad: List[AgentScratchpadUnit]) -> str: def _convert_scratchpad_list_to_str(self, agent_scratchpad: List[AgentScratchpadUnit]) -> str:
""" """
convert agent scratchpad list to str convert agent scratchpad list to str
""" """
...@@ -481,7 +481,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -481,7 +481,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
return result return result
def _originze_cot_prompt_messages(self, mode: Literal["completion", "chat"], def _organize_cot_prompt_messages(self, mode: Literal["completion", "chat"],
prompt_messages: List[PromptMessage], prompt_messages: List[PromptMessage],
tools: List[PromptMessageTool], tools: List[PromptMessageTool],
agent_scratchpad: List[AgentScratchpadUnit], agent_scratchpad: List[AgentScratchpadUnit],
...@@ -490,7 +490,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -490,7 +490,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
input: str, input: str,
) -> List[PromptMessage]: ) -> List[PromptMessage]:
""" """
originze chain of thought prompt messages, a standard prompt message is like: organize chain of thought prompt messages, a standard prompt message is like:
Respond to the human as helpfully and accurately as possible. Respond to the human as helpfully and accurately as possible.
{{instruction}} {{instruction}}
...@@ -528,7 +528,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -528,7 +528,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
.replace("{{tools}}", tools_str) \ .replace("{{tools}}", tools_str) \
.replace("{{tool_names}}", tool_names) .replace("{{tool_names}}", tool_names)
# originze prompt messages # organize prompt messages
if mode == "chat": if mode == "chat":
# override system message # override system message
overrided = False overrided = False
...@@ -559,7 +559,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner): ...@@ -559,7 +559,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
return prompt_messages return prompt_messages
elif mode == "completion": elif mode == "completion":
# parse agent scratchpad # parse agent scratchpad
agent_scratchpad_str = self._convert_strachpad_list_to_str(agent_scratchpad) agent_scratchpad_str = self._convert_scratchpad_list_to_str(agent_scratchpad)
# parse prompt messages # parse prompt messages
return [UserPromptMessage( return [UserPromptMessage(
content=first_prompt.replace("{{instruction}}", instruction) content=first_prompt.replace("{{instruction}}", instruction)
......
...@@ -10,7 +10,7 @@ from core.model_manager import ModelInstance ...@@ -10,7 +10,7 @@ from core.model_manager import ModelInstance
from core.application_queue_manager import PublishFrom from core.application_queue_manager import PublishFrom
from core.tools.errors import ToolInvokeError, ToolNotFoundError, \ from core.tools.errors import ToolInvokeError, ToolNotFoundError, \
ToolNotSupportedError, ToolProviderNotFoundError, ToolParamterValidationError, \ ToolNotSupportedError, ToolProviderNotFoundError, ToolParameterValidationError, \
ToolProviderCredentialValidationError ToolProviderCredentialValidationError
from core.features.assistant_base_runner import BaseAssistantApplicationRunner from core.features.assistant_base_runner import BaseAssistantApplicationRunner
...@@ -247,7 +247,7 @@ class AssistantFunctionCallApplicationRunner(BaseAssistantApplicationRunner): ...@@ -247,7 +247,7 @@ class AssistantFunctionCallApplicationRunner(BaseAssistantApplicationRunner):
try: try:
tool_invoke_message = tool_instance.invoke( tool_invoke_message = tool_instance.invoke(
user_id=self.user_id, user_id=self.user_id,
tool_paramters=tool_call_args, tool_parameters=tool_call_args,
) )
# transform tool invoke message to get LLM friendly message # transform tool invoke message to get LLM friendly message
tool_invoke_message = self.transform_tool_invoke_messages(tool_invoke_message) tool_invoke_message = self.transform_tool_invoke_messages(tool_invoke_message)
...@@ -272,9 +272,9 @@ class AssistantFunctionCallApplicationRunner(BaseAssistantApplicationRunner): ...@@ -272,9 +272,9 @@ class AssistantFunctionCallApplicationRunner(BaseAssistantApplicationRunner):
) as e: ) as e:
error_response = f"there is not a tool named {tool_call_name}" error_response = f"there is not a tool named {tool_call_name}"
except ( except (
ToolParamterValidationError ToolParameterValidationError
) as e: ) as e:
error_response = f"tool paramters validation error: {e}, please check your tool paramters" error_response = f"tool parameters validation error: {e}, please check your tool parameters"
except ToolInvokeError as e: except ToolInvokeError as e:
error_response = f"tool invoke error: {e}" error_response = f"tool invoke error: {e}"
except Exception as e: except Exception as e:
......
...@@ -41,7 +41,7 @@ class OpenLLMGenerate(object): ...@@ -41,7 +41,7 @@ class OpenLLMGenerate(object):
if not server_url: if not server_url:
raise InvalidAuthenticationError('Invalid server URL') raise InvalidAuthenticationError('Invalid server URL')
defautl_llm_config = { default_llm_config = {
"max_new_tokens": 128, "max_new_tokens": 128,
"min_length": 0, "min_length": 0,
"early_stopping": False, "early_stopping": False,
...@@ -75,19 +75,19 @@ class OpenLLMGenerate(object): ...@@ -75,19 +75,19 @@ class OpenLLMGenerate(object):
} }
if 'max_tokens' in model_parameters and type(model_parameters['max_tokens']) == int: if 'max_tokens' in model_parameters and type(model_parameters['max_tokens']) == int:
defautl_llm_config['max_new_tokens'] = model_parameters['max_tokens'] default_llm_config['max_new_tokens'] = model_parameters['max_tokens']
if 'temperature' in model_parameters and type(model_parameters['temperature']) == float: if 'temperature' in model_parameters and type(model_parameters['temperature']) == float:
defautl_llm_config['temperature'] = model_parameters['temperature'] default_llm_config['temperature'] = model_parameters['temperature']
if 'top_p' in model_parameters and type(model_parameters['top_p']) == float: if 'top_p' in model_parameters and type(model_parameters['top_p']) == float:
defautl_llm_config['top_p'] = model_parameters['top_p'] default_llm_config['top_p'] = model_parameters['top_p']
if 'top_k' in model_parameters and type(model_parameters['top_k']) == int: if 'top_k' in model_parameters and type(model_parameters['top_k']) == int:
defautl_llm_config['top_k'] = model_parameters['top_k'] default_llm_config['top_k'] = model_parameters['top_k']
if 'use_cache' in model_parameters and type(model_parameters['use_cache']) == bool: if 'use_cache' in model_parameters and type(model_parameters['use_cache']) == bool:
defautl_llm_config['use_cache'] = model_parameters['use_cache'] default_llm_config['use_cache'] = model_parameters['use_cache']
headers = { headers = {
'Content-Type': 'application/json', 'Content-Type': 'application/json',
...@@ -104,7 +104,7 @@ class OpenLLMGenerate(object): ...@@ -104,7 +104,7 @@ class OpenLLMGenerate(object):
data = { data = {
'stop': stop if stop else [], 'stop': stop if stop else [],
'prompt': '\n'.join([message.content for message in prompt_messages]), 'prompt': '\n'.join([message.content for message in prompt_messages]),
'llm_config': defautl_llm_config, 'llm_config': default_llm_config,
} }
try: try:
......
...@@ -125,7 +125,7 @@ from openai import OpenAI ...@@ -125,7 +125,7 @@ from openai import OpenAI
class DallE3Tool(BuiltinTool): class DallE3Tool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -135,7 +135,7 @@ class DallE3Tool(BuiltinTool): ...@@ -135,7 +135,7 @@ class DallE3Tool(BuiltinTool):
) )
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
...@@ -163,7 +163,7 @@ Next, we use Vectorizer.AI to convert the PNG icon generated by DallE3 into a ve ...@@ -163,7 +163,7 @@ Next, we use Vectorizer.AI to convert the PNG icon generated by DallE3 into a ve
```python ```python
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
...@@ -171,20 +171,20 @@ from httpx import post ...@@ -171,20 +171,20 @@ from httpx import post
from base64 import b64decode from base64 import b64decode
class VectorizerTool(BuiltinTool): class VectorizerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
Tool invocation, the image variable name needs to be passed in from here, so that we can get the image from the variable pool Tool invocation, the image variable name needs to be passed in from here, so that we can get the image from the variable pool
""" """
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
Override the tool parameter list, we can dynamically generate the parameter list based on the actual situation in the current variable pool, so that the LLM can generate the form based on the parameter list Override the tool parameter list, we can dynamically generate the parameter list based on the actual situation in the current variable pool, so that the LLM can generate the form based on the parameter list
""" """
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
""" """
Whether the current tool is available, if there is no image in the current variable pool, then we don't need to display this tool, just return False here Whether the current tool is available, if there is no image in the current variable pool, then we don't need to display this tool, just return False here
""" """
...@@ -194,7 +194,7 @@ Next, let's implement these three functions ...@@ -194,7 +194,7 @@ Next, let's implement these three functions
```python ```python
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
...@@ -202,7 +202,7 @@ from httpx import post ...@@ -202,7 +202,7 @@ from httpx import post
from base64 import b64decode from base64 import b64decode
class VectorizerTool(BuiltinTool): class VectorizerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -214,7 +214,7 @@ class VectorizerTool(BuiltinTool): ...@@ -214,7 +214,7 @@ class VectorizerTool(BuiltinTool):
raise ToolProviderCredentialValidationError('Please input api key name and value') raise ToolProviderCredentialValidationError('Please input api key name and value')
# Get image_id, the definition of image_id can be found in get_runtime_parameters # Get image_id, the definition of image_id can be found in get_runtime_parameters
image_id = tool_paramters.get('image_id', '') image_id = tool_parameters.get('image_id', '')
if not image_id: if not image_id:
return self.create_text_message('Please input image id') return self.create_text_message('Please input image id')
...@@ -241,24 +241,24 @@ class VectorizerTool(BuiltinTool): ...@@ -241,24 +241,24 @@ class VectorizerTool(BuiltinTool):
meta={'mime_type': 'image/svg+xml'}) meta={'mime_type': 'image/svg+xml'})
] ]
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
override the runtime parameters override the runtime parameters
""" """
# Here, we override the tool parameter list, define the image_id, and set its option list to all images in the current variable pool. The configuration here is consistent with the configuration in yaml. # Here, we override the tool parameter list, define the image_id, and set its option list to all images in the current variable pool. The configuration here is consistent with the configuration in yaml.
return [ return [
ToolParamter.get_simple_instance( ToolParameter.get_simple_instance(
name='image_id', name='image_id',
llm_description=f'the image id that you want to vectorize, \ llm_description=f'the image id that you want to vectorize, \
and the image id should be specified in \ and the image id should be specified in \
{[i.name for i in self.list_default_image_variables()]}', {[i.name for i in self.list_default_image_variables()]}',
type=ToolParamter.ToolParameterType.SELECT, type=ToolParameter.ToolParameterType.SELECT,
required=True, required=True,
options=[i.name for i in self.list_default_image_variables()] options=[i.name for i in self.list_default_image_variables()]
) )
] ]
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
# Only when there are images in the variable pool, the LLM needs to use this tool # Only when there are images in the variable pool, the LLM needs to use this tool
return len(self.list_default_image_variables()) > 0 return len(self.list_default_image_variables()) > 0
``` ```
......
...@@ -146,13 +146,13 @@ from typing import Any, Dict, List, Union ...@@ -146,13 +146,13 @@ from typing import Any, Dict, List, Union
class GoogleSearchTool(BuiltinTool): class GoogleSearchTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters['query'] query = tool_parameters['query']
result_type = tool_paramters['result_type'] result_type = tool_parameters['result_type']
api_key = self.runtime.credentials['serpapi_api_key'] api_key = self.runtime.credentials['serpapi_api_key']
# TODO: search with serpapi # TODO: search with serpapi
result = SerpAPI(api_key).run(query, result_type=result_type) result = SerpAPI(api_key).run(query, result_type=result_type)
...@@ -163,7 +163,7 @@ class GoogleSearchTool(BuiltinTool): ...@@ -163,7 +163,7 @@ class GoogleSearchTool(BuiltinTool):
``` ```
### Parameters ### Parameters
The overall logic of the tool is in the `_invoke` method, this method accepts two parameters: `user_id` and `tool_paramters`, which represent the user ID and tool parameters respectively The overall logic of the tool is in the `_invoke` method, this method accepts two parameters: `user_id` and `tool_parameters`, which represent the user ID and tool parameters respectively
### Return Data ### Return Data
When the tool returns, you can choose to return one message or multiple messages, here we return one message, using `create_text_message` and `create_link_message` can create a text message or a link message. When the tool returns, you can choose to return one message or multiple messages, here we return one message, using `create_text_message` and `create_link_message` can create a text message or a link message.
...@@ -195,7 +195,7 @@ class GoogleProvider(BuiltinToolProviderController): ...@@ -195,7 +195,7 @@ class GoogleProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"query": "test", "query": "test",
"result_type": "link" "result_type": "link"
}, },
......
...@@ -125,7 +125,7 @@ from openai import OpenAI ...@@ -125,7 +125,7 @@ from openai import OpenAI
class DallE3Tool(BuiltinTool): class DallE3Tool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -135,7 +135,7 @@ class DallE3Tool(BuiltinTool): ...@@ -135,7 +135,7 @@ class DallE3Tool(BuiltinTool):
) )
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
...@@ -163,7 +163,7 @@ class DallE3Tool(BuiltinTool): ...@@ -163,7 +163,7 @@ class DallE3Tool(BuiltinTool):
```python ```python
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
...@@ -171,20 +171,20 @@ from httpx import post ...@@ -171,20 +171,20 @@ from httpx import post
from base64 import b64decode from base64 import b64decode
class VectorizerTool(BuiltinTool): class VectorizerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
工具调用,图片变量名需要从这里传递进来,从而我们就可以从变量池中获取到图片 工具调用,图片变量名需要从这里传递进来,从而我们就可以从变量池中获取到图片
""" """
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
重写工具参数列表,我们可以根据当前变量池里的实际情况来动态生成参数列表,从而LLM可以根据参数列表来生成表单 重写工具参数列表,我们可以根据当前变量池里的实际情况来动态生成参数列表,从而LLM可以根据参数列表来生成表单
""" """
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
""" """
当前工具是否可用,如果当前变量池中没有图片,那么我们就不需要展示这个工具,这里返回False即可 当前工具是否可用,如果当前变量池中没有图片,那么我们就不需要展示这个工具,这里返回False即可
""" """
...@@ -194,7 +194,7 @@ class VectorizerTool(BuiltinTool): ...@@ -194,7 +194,7 @@ class VectorizerTool(BuiltinTool):
```python ```python
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
...@@ -202,7 +202,7 @@ from httpx import post ...@@ -202,7 +202,7 @@ from httpx import post
from base64 import b64decode from base64 import b64decode
class VectorizerTool(BuiltinTool): class VectorizerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -214,7 +214,7 @@ class VectorizerTool(BuiltinTool): ...@@ -214,7 +214,7 @@ class VectorizerTool(BuiltinTool):
raise ToolProviderCredentialValidationError('Please input api key name and value') raise ToolProviderCredentialValidationError('Please input api key name and value')
# 获取image_id,image_id的定义可以在get_runtime_parameters中找到 # 获取image_id,image_id的定义可以在get_runtime_parameters中找到
image_id = tool_paramters.get('image_id', '') image_id = tool_parameters.get('image_id', '')
if not image_id: if not image_id:
return self.create_text_message('Please input image id') return self.create_text_message('Please input image id')
...@@ -241,24 +241,24 @@ class VectorizerTool(BuiltinTool): ...@@ -241,24 +241,24 @@ class VectorizerTool(BuiltinTool):
meta={'mime_type': 'image/svg+xml'}) meta={'mime_type': 'image/svg+xml'})
] ]
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
override the runtime parameters override the runtime parameters
""" """
# 这里,我们重写了工具参数列表,定义了image_id,并设置了它的选项列表为当前变量池中的所有图片,这里的配置与yaml中的配置是一致的 # 这里,我们重写了工具参数列表,定义了image_id,并设置了它的选项列表为当前变量池中的所有图片,这里的配置与yaml中的配置是一致的
return [ return [
ToolParamter.get_simple_instance( ToolParameter.get_simple_instance(
name='image_id', name='image_id',
llm_description=f'the image id that you want to vectorize, \ llm_description=f'the image id that you want to vectorize, \
and the image id should be specified in \ and the image id should be specified in \
{[i.name for i in self.list_default_image_variables()]}', {[i.name for i in self.list_default_image_variables()]}',
type=ToolParamter.ToolParameterType.SELECT, type=ToolParameter.ToolParameterType.SELECT,
required=True, required=True,
options=[i.name for i in self.list_default_image_variables()] options=[i.name for i in self.list_default_image_variables()]
) )
] ]
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
# 只有当变量池中有图片时,LLM才需要使用这个工具 # 只有当变量池中有图片时,LLM才需要使用这个工具
return len(self.list_default_image_variables()) > 0 return len(self.list_default_image_variables()) > 0
``` ```
......
...@@ -146,13 +146,13 @@ from typing import Any, Dict, List, Union ...@@ -146,13 +146,13 @@ from typing import Any, Dict, List, Union
class GoogleSearchTool(BuiltinTool): class GoogleSearchTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters['query'] query = tool_parameters['query']
result_type = tool_paramters['result_type'] result_type = tool_parameters['result_type']
api_key = self.runtime.credentials['serpapi_api_key'] api_key = self.runtime.credentials['serpapi_api_key']
# TODO: search with serpapi # TODO: search with serpapi
result = SerpAPI(api_key).run(query, result_type=result_type) result = SerpAPI(api_key).run(query, result_type=result_type)
...@@ -163,7 +163,7 @@ class GoogleSearchTool(BuiltinTool): ...@@ -163,7 +163,7 @@ class GoogleSearchTool(BuiltinTool):
``` ```
### 参数 ### 参数
工具的整体逻辑都在`_invoke`方法中,这个方法接收两个参数:`user_id``tool_paramters`,分别表示用户ID和工具参数 工具的整体逻辑都在`_invoke`方法中,这个方法接收两个参数:`user_id``tool_parameters`,分别表示用户ID和工具参数
### 返回数据 ### 返回数据
在工具返回时,你可以选择返回一个消息或者多个消息,这里我们返回一个消息,使用`create_text_message``create_link_message`可以创建一个文本消息或者一个链接消息。 在工具返回时,你可以选择返回一个消息或者多个消息,这里我们返回一个消息,使用`create_text_message``create_link_message`可以创建一个文本消息或者一个链接消息。
...@@ -195,7 +195,7 @@ class GoogleProvider(BuiltinToolProviderController): ...@@ -195,7 +195,7 @@ class GoogleProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"query": "test", "query": "test",
"result_type": "link" "result_type": "link"
}, },
......
from pydantic import BaseModel from pydantic import BaseModel
from typing import Dict, Optional, Any, List from typing import Dict, Optional, Any, List
from core.tools.entities.tool_entities import ToolProviderType, ToolParamter from core.tools.entities.tool_entities import ToolProviderType, ToolParameter
class ApiBasedToolBundle(BaseModel): class ApiBasedToolBundle(BaseModel):
""" """
...@@ -16,7 +16,7 @@ class ApiBasedToolBundle(BaseModel): ...@@ -16,7 +16,7 @@ class ApiBasedToolBundle(BaseModel):
# operation_id # operation_id
operation_id: str = None operation_id: str = None
# parameters # parameters
parameters: Optional[List[ToolParamter]] = None parameters: Optional[List[ToolParameter]] = None
# author # author
author: str author: str
# icon # icon
......
...@@ -88,11 +88,11 @@ class ToolInvokeMessageBinary(BaseModel): ...@@ -88,11 +88,11 @@ class ToolInvokeMessageBinary(BaseModel):
url: str = Field(..., description="The url of the binary") url: str = Field(..., description="The url of the binary")
save_as: str = '' save_as: str = ''
class ToolParamterOption(BaseModel): class ToolParameterOption(BaseModel):
value: str = Field(..., description="The value of the option") value: str = Field(..., description="The value of the option")
label: I18nObject = Field(..., description="The label of the option") label: I18nObject = Field(..., description="The label of the option")
class ToolParamter(BaseModel): class ToolParameter(BaseModel):
class ToolParameterType(Enum): class ToolParameterType(Enum):
STRING = "string" STRING = "string"
NUMBER = "number" NUMBER = "number"
...@@ -114,12 +114,12 @@ class ToolParamter(BaseModel): ...@@ -114,12 +114,12 @@ class ToolParamter(BaseModel):
default: Optional[str] = None default: Optional[str] = None
min: Optional[Union[float, int]] = None min: Optional[Union[float, int]] = None
max: Optional[Union[float, int]] = None max: Optional[Union[float, int]] = None
options: Optional[List[ToolParamterOption]] = None options: Optional[List[ToolParameterOption]] = None
@classmethod @classmethod
def get_simple_instance(cls, def get_simple_instance(cls,
name: str, llm_description: str, type: ToolParameterType, name: str, llm_description: str, type: ToolParameterType,
required: bool, options: Optional[List[str]] = None) -> 'ToolParamter': required: bool, options: Optional[List[str]] = None) -> 'ToolParameter':
""" """
get a simple tool parameter get a simple tool parameter
...@@ -129,9 +129,9 @@ class ToolParamter(BaseModel): ...@@ -129,9 +129,9 @@ class ToolParamter(BaseModel):
:param required: if the parameter is required :param required: if the parameter is required
:param options: the options of the parameter :param options: the options of the parameter
""" """
# convert options to ToolParamterOption # convert options to ToolParameterOption
if options: if options:
options = [ToolParamterOption(value=option, label=I18nObject(en_US=option, zh_Hans=option)) for option in options] options = [ToolParameterOption(value=option, label=I18nObject(en_US=option, zh_Hans=option)) for option in options]
return cls( return cls(
name=name, name=name,
label=I18nObject(en_US='', zh_Hans=''), label=I18nObject(en_US='', zh_Hans=''),
...@@ -183,7 +183,7 @@ class ToolProviderCredentials(BaseModel): ...@@ -183,7 +183,7 @@ class ToolProviderCredentials(BaseModel):
raise ValueError(f'invalid mode value {value}') raise ValueError(f'invalid mode value {value}')
@staticmethod @staticmethod
def defaut(value: str) -> str: def default(value: str) -> str:
return "" return ""
name: str = Field(..., description="The name of the credentials") name: str = Field(..., description="The name of the credentials")
......
...@@ -4,7 +4,7 @@ from typing import List, Dict, Optional ...@@ -4,7 +4,7 @@ from typing import List, Dict, Optional
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
from core.tools.entities.tool_entities import ToolProviderCredentials from core.tools.entities.tool_entities import ToolProviderCredentials
from core.tools.tool.tool import ToolParamter from core.tools.tool.tool import ToolParameter
class UserToolProvider(BaseModel): class UserToolProvider(BaseModel):
class ProviderType(Enum): class ProviderType(Enum):
...@@ -45,4 +45,4 @@ class UserTool(BaseModel): ...@@ -45,4 +45,4 @@ class UserTool(BaseModel):
name: str # identifier name: str # identifier
label: I18nObject # label label: I18nObject # label
description: I18nObject description: I18nObject
parameters: Optional[List[ToolParamter]] parameters: Optional[List[ToolParameter]]
\ No newline at end of file \ No newline at end of file
...@@ -4,7 +4,7 @@ class ToolProviderNotFoundError(ValueError): ...@@ -4,7 +4,7 @@ class ToolProviderNotFoundError(ValueError):
class ToolNotFoundError(ValueError): class ToolNotFoundError(ValueError):
pass pass
class ToolParamterValidationError(ValueError): class ToolParameterValidationError(ValueError):
pass pass
class ToolProviderCredentialValidationError(ValueError): class ToolProviderCredentialValidationError(ValueError):
......
...@@ -123,12 +123,12 @@ class ApiBasedToolProviderController(ToolProviderController): ...@@ -123,12 +123,12 @@ class ApiBasedToolProviderController(ToolProviderController):
return self.tools return self.tools
def get_tools(self, user_id: str, tanent_id: str) -> List[ApiTool]: def get_tools(self, user_id: str, tenant_id: str) -> List[ApiTool]:
""" """
fetch tools from database fetch tools from database
:param user_id: the user id :param user_id: the user id
:param tanent_id: the tanent id :param tenant_id: the tenant id
:return: the tools :return: the tools
""" """
if self.tools is not None: if self.tools is not None:
...@@ -136,9 +136,9 @@ class ApiBasedToolProviderController(ToolProviderController): ...@@ -136,9 +136,9 @@ class ApiBasedToolProviderController(ToolProviderController):
tools: List[Tool] = [] tools: List[Tool] = []
# get tanent api providers # get tenant api providers
db_providers: List[ApiToolProvider] = db.session.query(ApiToolProvider).filter( db_providers: List[ApiToolProvider] = db.session.query(ApiToolProvider).filter(
ApiToolProvider.tenant_id == tanent_id, ApiToolProvider.tenant_id == tenant_id,
ApiToolProvider.name == self.identity.name ApiToolProvider.name == self.identity.name
).all() ).all()
......
from typing import Any, Dict, List from typing import Any, Dict, List
from core.tools.entities.tool_entities import ToolProviderType, ToolParamter, ToolParamterOption from core.tools.entities.tool_entities import ToolProviderType, ToolParameter, ToolParameterOption
from core.tools.tool.tool import Tool from core.tools.tool.tool import Tool
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
from core.tools.provider.tool_provider import ToolProviderController from core.tools.provider.tool_provider import ToolProviderController
...@@ -71,7 +71,7 @@ class AppBasedToolProviderEntity(ToolProviderController): ...@@ -71,7 +71,7 @@ class AppBasedToolProviderEntity(ToolProviderController):
variable_name = input_form[form_type]['variable_name'] variable_name = input_form[form_type]['variable_name']
options = input_form[form_type].get('options', []) options = input_form[form_type].get('options', [])
if form_type == 'paragraph' or form_type == 'text-input': if form_type == 'paragraph' or form_type == 'text-input':
tool['parameters'].append(ToolParamter( tool['parameters'].append(ToolParameter(
name=variable_name, name=variable_name,
label=I18nObject( label=I18nObject(
en_US=label, en_US=label,
...@@ -82,13 +82,13 @@ class AppBasedToolProviderEntity(ToolProviderController): ...@@ -82,13 +82,13 @@ class AppBasedToolProviderEntity(ToolProviderController):
zh_Hans=label zh_Hans=label
), ),
llm_description=label, llm_description=label,
form=ToolParamter.ToolParameterForm.FORM, form=ToolParameter.ToolParameterForm.FORM,
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
required=required, required=required,
default=default default=default
)) ))
elif form_type == 'select': elif form_type == 'select':
tool['parameters'].append(ToolParamter( tool['parameters'].append(ToolParameter(
name=variable_name, name=variable_name,
label=I18nObject( label=I18nObject(
en_US=label, en_US=label,
...@@ -99,11 +99,11 @@ class AppBasedToolProviderEntity(ToolProviderController): ...@@ -99,11 +99,11 @@ class AppBasedToolProviderEntity(ToolProviderController):
zh_Hans=label zh_Hans=label
), ),
llm_description=label, llm_description=label,
form=ToolParamter.ToolParameterForm.FORM, form=ToolParameter.ToolParameterForm.FORM,
type=ToolParamter.ToolParameterType.SELECT, type=ToolParameter.ToolParameterType.SELECT,
required=required, required=required,
default=default, default=default,
options=[ToolParamterOption( options=[ToolParameterOption(
value=option, value=option,
label=I18nObject( label=I18nObject(
en_US=option, en_US=option,
......
...@@ -13,7 +13,7 @@ class AzureDALLEProvider(BuiltinToolProviderController): ...@@ -13,7 +13,7 @@ class AzureDALLEProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"prompt": "cute girl, blue eyes, white hair, anime style", "prompt": "cute girl, blue eyes, white hair, anime style",
"size": "square", "size": "square",
"n": 1 "n": 1
......
...@@ -10,7 +10,7 @@ from openai import AzureOpenAI ...@@ -10,7 +10,7 @@ from openai import AzureOpenAI
class DallE3Tool(BuiltinTool): class DallE3Tool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -28,19 +28,19 @@ class DallE3Tool(BuiltinTool): ...@@ -28,19 +28,19 @@ class DallE3Tool(BuiltinTool):
} }
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
# get size # get size
size = SIZE_MAPPING[tool_paramters.get('size', 'square')] size = SIZE_MAPPING[tool_parameters.get('size', 'square')]
# get n # get n
n = tool_paramters.get('n', 1) n = tool_parameters.get('n', 1)
# get quality # get quality
quality = tool_paramters.get('quality', 'standard') quality = tool_parameters.get('quality', 'standard')
if quality not in ['standard', 'hd']: if quality not in ['standard', 'hd']:
return self.create_text_message('Invalid quality') return self.create_text_message('Invalid quality')
# get style # get style
style = tool_paramters.get('style', 'vivid') style = tool_parameters.get('style', 'vivid')
if style not in ['natural', 'vivid']: if style not in ['natural', 'vivid']:
return self.create_text_message('Invalid style') return self.create_text_message('Invalid style')
......
...@@ -16,7 +16,7 @@ class ChartProvider(BuiltinToolProviderController): ...@@ -16,7 +16,7 @@ class ChartProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"data": "1,3,5,7,9,2,4,6,8,10", "data": "1,3,5,7,9,2,4,6,8,10",
}, },
) )
......
...@@ -6,9 +6,9 @@ import io ...@@ -6,9 +6,9 @@ import io
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
class BarChartTool(BuiltinTool): class BarChartTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
data = tool_paramters.get('data', '') data = tool_parameters.get('data', '')
if not data: if not data:
return self.create_text_message('Please input data') return self.create_text_message('Please input data')
data = data.split(';') data = data.split(';')
...@@ -19,7 +19,7 @@ class BarChartTool(BuiltinTool): ...@@ -19,7 +19,7 @@ class BarChartTool(BuiltinTool):
else: else:
data = [float(i) for i in data] data = [float(i) for i in data]
axis = tool_paramters.get('x_axis', None) or None axis = tool_parameters.get('x_axis', None) or None
if axis: if axis:
axis = axis.split(';') axis = axis.split(';')
if len(axis) != len(data): if len(axis) != len(data):
......
...@@ -8,14 +8,14 @@ from typing import Any, Dict, List, Union ...@@ -8,14 +8,14 @@ from typing import Any, Dict, List, Union
class LinearChartTool(BuiltinTool): class LinearChartTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
data = tool_paramters.get('data', '') data = tool_parameters.get('data', '')
if not data: if not data:
return self.create_text_message('Please input data') return self.create_text_message('Please input data')
data = data.split(';') data = data.split(';')
axis = tool_paramters.get('x_axis', None) or None axis = tool_parameters.get('x_axis', None) or None
if axis: if axis:
axis = axis.split(';') axis = axis.split(';')
if len(axis) != len(data): if len(axis) != len(data):
......
...@@ -8,13 +8,13 @@ from typing import Any, Dict, List, Union ...@@ -8,13 +8,13 @@ from typing import Any, Dict, List, Union
class PieChartTool(BuiltinTool): class PieChartTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
data = tool_paramters.get('data', '') data = tool_parameters.get('data', '')
if not data: if not data:
return self.create_text_message('Please input data') return self.create_text_message('Please input data')
data = data.split(';') data = data.split(';')
categories = tool_paramters.get('categories', None) or None categories = tool_parameters.get('categories', None) or None
# if all data is int, convert to int # if all data is int, convert to int
if all([i.isdigit() for i in data]): if all([i.isdigit() for i in data]):
......
...@@ -13,7 +13,7 @@ class DALLEProvider(BuiltinToolProviderController): ...@@ -13,7 +13,7 @@ class DALLEProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"prompt": "cute girl, blue eyes, white hair, anime style", "prompt": "cute girl, blue eyes, white hair, anime style",
"size": "small", "size": "small",
"n": 1 "n": 1
......
...@@ -10,7 +10,7 @@ from openai import OpenAI ...@@ -10,7 +10,7 @@ from openai import OpenAI
class DallE2Tool(BuiltinTool): class DallE2Tool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -37,15 +37,15 @@ class DallE2Tool(BuiltinTool): ...@@ -37,15 +37,15 @@ class DallE2Tool(BuiltinTool):
} }
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
# get size # get size
size = SIZE_MAPPING[tool_paramters.get('size', 'large')] size = SIZE_MAPPING[tool_parameters.get('size', 'large')]
# get n # get n
n = tool_paramters.get('n', 1) n = tool_parameters.get('n', 1)
# call openapi dalle2 # call openapi dalle2
response = client.images.generate( response = client.images.generate(
......
...@@ -10,7 +10,7 @@ from openai import OpenAI ...@@ -10,7 +10,7 @@ from openai import OpenAI
class DallE3Tool(BuiltinTool): class DallE3Tool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -37,19 +37,19 @@ class DallE3Tool(BuiltinTool): ...@@ -37,19 +37,19 @@ class DallE3Tool(BuiltinTool):
} }
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
# get size # get size
size = SIZE_MAPPING[tool_paramters.get('size', 'square')] size = SIZE_MAPPING[tool_parameters.get('size', 'square')]
# get n # get n
n = tool_paramters.get('n', 1) n = tool_parameters.get('n', 1)
# get quality # get quality
quality = tool_paramters.get('quality', 'standard') quality = tool_parameters.get('quality', 'standard')
if quality not in ['standard', 'hd']: if quality not in ['standard', 'hd']:
return self.create_text_message('Invalid quality') return self.create_text_message('Invalid quality')
# get style # get style
style = tool_paramters.get('style', 'vivid') style = tool_parameters.get('style', 'vivid')
if style not in ['natural', 'vivid']: if style not in ['natural', 'vivid']:
return self.create_text_message('Invalid style') return self.create_text_message('Invalid style')
......
...@@ -6,11 +6,11 @@ from typing import Any, Dict, List, Union ...@@ -6,11 +6,11 @@ from typing import Any, Dict, List, Union
class GaodeRepositoriesTool(BuiltinTool): class GaodeRepositoriesTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
city = tool_paramters.get('city', '') city = tool_parameters.get('city', '')
if not city: if not city:
return self.create_text_message('Please tell me your city') return self.create_text_message('Please tell me your city')
......
...@@ -9,12 +9,12 @@ from typing import Any, Dict, List, Union ...@@ -9,12 +9,12 @@ from typing import Any, Dict, List, Union
class GihubRepositoriesTool(BuiltinTool): class GihubRepositoriesTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
top_n = tool_paramters.get('top_n', 5) top_n = tool_parameters.get('top_n', 5)
query = tool_paramters.get('query', '') query = tool_parameters.get('query', '')
if not query: if not query:
return self.create_text_message('Please input symbol') return self.create_text_message('Please input symbol')
......
...@@ -14,7 +14,7 @@ class GoogleProvider(BuiltinToolProviderController): ...@@ -14,7 +14,7 @@ class GoogleProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"query": "test", "query": "test",
"result_type": "link" "result_type": "link"
}, },
......
...@@ -148,13 +148,13 @@ class SerpAPI: ...@@ -148,13 +148,13 @@ class SerpAPI:
class GoogleSearchTool(BuiltinTool): class GoogleSearchTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters['query'] query = tool_parameters['query']
result_type = tool_paramters['result_type'] result_type = tool_parameters['result_type']
api_key = self.runtime.credentials['serpapi_api_key'] api_key = self.runtime.credentials['serpapi_api_key']
result = SerpAPI(api_key).run(query, result_type=result_type) result = SerpAPI(api_key).run(query, result_type=result_type)
if result_type == 'text': if result_type == 'text':
......
...@@ -14,7 +14,7 @@ class StableDiffusionProvider(BuiltinToolProviderController): ...@@ -14,7 +14,7 @@ class StableDiffusionProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"prompt": "cat", "prompt": "cat",
"lora": "", "lora": "",
"steps": 1, "steps": 1,
......
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter, ToolParamterOption from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter, ToolParameterOption
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
...@@ -60,7 +60,7 @@ DRAW_TEXT_OPTIONS = { ...@@ -60,7 +60,7 @@ DRAW_TEXT_OPTIONS = {
} }
class StableDiffusionTool(BuiltinTool): class StableDiffusionTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
...@@ -86,25 +86,25 @@ class StableDiffusionTool(BuiltinTool): ...@@ -86,25 +86,25 @@ class StableDiffusionTool(BuiltinTool):
# prompt # prompt
prompt = tool_paramters.get('prompt', '') prompt = tool_parameters.get('prompt', '')
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
# get negative prompt # get negative prompt
negative_prompt = tool_paramters.get('negative_prompt', '') negative_prompt = tool_parameters.get('negative_prompt', '')
# get size # get size
width = tool_paramters.get('width', 1024) width = tool_parameters.get('width', 1024)
height = tool_paramters.get('height', 1024) height = tool_parameters.get('height', 1024)
# get steps # get steps
steps = tool_paramters.get('steps', 1) steps = tool_parameters.get('steps', 1)
# get lora # get lora
lora = tool_paramters.get('lora', '') lora = tool_parameters.get('lora', '')
# get image id # get image id
image_id = tool_paramters.get('image_id', '') image_id = tool_parameters.get('image_id', '')
if image_id.strip(): if image_id.strip():
image_variable = self.get_default_image_variable() image_variable = self.get_default_image_variable()
if image_variable: if image_variable:
...@@ -212,32 +212,32 @@ class StableDiffusionTool(BuiltinTool): ...@@ -212,32 +212,32 @@ class StableDiffusionTool(BuiltinTool):
return self.create_text_message('Failed to generate image') return self.create_text_message('Failed to generate image')
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
parameters = [ parameters = [
ToolParamter(name='prompt', ToolParameter(name='prompt',
label=I18nObject(en_US='Prompt', zh_Hans='Prompt'), label=I18nObject(en_US='Prompt', zh_Hans='Prompt'),
human_description=I18nObject( human_description=I18nObject(
en_US='Image prompt, you can check the official documentation of Stable Diffusion', en_US='Image prompt, you can check the official documentation of Stable Diffusion',
zh_Hans='图像提示词,您可以查看 Stable Diffusion 的官方文档', zh_Hans='图像提示词,您可以查看 Stable Diffusion 的官方文档',
), ),
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
form=ToolParamter.ToolParameterForm.LLM, form=ToolParameter.ToolParameterForm.LLM,
llm_description='Image prompt of Stable Diffusion, you should describe the image you want to generate as a list of words as possible as detailed, the prompt must be written in English.', llm_description='Image prompt of Stable Diffusion, you should describe the image you want to generate as a list of words as possible as detailed, the prompt must be written in English.',
required=True), required=True),
] ]
if len(self.list_default_image_variables()) != 0: if len(self.list_default_image_variables()) != 0:
parameters.append( parameters.append(
ToolParamter(name='image_id', ToolParameter(name='image_id',
label=I18nObject(en_US='image_id', zh_Hans='image_id'), label=I18nObject(en_US='image_id', zh_Hans='image_id'),
human_description=I18nObject( human_description=I18nObject(
en_US='Image id of the image you want to generate based on, if you want to generate image based on the default image, you can leave this field empty.', en_US='Image id of the image you want to generate based on, if you want to generate image based on the default image, you can leave this field empty.',
zh_Hans='您想要生成的图像的图像 ID,如果您想要基于默认图像生成图像,则可以将此字段留空。', zh_Hans='您想要生成的图像的图像 ID,如果您想要基于默认图像生成图像,则可以将此字段留空。',
), ),
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
form=ToolParamter.ToolParameterForm.LLM, form=ToolParameter.ToolParameterForm.LLM,
llm_description='Image id of the original image, you can leave this field empty if you want to generate a new image.', llm_description='Image id of the original image, you can leave this field empty if you want to generate a new image.',
required=True, required=True,
options=[ToolParamterOption( options=[ToolParameterOption(
value=i.name, value=i.name,
label=I18nObject(en_US=i.name, zh_Hans=i.name) label=I18nObject(en_US=i.name, zh_Hans=i.name)
) for i in self.list_default_image_variables()]) ) for i in self.list_default_image_variables()])
......
...@@ -10,7 +10,7 @@ class WikiPediaProvider(BuiltinToolProviderController): ...@@ -10,7 +10,7 @@ class WikiPediaProvider(BuiltinToolProviderController):
try: try:
CurrentTimeTool().invoke( CurrentTimeTool().invoke(
user_id='', user_id='',
tool_paramters={}, tool_parameters={},
) )
except Exception as e: except Exception as e:
raise ToolProviderCredentialValidationError(str(e)) raise ToolProviderCredentialValidationError(str(e))
\ No newline at end of file
...@@ -8,7 +8,7 @@ from datetime import datetime, timezone ...@@ -8,7 +8,7 @@ from datetime import datetime, timezone
class CurrentTimeTool(BuiltinTool): class CurrentTimeTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
......
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.provider.builtin.vectorizer.tools.test_data import VECTORIZER_ICON_PNG from core.tools.provider.builtin.vectorizer.tools.test_data import VECTORIZER_ICON_PNG
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
...@@ -8,21 +8,21 @@ from httpx import post ...@@ -8,21 +8,21 @@ from httpx import post
from base64 import b64decode from base64 import b64decode
class VectorizerTool(BuiltinTool): class VectorizerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
api_key_name = self.runtime.credentials.get('api_key_name', None) api_key_name = self.runtime.credentials.get('api_key_name', None)
api_key_value = self.runtime.credentials.get('api_key_value', None) api_key_value = self.runtime.credentials.get('api_key_value', None)
mode = tool_paramters.get('mode', 'test') mode = tool_parameters.get('mode', 'test')
if mode == 'production': if mode == 'production':
mode = 'preview' mode = 'preview'
if not api_key_name or not api_key_value: if not api_key_name or not api_key_value:
raise ToolProviderCredentialValidationError('Please input api key name and value') raise ToolProviderCredentialValidationError('Please input api key name and value')
image_id = tool_paramters.get('image_id', '') image_id = tool_parameters.get('image_id', '')
if not image_id: if not image_id:
return self.create_text_message('Please input image id') return self.create_text_message('Please input image id')
...@@ -54,21 +54,21 @@ class VectorizerTool(BuiltinTool): ...@@ -54,21 +54,21 @@ class VectorizerTool(BuiltinTool):
meta={'mime_type': 'image/svg+xml'}) meta={'mime_type': 'image/svg+xml'})
] ]
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
override the runtime parameters override the runtime parameters
""" """
return [ return [
ToolParamter.get_simple_instance( ToolParameter.get_simple_instance(
name='image_id', name='image_id',
llm_description=f'the image id that you want to vectorize, \ llm_description=f'the image id that you want to vectorize, \
and the image id should be specified in \ and the image id should be specified in \
{[i.name for i in self.list_default_image_variables()]}', {[i.name for i in self.list_default_image_variables()]}',
type=ToolParamter.ToolParameterType.SELECT, type=ToolParameter.ToolParameterType.SELECT,
required=True, required=True,
options=[i.name for i in self.list_default_image_variables()] options=[i.name for i in self.list_default_image_variables()]
) )
] ]
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
return len(self.list_default_image_variables()) > 0 return len(self.list_default_image_variables()) > 0
\ No newline at end of file
...@@ -14,7 +14,7 @@ class VectorizerProvider(BuiltinToolProviderController): ...@@ -14,7 +14,7 @@ class VectorizerProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"mode": "test", "mode": "test",
"image_id": "__test_123" "image_id": "__test_123"
}, },
......
...@@ -7,14 +7,14 @@ from typing import Any, Dict, List, Union ...@@ -7,14 +7,14 @@ from typing import Any, Dict, List, Union
class WebscraperTool(BuiltinTool): class WebscraperTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
try: try:
url = tool_paramters.get('url', '') url = tool_parameters.get('url', '')
user_agent = tool_paramters.get('user_agent', '') user_agent = tool_parameters.get('user_agent', '')
if not url: if not url:
return self.create_text_message('Please input url') return self.create_text_message('Please input url')
......
...@@ -14,7 +14,7 @@ class WebscraperProvider(BuiltinToolProviderController): ...@@ -14,7 +14,7 @@ class WebscraperProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
'url': 'https://www.google.com', 'url': 'https://www.google.com',
'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ' 'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
}, },
......
...@@ -14,12 +14,12 @@ class WikipediaInput(BaseModel): ...@@ -14,12 +14,12 @@ class WikipediaInput(BaseModel):
class WikiPediaSearchTool(BuiltinTool): class WikiPediaSearchTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters.get('query', '') query = tool_parameters.get('query', '')
if not query: if not query:
return self.create_text_message('Please input query') return self.create_text_message('Please input query')
......
...@@ -12,7 +12,7 @@ class WikiPediaProvider(BuiltinToolProviderController): ...@@ -12,7 +12,7 @@ class WikiPediaProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"query": "misaka mikoto", "query": "misaka mikoto",
}, },
) )
......
...@@ -11,12 +11,12 @@ class WolframAlphaTool(BuiltinTool): ...@@ -11,12 +11,12 @@ class WolframAlphaTool(BuiltinTool):
def _invoke(self, def _invoke(self,
user_id: str, user_id: str,
tool_paramters: Dict[str, Any], tool_parameters: Dict[str, Any],
) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters.get('query', '') query = tool_parameters.get('query', '')
if not query: if not query:
return self.create_text_message('Please input query') return self.create_text_message('Please input query')
appid = self.runtime.credentials.get('appid', '') appid = self.runtime.credentials.get('appid', '')
......
...@@ -16,7 +16,7 @@ class GoogleProvider(BuiltinToolProviderController): ...@@ -16,7 +16,7 @@ class GoogleProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"query": "1+2+....+111", "query": "1+2+....+111",
}, },
) )
......
...@@ -9,23 +9,23 @@ from yfinance import download ...@@ -9,23 +9,23 @@ from yfinance import download
import pandas as pd import pandas as pd
class YahooFinanceAnalyticsTool(BuiltinTool): class YahooFinanceAnalyticsTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
symbol = tool_paramters.get('symbol', '') symbol = tool_parameters.get('symbol', '')
if not symbol: if not symbol:
return self.create_text_message('Please input symbol') return self.create_text_message('Please input symbol')
time_range = [None, None] time_range = [None, None]
start_date = tool_paramters.get('start_date', '') start_date = tool_parameters.get('start_date', '')
if start_date: if start_date:
time_range[0] = start_date time_range[0] = start_date
else: else:
time_range[0] = '1800-01-01' time_range[0] = '1800-01-01'
end_date = tool_paramters.get('end_date', '') end_date = tool_parameters.get('end_date', '')
if end_date: if end_date:
time_range[1] = end_date time_range[1] = end_date
else: else:
......
...@@ -7,13 +7,13 @@ from requests.exceptions import HTTPError, ReadTimeout ...@@ -7,13 +7,13 @@ from requests.exceptions import HTTPError, ReadTimeout
import yfinance import yfinance
class YahooFinanceSearchTickerTool(BuiltinTool): class YahooFinanceSearchTickerTool(BuiltinTool):
def _invoke(self,user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self,user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
''' '''
invoke tools invoke tools
''' '''
query = tool_paramters.get('symbol', '') query = tool_parameters.get('symbol', '')
if not query: if not query:
return self.create_text_message('Please input symbol') return self.create_text_message('Please input symbol')
......
...@@ -7,12 +7,12 @@ from requests.exceptions import HTTPError, ReadTimeout ...@@ -7,12 +7,12 @@ from requests.exceptions import HTTPError, ReadTimeout
from yfinance import Ticker from yfinance import Ticker
class YahooFinanceSearchTickerTool(BuiltinTool): class YahooFinanceSearchTickerTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
query = tool_paramters.get('symbol', '') query = tool_parameters.get('symbol', '')
if not query: if not query:
return self.create_text_message('Please input symbol') return self.create_text_message('Please input symbol')
......
...@@ -12,7 +12,7 @@ class YahooFinanceProvider(BuiltinToolProviderController): ...@@ -12,7 +12,7 @@ class YahooFinanceProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"ticker": "MSFT", "ticker": "MSFT",
}, },
) )
......
...@@ -7,23 +7,23 @@ from datetime import datetime ...@@ -7,23 +7,23 @@ from datetime import datetime
from googleapiclient.discovery import build from googleapiclient.discovery import build
class YoutubeVideosAnalyticsTool(BuiltinTool): class YoutubeVideosAnalyticsTool(BuiltinTool):
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) \ def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) \
-> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
""" """
invoke tools invoke tools
""" """
channel = tool_paramters.get('channel', '') channel = tool_parameters.get('channel', '')
if not channel: if not channel:
return self.create_text_message('Please input symbol') return self.create_text_message('Please input symbol')
time_range = [None, None] time_range = [None, None]
start_date = tool_paramters.get('start_date', '') start_date = tool_parameters.get('start_date', '')
if start_date: if start_date:
time_range[0] = start_date time_range[0] = start_date
else: else:
time_range[0] = '1800-01-01' time_range[0] = '1800-01-01'
end_date = tool_paramters.get('end_date', '') end_date = tool_parameters.get('end_date', '')
if end_date: if end_date:
time_range[1] = end_date time_range[1] = end_date
else: else:
......
...@@ -12,7 +12,7 @@ class YahooFinanceProvider(BuiltinToolProviderController): ...@@ -12,7 +12,7 @@ class YahooFinanceProvider(BuiltinToolProviderController):
} }
).invoke( ).invoke(
user_id='', user_id='',
tool_paramters={ tool_parameters={
"channel": "TOKYO GIRLS COLLECTION", "channel": "TOKYO GIRLS COLLECTION",
"start_date": "2020-01-01", "start_date": "2020-01-01",
"end_date": "2024-12-31", "end_date": "2024-12-31",
......
...@@ -5,13 +5,13 @@ from os import path, listdir ...@@ -5,13 +5,13 @@ from os import path, listdir
from yaml import load, FullLoader from yaml import load, FullLoader
from core.tools.entities.tool_entities import ToolProviderType, \ from core.tools.entities.tool_entities import ToolProviderType, \
ToolParamter, ToolProviderCredentials ToolParameter, ToolProviderCredentials
from core.tools.tool.tool import Tool from core.tools.tool.tool import Tool
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
from core.tools.provider.tool_provider import ToolProviderController from core.tools.provider.tool_provider import ToolProviderController
from core.tools.entities.user_entities import UserToolProviderCredentials from core.tools.entities.user_entities import UserToolProviderCredentials
from core.tools.errors import ToolNotFoundError, ToolProviderNotFoundError, \ from core.tools.errors import ToolNotFoundError, ToolProviderNotFoundError, \
ToolParamterValidationError, ToolProviderCredentialValidationError ToolParameterValidationError, ToolProviderCredentialValidationError
import importlib import importlib
...@@ -109,7 +109,7 @@ class BuiltinToolProviderController(ToolProviderController): ...@@ -109,7 +109,7 @@ class BuiltinToolProviderController(ToolProviderController):
""" """
return next(filter(lambda x: x.identity.name == tool_name, self.get_tools()), None) return next(filter(lambda x: x.identity.name == tool_name, self.get_tools()), None)
def get_parameters(self, tool_name: str) -> List[ToolParamter]: def get_parameters(self, tool_name: str) -> List[ToolParameter]:
""" """
returns the parameters of the tool returns the parameters of the tool
...@@ -148,62 +148,62 @@ class BuiltinToolProviderController(ToolProviderController): ...@@ -148,62 +148,62 @@ class BuiltinToolProviderController(ToolProviderController):
""" """
tool_parameters_schema = self.get_parameters(tool_name) tool_parameters_schema = self.get_parameters(tool_name)
tool_parameters_need_to_validate: Dict[str, ToolParamter] = {} tool_parameters_need_to_validate: Dict[str, ToolParameter] = {}
for parameter in tool_parameters_schema: for parameter in tool_parameters_schema:
tool_parameters_need_to_validate[parameter.name] = parameter tool_parameters_need_to_validate[parameter.name] = parameter
for parameter in tool_parameters: for parameter in tool_parameters:
if parameter not in tool_parameters_need_to_validate: if parameter not in tool_parameters_need_to_validate:
raise ToolParamterValidationError(f'parameter {parameter} not found in tool {tool_name}') raise ToolParameterValidationError(f'parameter {parameter} not found in tool {tool_name}')
# check type # check type
parameter_schema = tool_parameters_need_to_validate[parameter] parameter_schema = tool_parameters_need_to_validate[parameter]
if parameter_schema.type == ToolParamter.ToolParameterType.STRING: if parameter_schema.type == ToolParameter.ToolParameterType.STRING:
if not isinstance(tool_parameters[parameter], str): if not isinstance(tool_parameters[parameter], str):
raise ToolParamterValidationError(f'parameter {parameter} should be string') raise ToolParameterValidationError(f'parameter {parameter} should be string')
elif parameter_schema.type == ToolParamter.ToolParameterType.NUMBER: elif parameter_schema.type == ToolParameter.ToolParameterType.NUMBER:
if not isinstance(tool_parameters[parameter], (int, float)): if not isinstance(tool_parameters[parameter], (int, float)):
raise ToolParamterValidationError(f'parameter {parameter} should be number') raise ToolParameterValidationError(f'parameter {parameter} should be number')
if parameter_schema.min is not None and tool_parameters[parameter] < parameter_schema.min: if parameter_schema.min is not None and tool_parameters[parameter] < parameter_schema.min:
raise ToolParamterValidationError(f'parameter {parameter} should be greater than {parameter_schema.min}') raise ToolParameterValidationError(f'parameter {parameter} should be greater than {parameter_schema.min}')
if parameter_schema.max is not None and tool_parameters[parameter] > parameter_schema.max: if parameter_schema.max is not None and tool_parameters[parameter] > parameter_schema.max:
raise ToolParamterValidationError(f'parameter {parameter} should be less than {parameter_schema.max}') raise ToolParameterValidationError(f'parameter {parameter} should be less than {parameter_schema.max}')
elif parameter_schema.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter_schema.type == ToolParameter.ToolParameterType.BOOLEAN:
if not isinstance(tool_parameters[parameter], bool): if not isinstance(tool_parameters[parameter], bool):
raise ToolParamterValidationError(f'parameter {parameter} should be boolean') raise ToolParameterValidationError(f'parameter {parameter} should be boolean')
elif parameter_schema.type == ToolParamter.ToolParameterType.SELECT: elif parameter_schema.type == ToolParameter.ToolParameterType.SELECT:
if not isinstance(tool_parameters[parameter], str): if not isinstance(tool_parameters[parameter], str):
raise ToolParamterValidationError(f'parameter {parameter} should be string') raise ToolParameterValidationError(f'parameter {parameter} should be string')
options = parameter_schema.options options = parameter_schema.options
if not isinstance(options, list): if not isinstance(options, list):
raise ToolParamterValidationError(f'parameter {parameter} options should be list') raise ToolParameterValidationError(f'parameter {parameter} options should be list')
if tool_parameters[parameter] not in [x.value for x in options]: if tool_parameters[parameter] not in [x.value for x in options]:
raise ToolParamterValidationError(f'parameter {parameter} should be one of {options}') raise ToolParameterValidationError(f'parameter {parameter} should be one of {options}')
tool_parameters_need_to_validate.pop(parameter) tool_parameters_need_to_validate.pop(parameter)
for parameter in tool_parameters_need_to_validate: for parameter in tool_parameters_need_to_validate:
parameter_schema = tool_parameters_need_to_validate[parameter] parameter_schema = tool_parameters_need_to_validate[parameter]
if parameter_schema.required: if parameter_schema.required:
raise ToolParamterValidationError(f'parameter {parameter} is required') raise ToolParameterValidationError(f'parameter {parameter} is required')
# the parameter is not set currently, set the default value if needed # the parameter is not set currently, set the default value if needed
if parameter_schema.default is not None: if parameter_schema.default is not None:
default_value = parameter_schema.default default_value = parameter_schema.default
# parse default value into the correct type # parse default value into the correct type
if parameter_schema.type == ToolParamter.ToolParameterType.STRING or \ if parameter_schema.type == ToolParameter.ToolParameterType.STRING or \
parameter_schema.type == ToolParamter.ToolParameterType.SELECT: parameter_schema.type == ToolParameter.ToolParameterType.SELECT:
default_value = str(default_value) default_value = str(default_value)
elif parameter_schema.type == ToolParamter.ToolParameterType.NUMBER: elif parameter_schema.type == ToolParameter.ToolParameterType.NUMBER:
default_value = float(default_value) default_value = float(default_value)
elif parameter_schema.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter_schema.type == ToolParameter.ToolParameterType.BOOLEAN:
default_value = bool(default_value) default_value = bool(default_value)
tool_parameters[parameter] = default_value tool_parameters[parameter] = default_value
......
...@@ -4,11 +4,11 @@ from typing import List, Dict, Any, Optional ...@@ -4,11 +4,11 @@ from typing import List, Dict, Any, Optional
from pydantic import BaseModel from pydantic import BaseModel
from core.tools.entities.tool_entities import ToolProviderType, \ from core.tools.entities.tool_entities import ToolProviderType, \
ToolProviderIdentity, ToolParamter, ToolProviderCredentials ToolProviderIdentity, ToolParameter, ToolProviderCredentials
from core.tools.tool.tool import Tool from core.tools.tool.tool import Tool
from core.tools.entities.user_entities import UserToolProviderCredentials from core.tools.entities.user_entities import UserToolProviderCredentials
from core.tools.errors import ToolNotFoundError, \ from core.tools.errors import ToolNotFoundError, \
ToolParamterValidationError, ToolProviderCredentialValidationError ToolParameterValidationError, ToolProviderCredentialValidationError
class ToolProviderController(BaseModel, ABC): class ToolProviderController(BaseModel, ABC):
identity: Optional[ToolProviderIdentity] = None identity: Optional[ToolProviderIdentity] = None
...@@ -50,7 +50,7 @@ class ToolProviderController(BaseModel, ABC): ...@@ -50,7 +50,7 @@ class ToolProviderController(BaseModel, ABC):
""" """
pass pass
def get_parameters(self, tool_name: str) -> List[ToolParamter]: def get_parameters(self, tool_name: str) -> List[ToolParameter]:
""" """
returns the parameters of the tool returns the parameters of the tool
...@@ -80,62 +80,62 @@ class ToolProviderController(BaseModel, ABC): ...@@ -80,62 +80,62 @@ class ToolProviderController(BaseModel, ABC):
""" """
tool_parameters_schema = self.get_parameters(tool_name) tool_parameters_schema = self.get_parameters(tool_name)
tool_parameters_need_to_validate: Dict[str, ToolParamter] = {} tool_parameters_need_to_validate: Dict[str, ToolParameter] = {}
for parameter in tool_parameters_schema: for parameter in tool_parameters_schema:
tool_parameters_need_to_validate[parameter.name] = parameter tool_parameters_need_to_validate[parameter.name] = parameter
for parameter in tool_parameters: for parameter in tool_parameters:
if parameter not in tool_parameters_need_to_validate: if parameter not in tool_parameters_need_to_validate:
raise ToolParamterValidationError(f'parameter {parameter} not found in tool {tool_name}') raise ToolParameterValidationError(f'parameter {parameter} not found in tool {tool_name}')
# check type # check type
parameter_schema = tool_parameters_need_to_validate[parameter] parameter_schema = tool_parameters_need_to_validate[parameter]
if parameter_schema.type == ToolParamter.ToolParameterType.STRING: if parameter_schema.type == ToolParameter.ToolParameterType.STRING:
if not isinstance(tool_parameters[parameter], str): if not isinstance(tool_parameters[parameter], str):
raise ToolParamterValidationError(f'parameter {parameter} should be string') raise ToolParameterValidationError(f'parameter {parameter} should be string')
elif parameter_schema.type == ToolParamter.ToolParameterType.NUMBER: elif parameter_schema.type == ToolParameter.ToolParameterType.NUMBER:
if not isinstance(tool_parameters[parameter], (int, float)): if not isinstance(tool_parameters[parameter], (int, float)):
raise ToolParamterValidationError(f'parameter {parameter} should be number') raise ToolParameterValidationError(f'parameter {parameter} should be number')
if parameter_schema.min is not None and tool_parameters[parameter] < parameter_schema.min: if parameter_schema.min is not None and tool_parameters[parameter] < parameter_schema.min:
raise ToolParamterValidationError(f'parameter {parameter} should be greater than {parameter_schema.min}') raise ToolParameterValidationError(f'parameter {parameter} should be greater than {parameter_schema.min}')
if parameter_schema.max is not None and tool_parameters[parameter] > parameter_schema.max: if parameter_schema.max is not None and tool_parameters[parameter] > parameter_schema.max:
raise ToolParamterValidationError(f'parameter {parameter} should be less than {parameter_schema.max}') raise ToolParameterValidationError(f'parameter {parameter} should be less than {parameter_schema.max}')
elif parameter_schema.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter_schema.type == ToolParameter.ToolParameterType.BOOLEAN:
if not isinstance(tool_parameters[parameter], bool): if not isinstance(tool_parameters[parameter], bool):
raise ToolParamterValidationError(f'parameter {parameter} should be boolean') raise ToolParameterValidationError(f'parameter {parameter} should be boolean')
elif parameter_schema.type == ToolParamter.ToolParameterType.SELECT: elif parameter_schema.type == ToolParameter.ToolParameterType.SELECT:
if not isinstance(tool_parameters[parameter], str): if not isinstance(tool_parameters[parameter], str):
raise ToolParamterValidationError(f'parameter {parameter} should be string') raise ToolParameterValidationError(f'parameter {parameter} should be string')
options = parameter_schema.options options = parameter_schema.options
if not isinstance(options, list): if not isinstance(options, list):
raise ToolParamterValidationError(f'parameter {parameter} options should be list') raise ToolParameterValidationError(f'parameter {parameter} options should be list')
if tool_parameters[parameter] not in [x.value for x in options]: if tool_parameters[parameter] not in [x.value for x in options]:
raise ToolParamterValidationError(f'parameter {parameter} should be one of {options}') raise ToolParameterValidationError(f'parameter {parameter} should be one of {options}')
tool_parameters_need_to_validate.pop(parameter) tool_parameters_need_to_validate.pop(parameter)
for parameter in tool_parameters_need_to_validate: for parameter in tool_parameters_need_to_validate:
parameter_schema = tool_parameters_need_to_validate[parameter] parameter_schema = tool_parameters_need_to_validate[parameter]
if parameter_schema.required: if parameter_schema.required:
raise ToolParamterValidationError(f'parameter {parameter} is required') raise ToolParameterValidationError(f'parameter {parameter} is required')
# the parameter is not set currently, set the default value if needed # the parameter is not set currently, set the default value if needed
if parameter_schema.default is not None: if parameter_schema.default is not None:
default_value = parameter_schema.default default_value = parameter_schema.default
# parse default value into the correct type # parse default value into the correct type
if parameter_schema.type == ToolParamter.ToolParameterType.STRING or \ if parameter_schema.type == ToolParameter.ToolParameterType.STRING or \
parameter_schema.type == ToolParamter.ToolParameterType.SELECT: parameter_schema.type == ToolParameter.ToolParameterType.SELECT:
default_value = str(default_value) default_value = str(default_value)
elif parameter_schema.type == ToolParamter.ToolParameterType.NUMBER: elif parameter_schema.type == ToolParameter.ToolParameterType.NUMBER:
default_value = float(default_value) default_value = float(default_value)
elif parameter_schema.type == ToolParamter.ToolParameterType.BOOLEAN: elif parameter_schema.type == ToolParameter.ToolParameterType.BOOLEAN:
default_value = bool(default_value) default_value = bool(default_value)
tool_parameters[parameter] = default_value tool_parameters[parameter] = default_value
......
...@@ -223,15 +223,15 @@ class ApiTool(Tool): ...@@ -223,15 +223,15 @@ class ApiTool(Tool):
return response return response
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> ToolInvokeMessage | List[ToolInvokeMessage]: def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> ToolInvokeMessage | List[ToolInvokeMessage]:
""" """
invoke http request invoke http request
""" """
# assemble request # assemble request
headers = self.assembling_request(tool_paramters) headers = self.assembling_request(tool_parameters)
# do http request # do http request
response = self.do_http_request(self.api_bundle.server_url, self.api_bundle.method, headers, tool_paramters) response = self.do_http_request(self.api_bundle.server_url, self.api_bundle.method, headers, tool_parameters)
# validate response # validate response
response = self.validate_and_parse_response(response) response = self.validate_and_parse_response(response)
......
from typing import Any, Dict, List, Union from typing import Any, Dict, List, Union
from core.features.dataset_retrieval import DatasetRetrievalFeature from core.features.dataset_retrieval import DatasetRetrievalFeature
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParamter, ToolIdentity, ToolDescription from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter, ToolIdentity, ToolDescription
from core.tools.tool.tool import Tool from core.tools.tool.tool import Tool
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
...@@ -63,23 +63,23 @@ class DatasetRetrieverTool(Tool): ...@@ -63,23 +63,23 @@ class DatasetRetrieverTool(Tool):
return tools return tools
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
return [ return [
ToolParamter(name='query', ToolParameter(name='query',
label=I18nObject(en_US='', zh_Hans=''), label=I18nObject(en_US='', zh_Hans=''),
human_description=I18nObject(en_US='', zh_Hans=''), human_description=I18nObject(en_US='', zh_Hans=''),
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
form=ToolParamter.ToolParameterForm.LLM, form=ToolParameter.ToolParameterForm.LLM,
llm_description='Query for the dataset to be used to retrieve the dataset.', llm_description='Query for the dataset to be used to retrieve the dataset.',
required=True, required=True,
default=''), default=''),
] ]
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> ToolInvokeMessage | List[ToolInvokeMessage]: def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> ToolInvokeMessage | List[ToolInvokeMessage]:
""" """
invoke dataset retriever tool invoke dataset retriever tool
""" """
query = tool_paramters.get('query', None) query = tool_parameters.get('query', None)
if not query: if not query:
return self.create_text_message(text='please input query') return self.create_text_message(text='please input query')
......
...@@ -5,13 +5,13 @@ from abc import abstractmethod, ABC ...@@ -5,13 +5,13 @@ from abc import abstractmethod, ABC
from enum import Enum from enum import Enum
from core.tools.entities.tool_entities import ToolIdentity, ToolInvokeMessage,\ from core.tools.entities.tool_entities import ToolIdentity, ToolInvokeMessage,\
ToolParamter, ToolDescription, ToolRuntimeVariablePool, ToolRuntimeVariable, ToolRuntimeImageVariable ToolParameter, ToolDescription, ToolRuntimeVariablePool, ToolRuntimeVariable, ToolRuntimeImageVariable
from core.tools.tool_file_manager import ToolFileManager from core.tools.tool_file_manager import ToolFileManager
from core.callback_handler.agent_tool_callback_handler import DifyAgentCallbackHandler from core.callback_handler.agent_tool_callback_handler import DifyAgentCallbackHandler
class Tool(BaseModel, ABC): class Tool(BaseModel, ABC):
identity: ToolIdentity = None identity: ToolIdentity = None
parameters: Optional[List[ToolParamter]] = None parameters: Optional[List[ToolParameter]] = None
description: ToolDescription = None description: ToolDescription = None
is_team_authorization: bool = False is_team_authorization: bool = False
agent_callback: Optional[DifyAgentCallbackHandler] = None agent_callback: Optional[DifyAgentCallbackHandler] = None
...@@ -166,22 +166,22 @@ class Tool(BaseModel, ABC): ...@@ -166,22 +166,22 @@ class Tool(BaseModel, ABC):
return result return result
def invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> List[ToolInvokeMessage]: def invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> List[ToolInvokeMessage]:
# update tool_paramters # update tool_parameters
if self.runtime.runtime_parameters: if self.runtime.runtime_parameters:
tool_paramters.update(self.runtime.runtime_parameters) tool_parameters.update(self.runtime.runtime_parameters)
# hit callback # hit callback
if self.use_callback: if self.use_callback:
self.agent_callback.on_tool_start( self.agent_callback.on_tool_start(
tool_name=self.identity.name, tool_name=self.identity.name,
tool_inputs=tool_paramters tool_inputs=tool_parameters
) )
try: try:
result = self._invoke( result = self._invoke(
user_id=user_id, user_id=user_id,
tool_paramters=tool_paramters, tool_parameters=tool_parameters,
) )
except Exception as e: except Exception as e:
if self.use_callback: if self.use_callback:
...@@ -195,7 +195,7 @@ class Tool(BaseModel, ABC): ...@@ -195,7 +195,7 @@ class Tool(BaseModel, ABC):
if self.use_callback: if self.use_callback:
self.agent_callback.on_tool_end( self.agent_callback.on_tool_end(
tool_name=self.identity.name, tool_name=self.identity.name,
tool_inputs=tool_paramters, tool_inputs=tool_parameters,
tool_outputs=self._convert_tool_response_to_str(result) tool_outputs=self._convert_tool_response_to_str(result)
) )
...@@ -210,7 +210,7 @@ class Tool(BaseModel, ABC): ...@@ -210,7 +210,7 @@ class Tool(BaseModel, ABC):
if response.type == ToolInvokeMessage.MessageType.TEXT: if response.type == ToolInvokeMessage.MessageType.TEXT:
result += response.message result += response.message
elif response.type == ToolInvokeMessage.MessageType.LINK: elif response.type == ToolInvokeMessage.MessageType.LINK:
result += f"result link: {response.message}. please dirct user to check it." result += f"result link: {response.message}. please tell user to check it."
elif response.type == ToolInvokeMessage.MessageType.IMAGE_LINK or \ elif response.type == ToolInvokeMessage.MessageType.IMAGE_LINK or \
response.type == ToolInvokeMessage.MessageType.IMAGE: response.type == ToolInvokeMessage.MessageType.IMAGE:
result += f"image has been created and sent to user already, you should tell user to check it now." result += f"image has been created and sent to user already, you should tell user to check it now."
...@@ -225,7 +225,7 @@ class Tool(BaseModel, ABC): ...@@ -225,7 +225,7 @@ class Tool(BaseModel, ABC):
return result return result
@abstractmethod @abstractmethod
def _invoke(self, user_id: str, tool_paramters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]: def _invoke(self, user_id: str, tool_parameters: Dict[str, Any]) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
pass pass
def validate_credentials(self, credentials: Dict[str, Any], parameters: Dict[str, Any]) -> None: def validate_credentials(self, credentials: Dict[str, Any], parameters: Dict[str, Any]) -> None:
...@@ -237,7 +237,7 @@ class Tool(BaseModel, ABC): ...@@ -237,7 +237,7 @@ class Tool(BaseModel, ABC):
""" """
pass pass
def get_runtime_parameters(self) -> List[ToolParamter]: def get_runtime_parameters(self) -> List[ToolParameter]:
""" """
get the runtime parameters get the runtime parameters
...@@ -247,11 +247,11 @@ class Tool(BaseModel, ABC): ...@@ -247,11 +247,11 @@ class Tool(BaseModel, ABC):
""" """
return self.parameters return self.parameters
def is_tool_avaliable(self) -> bool: def is_tool_available(self) -> bool:
""" """
check if the tool is avaliable check if the tool is available
:return: if the tool is avaliable :return: if the tool is available
""" """
return True return True
......
...@@ -12,7 +12,7 @@ from core.tools.errors import ToolProviderNotFoundError ...@@ -12,7 +12,7 @@ from core.tools.errors import ToolProviderNotFoundError
from core.tools.provider.api_tool_provider import ApiBasedToolProviderController from core.tools.provider.api_tool_provider import ApiBasedToolProviderController
from core.tools.provider.app_tool_provider import AppBasedToolProviderEntity from core.tools.provider.app_tool_provider import AppBasedToolProviderEntity
from core.tools.entities.user_entities import UserToolProvider from core.tools.entities.user_entities import UserToolProvider
from core.tools.utils.configration import ToolConfiguration from core.tools.utils.configuration import ToolConfiguration
from core.tools.utils.encoder import serialize_base_model_dict from core.tools.utils.encoder import serialize_base_model_dict
from core.tools.provider.builtin._positions import BuiltinToolProviderSort from core.tools.provider.builtin._positions import BuiltinToolProviderSort
...@@ -115,7 +115,7 @@ class ToolManager: ...@@ -115,7 +115,7 @@ class ToolManager:
return tool return tool
@staticmethod @staticmethod
def get_tool(provider_type: str, provider_id: str, tool_name: str, tanent_id: str = None) \ def get_tool(provider_type: str, provider_id: str, tool_name: str, tenant_id: str = None) \
-> Union[BuiltinTool, ApiTool]: -> Union[BuiltinTool, ApiTool]:
""" """
get the tool get the tool
...@@ -129,9 +129,9 @@ class ToolManager: ...@@ -129,9 +129,9 @@ class ToolManager:
if provider_type == 'builtin': if provider_type == 'builtin':
return ToolManager.get_builtin_tool(provider_id, tool_name) return ToolManager.get_builtin_tool(provider_id, tool_name)
elif provider_type == 'api': elif provider_type == 'api':
if tanent_id is None: if tenant_id is None:
raise ValueError('tanent id is required for api provider') raise ValueError('tenant id is required for api provider')
api_provider, _ = ToolManager.get_api_provider_controller(tanent_id, provider_id) api_provider, _ = ToolManager.get_api_provider_controller(tenant_id, provider_id)
return api_provider.get_tool(tool_name) return api_provider.get_tool(tool_name)
elif provider_type == 'app': elif provider_type == 'app':
raise NotImplementedError('app provider not implemented') raise NotImplementedError('app provider not implemented')
...@@ -139,7 +139,7 @@ class ToolManager: ...@@ -139,7 +139,7 @@ class ToolManager:
raise ToolProviderNotFoundError(f'provider type {provider_type} not found') raise ToolProviderNotFoundError(f'provider type {provider_type} not found')
@staticmethod @staticmethod
def get_tool_runtime(provider_type: str, provider_name: str, tool_name: str, tanent_id, def get_tool_runtime(provider_type: str, provider_name: str, tool_name: str, tenant_id,
agent_callback: DifyAgentCallbackHandler = None) \ agent_callback: DifyAgentCallbackHandler = None) \
-> Union[BuiltinTool, ApiTool]: -> Union[BuiltinTool, ApiTool]:
""" """
...@@ -158,13 +158,13 @@ class ToolManager: ...@@ -158,13 +158,13 @@ class ToolManager:
provider_controller = ToolManager.get_builtin_provider(provider_name) provider_controller = ToolManager.get_builtin_provider(provider_name)
if not provider_controller.need_credentials: if not provider_controller.need_credentials:
return builtin_tool.fork_tool_runtime(meta={ return builtin_tool.fork_tool_runtime(meta={
'tenant_id': tanent_id, 'tenant_id': tenant_id,
'credentials': {}, 'credentials': {},
}, agent_callback=agent_callback) }, agent_callback=agent_callback)
# get credentials # get credentials
builtin_provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter( builtin_provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter(
BuiltinToolProvider.tenant_id == tanent_id, BuiltinToolProvider.tenant_id == tenant_id,
BuiltinToolProvider.provider == provider_name, BuiltinToolProvider.provider == provider_name,
).first() ).first()
...@@ -174,28 +174,28 @@ class ToolManager: ...@@ -174,28 +174,28 @@ class ToolManager:
# decrypt the credentials # decrypt the credentials
credentials = builtin_provider.credentials credentials = builtin_provider.credentials
controller = ToolManager.get_builtin_provider(provider_name) controller = ToolManager.get_builtin_provider(provider_name)
tool_configuration = ToolConfiguration(tenant_id=tanent_id, provider_controller=controller) tool_configuration = ToolConfiguration(tenant_id=tenant_id, provider_controller=controller)
decrypted_credentials = tool_configuration.decrypt_tool_credentials(credentials) decrypted_credentials = tool_configuration.decrypt_tool_credentials(credentials)
return builtin_tool.fork_tool_runtime(meta={ return builtin_tool.fork_tool_runtime(meta={
'tenant_id': tanent_id, 'tenant_id': tenant_id,
'credentials': decrypted_credentials, 'credentials': decrypted_credentials,
'runtime_parameters': {} 'runtime_parameters': {}
}, agent_callback=agent_callback) }, agent_callback=agent_callback)
elif provider_type == 'api': elif provider_type == 'api':
if tanent_id is None: if tenant_id is None:
raise ValueError('tanent id is required for api provider') raise ValueError('tenant id is required for api provider')
api_provider, credentials = ToolManager.get_api_provider_controller(tanent_id, provider_name) api_provider, credentials = ToolManager.get_api_provider_controller(tenant_id, provider_name)
# decrypt the credentials # decrypt the credentials
tool_configuration = ToolConfiguration(tenant_id=tanent_id, provider_controller=api_provider) tool_configuration = ToolConfiguration(tenant_id=tenant_id, provider_controller=api_provider)
decrypted_credentials = tool_configuration.decrypt_tool_credentials(credentials) decrypted_credentials = tool_configuration.decrypt_tool_credentials(credentials)
return api_provider.get_tool(tool_name).fork_tool_runtime(meta={ return api_provider.get_tool(tool_name).fork_tool_runtime(meta={
'tenant_id': tanent_id, 'tenant_id': tenant_id,
'credentials': decrypted_credentials, 'credentials': decrypted_credentials,
}) })
elif provider_type == 'app': elif provider_type == 'app':
...@@ -321,7 +321,7 @@ class ToolManager: ...@@ -321,7 +321,7 @@ class ToolManager:
schema = provider.get_credentials_schema() schema = provider.get_credentials_schema()
for name, value in schema.items(): for name, value in schema.items():
result_providers[provider.identity.name].team_credentials[name] = \ result_providers[provider.identity.name].team_credentials[name] = \
ToolProviderCredentials.CredentialsType.defaut(value.type) ToolProviderCredentials.CredentialsType.default(value.type)
# check if the provider need credentials # check if the provider need credentials
if not provider.need_credentials: if not provider.need_credentials:
...@@ -398,7 +398,7 @@ class ToolManager: ...@@ -398,7 +398,7 @@ class ToolManager:
return BuiltinToolProviderSort.sort(list(result_providers.values())) return BuiltinToolProviderSort.sort(list(result_providers.values()))
@staticmethod @staticmethod
def get_api_provider_controller(tanent_id: str, provider_id: str) -> Tuple[ApiBasedToolProviderController, Dict[str, Any]]: def get_api_provider_controller(tenant_id: str, provider_id: str) -> Tuple[ApiBasedToolProviderController, Dict[str, Any]]:
""" """
get the api provider get the api provider
...@@ -408,7 +408,7 @@ class ToolManager: ...@@ -408,7 +408,7 @@ class ToolManager:
""" """
provider: ApiToolProvider = db.session.query(ApiToolProvider).filter( provider: ApiToolProvider = db.session.query(ApiToolProvider).filter(
ApiToolProvider.id == provider_id, ApiToolProvider.id == provider_id,
ApiToolProvider.tenant_id == tanent_id, ApiToolProvider.tenant_id == tenant_id,
).first() ).first()
if provider is None: if provider is None:
...@@ -435,7 +435,7 @@ class ToolManager: ...@@ -435,7 +435,7 @@ class ToolManager:
).first() ).first()
if provider is None: if provider is None:
raise ValueError(f'yout have not added provider {provider}') raise ValueError(f'you have not added provider {provider}')
try: try:
credentials = json.loads(provider.credentials_str) or {} credentials = json.loads(provider.credentials_str) or {}
......
...@@ -17,7 +17,7 @@ class ToolConfiguration(BaseModel): ...@@ -17,7 +17,7 @@ class ToolConfiguration(BaseModel):
def encrypt_tool_credentials(self, credentials: Dict[str, str]) -> Dict[str, str]: def encrypt_tool_credentials(self, credentials: Dict[str, str]) -> Dict[str, str]:
""" """
encrypt tool credentials with tanent id encrypt tool credentials with tenant id
return a deep copy of credentials with encrypted values return a deep copy of credentials with encrypted values
""" """
...@@ -58,7 +58,7 @@ class ToolConfiguration(BaseModel): ...@@ -58,7 +58,7 @@ class ToolConfiguration(BaseModel):
def decrypt_tool_credentials(self, credentials: Dict[str, str]) -> Dict[str, str]: def decrypt_tool_credentials(self, credentials: Dict[str, str]) -> Dict[str, str]:
""" """
decrypt tool credentials with tanent id decrypt tool credentials with tenant id
return a deep copy of credentials with decrypted values return a deep copy of credentials with decrypted values
""" """
......
from core.tools.entities.tool_bundle import ApiBasedToolBundle from core.tools.entities.tool_bundle import ApiBasedToolBundle
from core.tools.entities.tool_entities import ToolParamter, ToolParamterOption, ApiProviderSchemaType from core.tools.entities.tool_entities import ToolParameter, ToolParameterOption, ApiProviderSchemaType
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
from core.tools.errors import ToolProviderNotFoundError, ToolNotSupportedError, \ from core.tools.errors import ToolProviderNotFoundError, ToolNotSupportedError, \
ToolApiSchemaError ToolApiSchemaError
...@@ -47,7 +47,7 @@ class ApiBasedToolSchemaParser: ...@@ -47,7 +47,7 @@ class ApiBasedToolSchemaParser:
parameters = [] parameters = []
if 'parameters' in interface['operation']: if 'parameters' in interface['operation']:
for parameter in interface['operation']['parameters']: for parameter in interface['operation']['parameters']:
parameters.append(ToolParamter( parameters.append(ToolParameter(
name=parameter['name'], name=parameter['name'],
label=I18nObject( label=I18nObject(
en_US=parameter['name'], en_US=parameter['name'],
...@@ -57,9 +57,9 @@ class ApiBasedToolSchemaParser: ...@@ -57,9 +57,9 @@ class ApiBasedToolSchemaParser:
en_US=parameter.get('description', ''), en_US=parameter.get('description', ''),
zh_Hans=parameter.get('description', '') zh_Hans=parameter.get('description', '')
), ),
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
required=parameter.get('required', False), required=parameter.get('required', False),
form=ToolParamter.ToolParameterForm.LLM, form=ToolParameter.ToolParameterForm.LLM,
llm_description=parameter.get('description'), llm_description=parameter.get('description'),
default=parameter['default'] if 'default' in parameter else None, default=parameter['default'] if 'default' in parameter else None,
)) ))
...@@ -87,7 +87,7 @@ class ApiBasedToolSchemaParser: ...@@ -87,7 +87,7 @@ class ApiBasedToolSchemaParser:
required = body_schema['required'] if 'required' in body_schema else [] required = body_schema['required'] if 'required' in body_schema else []
properties = body_schema['properties'] if 'properties' in body_schema else {} properties = body_schema['properties'] if 'properties' in body_schema else {}
for name, property in properties.items(): for name, property in properties.items():
parameters.append(ToolParamter( parameters.append(ToolParameter(
name=name, name=name,
label=I18nObject( label=I18nObject(
en_US=name, en_US=name,
...@@ -97,9 +97,9 @@ class ApiBasedToolSchemaParser: ...@@ -97,9 +97,9 @@ class ApiBasedToolSchemaParser:
en_US=property['description'] if 'description' in property else '', en_US=property['description'] if 'description' in property else '',
zh_Hans=property['description'] if 'description' in property else '' zh_Hans=property['description'] if 'description' in property else ''
), ),
type=ToolParamter.ToolParameterType.STRING, type=ToolParameter.ToolParameterType.STRING,
required=name in required, required=name in required,
form=ToolParamter.ToolParameterForm.LLM, form=ToolParameter.ToolParameterForm.LLM,
llm_description=property['description'] if 'description' in property else '', llm_description=property['description'] if 'description' in property else '',
default=property['default'] if 'default' in property else None, default=property['default'] if 'default' in property else None,
)) ))
......
...@@ -100,7 +100,7 @@ class ApiToolProvider(db.Model): ...@@ -100,7 +100,7 @@ class ApiToolProvider(db.Model):
schema_type_str = db.Column(db.String(40), nullable=False) schema_type_str = db.Column(db.String(40), nullable=False)
# who created this tool # who created this tool
user_id = db.Column(UUID, nullable=False) user_id = db.Column(UUID, nullable=False)
# tanent id # tenant id
tenant_id = db.Column(UUID, nullable=False) tenant_id = db.Column(UUID, nullable=False)
# description of the provider # description of the provider
description = db.Column(db.Text, nullable=False) description = db.Column(db.Text, nullable=False)
...@@ -135,7 +135,7 @@ class ApiToolProvider(db.Model): ...@@ -135,7 +135,7 @@ class ApiToolProvider(db.Model):
return db.session.query(Account).filter(Account.id == self.user_id).first() return db.session.query(Account).filter(Account.id == self.user_id).first()
@property @property
def tanent(self) -> Tenant: def tenant(self) -> Tenant:
return db.session.query(Tenant).filter(Tenant.id == self.tenant_id).first() return db.session.query(Tenant).filter(Tenant.id == self.tenant_id).first()
class ToolModelInvoke(db.Model): class ToolModelInvoke(db.Model):
...@@ -150,7 +150,7 @@ class ToolModelInvoke(db.Model): ...@@ -150,7 +150,7 @@ class ToolModelInvoke(db.Model):
id = db.Column(UUID, server_default=db.text('uuid_generate_v4()')) id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
# who invoke this tool # who invoke this tool
user_id = db.Column(UUID, nullable=False) user_id = db.Column(UUID, nullable=False)
# tanent id # tenant id
tenant_id = db.Column(UUID, nullable=False) tenant_id = db.Column(UUID, nullable=False)
# provider # provider
provider = db.Column(db.String(40), nullable=False) provider = db.Column(db.String(40), nullable=False)
...@@ -190,7 +190,7 @@ class ToolConversationVariables(db.Model): ...@@ -190,7 +190,7 @@ class ToolConversationVariables(db.Model):
id = db.Column(UUID, server_default=db.text('uuid_generate_v4()')) id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
# conversation user id # conversation user id
user_id = db.Column(UUID, nullable=False) user_id = db.Column(UUID, nullable=False)
# tanent id # tenant id
tenant_id = db.Column(UUID, nullable=False) tenant_id = db.Column(UUID, nullable=False)
# conversation id # conversation id
conversation_id = db.Column(UUID, nullable=False) conversation_id = db.Column(UUID, nullable=False)
...@@ -218,7 +218,7 @@ class ToolFile(db.Model): ...@@ -218,7 +218,7 @@ class ToolFile(db.Model):
id = db.Column(UUID, server_default=db.text('uuid_generate_v4()')) id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
# conversation user id # conversation user id
user_id = db.Column(UUID, nullable=False) user_id = db.Column(UUID, nullable=False)
# tanent id # tenant id
tenant_id = db.Column(UUID, nullable=False) tenant_id = db.Column(UUID, nullable=False)
# conversation id # conversation id
conversation_id = db.Column(UUID, nullable=False) conversation_id = db.Column(UUID, nullable=False)
......
...@@ -12,7 +12,7 @@ from core.tools.provider.tool_provider import ToolProviderController ...@@ -12,7 +12,7 @@ from core.tools.provider.tool_provider import ToolProviderController
from core.tools.provider.api_tool_provider import ApiBasedToolProviderController from core.tools.provider.api_tool_provider import ApiBasedToolProviderController
from core.tools.utils.parser import ApiBasedToolSchemaParser 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.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 core.tools.errors import ToolProviderCredentialValidationError, ToolProviderNotFoundError, ToolNotFoundError
from extensions.ext_database import db from extensions.ext_database import db
...@@ -24,26 +24,26 @@ import json ...@@ -24,26 +24,26 @@ import json
class ToolManageService: class ToolManageService:
@staticmethod @staticmethod
def list_tool_providers(user_id: str, tanent_id: str): def list_tool_providers(user_id: str, tenant_id: str):
""" """
list tool providers list tool providers
:return: the list of tool providers :return: the list of tool providers
""" """
result = [provider.to_dict() for provider in ToolManager.user_list_providers( result = [provider.to_dict() for provider in ToolManager.user_list_providers(
user_id, tanent_id user_id, tenant_id
)] )]
# add icon url prefix # add icon url prefix
for provider in result: for provider in result:
ToolManageService.repacket_provider(provider) ToolManageService.repack_provider(provider)
return result return result
@staticmethod @staticmethod
def repacket_provider(provider: dict): def repack_provider(provider: dict):
""" """
repacket provider repack provider
:param provider: the provider dict :param provider: the provider dict
""" """
...@@ -286,7 +286,7 @@ class ToolManageService: ...@@ -286,7 +286,7 @@ class ToolManageService:
).first() ).first()
if provider is None: 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( return json.loads(
serialize_base_model_array([ serialize_base_model_array([
...@@ -424,7 +424,7 @@ class ToolManageService: ...@@ -424,7 +424,7 @@ class ToolManageService:
).first() ).first()
if provider is None: 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.delete(provider)
db.session.commit() db.session.commit()
...@@ -457,7 +457,7 @@ class ToolManageService: ...@@ -457,7 +457,7 @@ class ToolManageService:
).first() ).first()
if provider is None: 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.delete(provider)
db.session.commit() db.session.commit()
......
""" """
LocalAI Embedding Interface is temporarily unavaliable due to LocalAI Embedding Interface is temporarily unavailable due to
we could not find a way to test it for now. we could not find a way to test it for now.
""" """
\ No newline at end of file
...@@ -9,7 +9,7 @@ const translation = { ...@@ -9,7 +9,7 @@ const translation = {
three: 'Execute and finish', three: 'Execute and finish',
}, },
error: { error: {
unavailable: 'This Knowledge is not avaliable', unavailable: 'This Knowledge is not available',
}, },
stepOne: { stepOne: {
filePreview: 'File Preview', filePreview: 'File Preview',
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment