From 51cc63d9cef288b2691dd6b54d1b9e1c3b03003d Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Thu, 26 Dec 2024 14:12:43 +0800 Subject: [PATCH] fix: undefined dereference to ApiTool --- api/core/agent/base_agent_runner.py | 6 +++++- api/core/tools/custom_tool/provider.py | 2 -- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/api/core/agent/base_agent_runner.py b/api/core/agent/base_agent_runner.py index 2e270e304d..548631c19e 100644 --- a/api/core/agent/base_agent_runner.py +++ b/api/core/agent/base_agent_runner.py @@ -484,7 +484,11 @@ class BaseAgentRunner(AppRunner): files = db.session.query(MessageFile).filter(MessageFile.message_id == message.id).all() if not files: return UserPromptMessage(content=message.query) - file_extra_config = FileUploadConfigManager.convert(message.app_model_config.to_dict()) + if message.app_model_config: + file_extra_config = FileUploadConfigManager.convert(message.app_model_config.to_dict()) + else: + file_extra_config = None + if not file_extra_config: return UserPromptMessage(content=message.query) diff --git a/api/core/tools/custom_tool/provider.py b/api/core/tools/custom_tool/provider.py index 7efd7e00b8..fd7873b083 100644 --- a/api/core/tools/custom_tool/provider.py +++ b/api/core/tools/custom_tool/provider.py @@ -151,8 +151,6 @@ class ApiToolProviderController(ToolProviderController): """ if len(self.tools) > 0: return self.tools - if self.identity is None: - return None tools: list[ApiTool] = []