test

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

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

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()

api/core/tools/tool_engine.py

@@ -375,4 +375,4 @@ def _create_message_files(
 
         db.session.close()
 
-        return result
+        return result

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