fix mypy error

pull/19020/head
金鹏程 1 year ago
parent 1aa6a2126f
commit d02c48c058

@ -69,8 +69,11 @@ class CotAgentRunner(BaseAgentRunner, ABC):
tool_instances, prompt_messages_tools = self._init_prompt_tools() tool_instances, prompt_messages_tools = self._init_prompt_tools()
self._prompt_messages_tools = prompt_messages_tools self._prompt_messages_tools = prompt_messages_tools
# fix metadata filter not work
if app_config.dataset is not None:
metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions
for key, dataset_retriever_tool in tool_instances.items(): for key, dataset_retriever_tool in tool_instances.items():
if hasattr(dataset_retriever_tool, 'retrieval_tool'):
dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions
function_call_state = True function_call_state = True

@ -45,8 +45,11 @@ class FunctionCallAgentRunner(BaseAgentRunner):
# convert tools into ModelRuntime Tool format # convert tools into ModelRuntime Tool format
tool_instances, prompt_messages_tools = self._init_prompt_tools() tool_instances, prompt_messages_tools = self._init_prompt_tools()
# fix metadata filter not work
if app_config.dataset is not None:
metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions
for key, dataset_retriever_tool in tool_instances.items(): for key, dataset_retriever_tool in tool_instances.items():
if hasattr(dataset_retriever_tool, 'retrieval_tool'):
dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions
assert app_config.agent assert app_config.agent

Loading…
Cancel
Save