diff --git a/api/core/agent/cot_agent_runner.py b/api/core/agent/cot_agent_runner.py index 6083e0f72ebcfc..2f0643220c2f06 100644 --- a/api/core/agent/cot_agent_runner.py +++ b/api/core/agent/cot_agent_runner.py @@ -292,7 +292,9 @@ def _save_and_publish_final_thought(self, thought_id: str, final_answer: str, th answer=final_answer, messages_ids=[], ) - self.queue_manager.publish(QueueAgentThoughtEvent(agent_thought_id=thought_id), PublishFrom.APPLICATION_MANAGER) + self.queue_manager.publish( + QueueAgentThoughtEvent(agent_thought_id=thought_id), PublishFrom.APPLICATION_MANAGER + ) def _handle_invoke_action( self, @@ -450,4 +452,4 @@ def _organize_historic_prompt_messages( history_messages=result, memory=self.memory, ).get_prompt() - return historic_prompts + return historic_prompts \ No newline at end of file diff --git a/api/core/agent/fc_agent_runner.py b/api/core/agent/fc_agent_runner.py index e844048e84d7c1..62ecd5b185a094 100644 --- a/api/core/agent/fc_agent_runner.py +++ b/api/core/agent/fc_agent_runner.py @@ -674,4 +674,4 @@ def _organize_prompt_messages(self): if len(self._current_thoughts) != 0: # clear messages after the first iteration prompt_messages = self._clear_user_prompt_image_messages(prompt_messages) - return prompt_messages + return prompt_messages \ No newline at end of file diff --git a/api/core/tools/entities/tool_entities.py b/api/core/tools/entities/tool_entities.py index 23e6c0c5e11140..e25cc7357fd09c 100644 --- a/api/core/tools/entities/tool_entities.py +++ b/api/core/tools/entities/tool_entities.py @@ -491,4 +491,4 @@ class Parameter(BaseModel): tool_parameters: Mapping[str, Parameter] = Field(..., description="Parameters, type llm") def to_plugin_parameter(self) -> dict[str, Any]: - return self.model_dump() + return self.model_dump() \ No newline at end of file diff --git a/api/core/tools/tool_engine.py b/api/core/tools/tool_engine.py index 018c326ee0382e..fcae348c0599ff 100644 --- a/api/core/tools/tool_engine.py +++ b/api/core/tools/tool_engine.py @@ -375,4 +375,4 @@ def _create_message_files( db.session.close() - return result + return result \ No newline at end of file diff --git a/api/core/tools/workflow_as_tool/tool.py b/api/core/tools/workflow_as_tool/tool.py index 865a96b0477a91..b6bc3adf2b5d36 100644 --- a/api/core/tools/workflow_as_tool/tool.py +++ b/api/core/tools/workflow_as_tool/tool.py @@ -381,4 +381,4 @@ def _update_file_mapping(self, file_dict: dict): file_dict["tool_file_id"] = file_dict.get("related_id") elif transfer_method == FileTransferMethod.LOCAL_FILE: file_dict["upload_file_id"] = file_dict.get("related_id") - return file_dict + return file_dict \ No newline at end of file