format, lint, and mypy all the files I touched.

Author: spokeydokeys

examples/agent_patterns/agents_as_tools_with_history.py

@@ -1,5 +1,6 @@
 import asyncio
 import tempfile
+from typing import cast
 
 from agents import Agent, Runner, SQLiteSession, trace
 from agents.tool import function_tool
@@ -31,7 +32,6 @@
 
 
 async def main(session_id, file_path):
-
     @function_tool
     async def check_tool() -> str:
         """A tool that looks at the entire conversation, checks
@@ -61,8 +61,7 @@ async def check_tool() -> str:
             check_agent,
             input=input_messages,
         )
-        return output.final_output
-
+        return cast(str, output.final_output)
 
     orchestrator_agent = Agent(
         name="orchestrator_agent",
@@ -85,8 +84,8 @@ async def check_tool() -> str:
                 tool_name="translate_to_italian",
                 tool_description="Translate the user's message to Italian",
             ),
-            check_tool
-        ]
+            check_tool,
+        ],
     )
 
     msg = input("Hi! What would you like translated? ")

src/agents/run.py

@@ -844,11 +844,15 @@ async def _start_streaming(
                         streamed_result.is_complete = True
 
                         # Save the conversation to session if enabled
-                        await AgentRunner._save_result_to_session(session, [], turn_result.new_step_items)
+                        await AgentRunner._save_result_to_session(
+                            session, [], turn_result.new_step_items
+                        )
 
                         streamed_result._event_queue.put_nowait(QueueCompleteSentinel())
                     elif isinstance(turn_result.next_step, NextStepRunAgain):
-                        await AgentRunner._save_result_to_session(session, [], turn_result.new_step_items)
+                        await AgentRunner._save_result_to_session(
+                            session, [], turn_result.new_step_items
+                        )
                 except AgentsException as exc:
                     streamed_result.is_complete = True
                     streamed_result._event_queue.put_nowait(QueueCompleteSentinel())