reformat all files.

Author: gali-leilei

apps/eval-agent/src/eval_agent/__init__.py

@@ -2,5 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+
 def hello() -> str:
     return "Hello from eval-agent!"

apps/run-agent/common_benchmark.py

@@ -530,7 +530,7 @@ def load_tasks(self) -> List[BenchmarkTask]:
                 except json.JSONDecodeError as e:
                     print(f"Warning: Failed to parse line {i + 1}: {e}")
                     continue
-        tasks = tasks[:self.cfg.benchmark.execution.max_tasks]
+        tasks = tasks[: self.cfg.benchmark.execution.max_tasks]
         self.tasks = tasks
         print(f"Loaded {len(tasks)} tasks")
         return tasks

apps/run-agent/trace_single_task.py

@@ -53,7 +53,12 @@ async def single_task(
     )
 
 
-def main(*args, task_id: str = "task_1", task: str="Write a python code to say 'Hello, World!', use python to execute the code.",  task_file_name: str = ""):
+def main(
+    *args,
+    task_id: str = "task_1",
+    task: str = "Write a python code to say 'Hello, World!', use python to execute the code.",
+    task_file_name: str = "",
+):
     print("inside trace")
     dotenv.load_dotenv()
     with hydra.initialize_config_dir(config_dir=config_path(), version_base=None):

libs/miroflow-contrib/src/miroflow/contrib/pocket/core.py

@@ -99,9 +99,7 @@ def _exec(self, prep_res: _PrepResult) -> _ExecResult:
         raise RuntimeError("Should not reach here")
 
 
-class BatchNode[_PrepResult, _ExecResult](
-    Node[list[_PrepResult] , list[_ExecResult]]
-):
+class BatchNode[_PrepResult, _ExecResult](Node[list[_PrepResult], list[_ExecResult]]):
     def _exec(self, prep_res: list[_PrepResult]) -> list[_ExecResult]:
         return [super(BatchNode, self)._exec(i) for i in prep_res]
 
@@ -139,9 +137,7 @@ def _run(self, shared: SharedData) -> str:
         o = self._orch(shared)
         return self.post(shared, p, o)
 
-    def post(
-        self, shared: SharedData, prep_res: _PrepResult, exec_res: Any
-    ) -> str:
+    def post(self, shared: SharedData, prep_res: _PrepResult, exec_res: Any) -> str:
         return exec_res
 
 
@@ -195,9 +191,7 @@ def _run(self, shared: SharedData) -> str:
         raise RuntimeError("Use run_async.")
 
 
-class AsyncBatchNode(
-    AsyncNode[list[_PrepResult] | None, list[_ExecResult], str]
-):
+class AsyncBatchNode(AsyncNode[list[_PrepResult] | None, list[_ExecResult], str]):
     async def _exec(self, prep_res: list[_PrepResult] | None) -> list[_ExecResult]:
         if prep_res is None:
             return []
@@ -215,9 +209,7 @@ async def _exec(self, prep_res: list[_PrepResult] | None) -> list[_ExecResult]:
         )
 
 
-class AsyncFlow(
-    Flow[_PrepResult, Any], AsyncNode[_PrepResult, Any]
-):
+class AsyncFlow(Flow[_PrepResult, Any], AsyncNode[_PrepResult, Any]):
     async def _orch_async(
         self, shared: SharedData, params: Params | None = None
     ) -> Any:

libs/miroflow-contrib/src/miroflow/contrib/pocket/core_v2.py

@@ -7,8 +7,8 @@
 A, B = TypeVar("A"), TypeVar("B")
 Context = Any
 
-class Node[A, B](ABC):
 
+class Node[A, B](ABC):
     @abstractmethod
     def prep(self, ctx: Context) -> A: ...
 
@@ -23,30 +23,31 @@ def run(self, ctx: Context) -> str:
         e = self.exec(p)
         return self.post(ctx, p, e)
 
+
 class Flowable(Protocol):
     @property
     def successors(self) -> dict[str, "Node[Any, Any]"]: ...
 
+
 class FlowableMixin:
     def handle(self: Flowable, action: str) -> "Node[Any, Any] | None":
         return self.successors.get(action, None)
-    
-    def add_successor(self: Flowable, action: str, node: "Node[Any, Any]") -> "Node[Any, Any]":
+
+    def add_successor(
+        self: Flowable, action: str, node: "Node[Any, Any]"
+    ) -> "Node[Any, Any]":
         if action in self.successors:
             warnings.warn(f"Overwriting successor for action '{action}'")
         self.successors[action] = node
         return node
-    
+
+
 @dataclass
 class Flow:
     start: Node[Any, Any]
-    
-    
-    def flow(self, ctx: Context) -> str: 
-        curr,last_action = (
-            copy.copy(self.start),
-            "default"
-        )
+
+    def flow(self, ctx: Context) -> str:
+        curr, last_action = (copy.copy(self.start), "default")
         while curr is not None:
             last_action = curr.run(ctx)
             next_node = curr.successors.get(last_action, None)

libs/miroflow-contrib/tests/test_demo.py

@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-License-Identifier: MIT
 
+
 def test_hello():
     assert 1 + 2 == 3, "just a demo"
 

libs/miroflow-tool/tests/test_reading_mcp_server.py

@@ -17,6 +17,7 @@
 files_for_tests_dir = os.path.join(current_dir, "files-for-tests")
 pytest.skip("Skipping all tests in this file", allow_module_level=True)
 
+
 class TestReadingMCPServer:
     """Test suite for Reading MCP Server functionality."""
 

