lint

Author: thomas-lab-openai

examples/basic/max_turns_resume.py

@@ -28,9 +28,9 @@ def main():
             input="Give me the main stages of the water cycle.",
             max_turns=1,
         )
-    except MaxTurnsExceeded as exc:
+    except MaxTurnsExceeded as max_turns_exc:
         print("Reached the max turn limit. Asking the agent to finalize without tools...\n")
-        result = exc.resume_sync(
+        result = max_turns_exc.resume_sync(
             "Finish the answer using the gathered information without calling tools again."
         )
         print(result.final_output)

src/agents/exceptions.py

@@ -2,15 +2,15 @@
 
 from dataclasses import dataclass, replace
 from textwrap import dedent
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     from .agent import Agent
     from .guardrail import InputGuardrailResult, OutputGuardrailResult
     from .items import ModelResponse, RunItem, TResponseInputItem
-    from .run_context import RunContextWrapper
-    from .run import RunConfig
     from .result import RunResult
+    from .run import RunConfig
+    from .run_context import RunContextWrapper
     from .tool_guardrails import (
         ToolGuardrailFunctionOutput,
         ToolInputGuardrail,
@@ -54,14 +54,15 @@ class MaxTurnsExceeded(AgentsException):
 
     _DEFAULT_RESUME_PROMPT = """
     You reached the maximum number of turns.
-    Return a final answer to the query using ONLY the information already gathered in the conversation so far.
+    Return a final answer to the query using ONLY the information already gathered \
+    in the conversation so far.
     """
 
     def __init__(self, message: str):
         self.message = message
         super().__init__(message)
 
-    async def resume(self, prompt: Optional[str] = _DEFAULT_RESUME_PROMPT) -> RunResult:
+    async def resume(self, prompt: str | None = _DEFAULT_RESUME_PROMPT) -> RunResult:
         """Resume the failed run asynchronously with a final, tool-free turn.
 
         Note:
@@ -87,7 +88,7 @@ async def resume(self, prompt: Optional[str] = _DEFAULT_RESUME_PROMPT) -> RunRes
             run_config=run_config,
         )
 
-    def resume_sync(self, prompt: Optional[str] = _DEFAULT_RESUME_PROMPT) -> RunResult:
+    def resume_sync(self, prompt: str | None = _DEFAULT_RESUME_PROMPT) -> RunResult:
         """Resume the failed run synchronously with a final, tool-free turn.
 
         Note:
@@ -116,7 +117,7 @@ def resume_sync(self, prompt: Optional[str] = _DEFAULT_RESUME_PROMPT) -> RunResu
     def _prepare_resume_arguments(
         self,
         run_data: RunErrorDetails,
-        prompt: Optional[str] = None,
+        prompt: str | None = None,
     ) -> tuple[list[TResponseInputItem], RunConfig]:
         from .items import ItemHelpers
         from .model_settings import ModelSettings
@@ -142,7 +143,7 @@ def _prepare_resume_arguments(
             run_config,
         )
 
-    def _normalize_resume_prompt(self, prompt: Optional[str]) -> Optional[str]:
+    def _normalize_resume_prompt(self, prompt: str | None) -> str | None:
         if prompt is None:
             return None
         normalized = dedent(prompt).strip()
@@ -151,7 +152,8 @@ def _normalize_resume_prompt(self, prompt: Optional[str]) -> Optional[str]:
     def _require_run_data(self) -> RunErrorDetails:
         if self.run_data is None:
             raise RuntimeError(
-                "Run data is not available; resume() can only be called on exceptions raised by Runner."
+                "Run data is not available; resume() can only be called on\
+                exceptions raised by Runner."
             )
         return self.run_data
 

src/agents/result.py

@@ -70,7 +70,7 @@ class RunResultBase(abc.ABC):
     context_wrapper: RunContextWrapper[Any]
     """The context wrapper for the agent run."""
 
-    run_config: "RunConfig"
+    run_config: RunConfig
     """The run configuration that was used for the agent run."""
 
     @property

tests/test_max_turns.py

@@ -130,15 +130,12 @@ def test_max_turns_resume_sync_uses_default_prompt():
     with pytest.raises(MaxTurnsExceeded) as exc_info:
         Runner.run_sync(agent, input="user_message", max_turns=2)
 
-    result = exc_info.value.resume_sync()
+    resume_prompt = "Return a final answer to the query using ONLY the information already gathered"
+    result = exc_info.value.resume_sync(resume_prompt)
 
     assert result.final_output == final_answer
     resume_input = model.last_turn_args["input"]
-    expected_prompt = (
-        "You reached the maximum number of turns.\n"
-        "Return a final answer to the query using ONLY the information already gathered in the conversation so far."
-    )
-    assert resume_input[-1] == {"content": expected_prompt, "role": "user"}
+    assert resume_input[-1] == {"content": resume_prompt, "role": "user"}
     assert model.last_turn_args["model_settings"].tool_choice == "none"