libs/miroflow-tool/tests/test_searching_mcp_server.py

@@ -7,8 +7,10 @@
 from mcp import StdioServerParameters
 
 from miroflow.tool.manager import ToolManager
- 
+
 pytest.skip("Skipping all tests in this file", allow_module_level=True)
+
+
 class TestSearchingMCPServer:
     """Test suite for Searching MCP Server functionality."""
 
@@ -654,4 +656,3 @@ async def test_serper_api_key_not_set(self):
 def pytest_configure(config):
     """Configure pytest with custom markers."""
     config.addinivalue_line("markers", "integration: marks tests as integration tests")
-    

libs/miroflow/src/miroflow/prebuilt/orchestrator.py

@@ -694,7 +694,7 @@ async def _get_gaia_answer_type(self, task_description: str) -> str:
 Return exactly one of the [number, date, time, string], nothing else.
 """
         print(f"Answer type instruction: {instruction}")
-        
+
         message_id = _generate_message_id()
         response = await client.chat.completions.create(
             model="gpt-4.1",
@@ -704,18 +704,20 @@ async def _get_gaia_answer_type(self, task_description: str) -> str:
         # 检查结果是否为空，如果为空则抛出异常触发重试
         if not answer_type or not answer_type.strip():
             raise ValueError("answer type returned empty result")
-        
+
         print(f"Answer type: {answer_type}")
-        
+
         return answer_type.strip()
 
     @retry(wait=wait_exponential(multiplier=15), stop=stop_after_attempt(5))
-    async def _o3_extract_gaia_final_answer(self, answer_type: str, task_description_detail: str, summary: str) -> str:
+    async def _o3_extract_gaia_final_answer(
+        self, answer_type: str, task_description_detail: str, summary: str
+    ) -> str:
         """使用O3模型从summary中抽取最终答案"""
         client = AsyncOpenAI(api_key=self.cfg.env.openai_api_key, timeout=600)
-        
+
         full_prompts = {
-"time": f"""# Inputs
+            "time": f"""# Inputs
 
 * **Original Question**: `{task_description_detail}`
 * **Agent Summary**: `{summary}`
@@ -774,8 +776,7 @@ async def _o3_extract_gaia_final_answer(self, answer_type: str, task_description
 
 Return the step-by-step process and your final answer wrapped in \\boxed{{...}}, check the **Formatting rules**, **Additional constraints**, **Common pitfalls to avoid** and **Quick reference examples** step by step, and ensure the answer meet the requirements.
 """,
-
-"number": f"""# Inputs
+            "number": f"""# Inputs
 
 * **Original Question**: `{task_description_detail}`
 * **Agent Summary**: `{summary}`
@@ -854,8 +855,7 @@ async def _o3_extract_gaia_final_answer(self, answer_type: str, task_description
 
 Return the step-by-step process and your final answer wrapped in \\boxed{{...}}, check the **Formatting rules**, **Additional constraints**, **Common pitfalls to avoid** and **Quick reference examples** step by step, and ensure the answer meet the requirements.
 """,
-
-"string": f"""# Inputs
+            "string": f"""# Inputs
 
 * **Original Question**: `{task_description_detail}`
 * **Agent Summary**: `{summary}`
@@ -940,10 +940,12 @@ async def _o3_extract_gaia_final_answer(self, answer_type: str, task_description
 --- 
 # Output
 Return the step-by-step process and your final answer wrapped in \\boxed{{...}}, check the **Formatting rules**, **Additional constraints**, **Common pitfalls to avoid** and **Quick reference examples** step by step, and ensure the answer meet the requirements.
-"""
-}
-        full_prompt = full_prompts.get(answer_type if answer_type in ["number", "time"] else "string")
-        
+""",
+        }
+        full_prompt = full_prompts.get(
+            answer_type if answer_type in ["number", "time"] else "string"
+        )
+
         print("O3 Extract Final Answer Prompt:")
         print(full_prompt)
 
@@ -959,12 +961,12 @@ async def _o3_extract_gaia_final_answer(self, answer_type: str, task_description
         if not result or not result.strip():
             raise ValueError("O3 final answer extraction returned empty result")
 
-        match = re.search(r'\\boxed{([^}]*)}', result)
+        match = re.search(r"\\boxed{([^}]*)}", result)
         if not match:
             raise ValueError("O3 final answer extraction returned empty answer")
 
         print("response:", result)
-                
+
         return result
 
     async def run_main_agent(
@@ -1285,7 +1287,7 @@ async def run_main_agent(
                 # 执行O3最终答案提取
                 try:
                     answer_type = await self._get_gaia_answer_type(task_description)
-                
+
                     o3_extracted_answer = await self._o3_extract_gaia_final_answer(
                         answer_type,
                         task_description,

libs/miroflow/src/miroflow/utils/tool_utils.py

@@ -29,7 +29,6 @@ def create_mcp_server_parameters(
         if logs_dir:
             tool_code_env["LOGS_DIR"] = str(logs_dir)
 
-
         configs.append(
             {
                 "name": "tool-code",
@@ -133,7 +132,6 @@ def create_mcp_server_parameters(
         agent_cfg.get("tools", None) is not None
         and "tool-browsing" in agent_cfg["tools"]
     ):
-
         configs.append(
             {
                 "name": "tool-browsing",
@@ -265,4 +263,3 @@ def expose_sub_agents_as_tools(sub_agents_cfg: DictConfig):
                 )
             )
     return sub_agents_server_params
-