make lint

Author: stevenheidel

src/agents/model_settings.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from dataclasses import dataclass, asdict
+from dataclasses import dataclass
 from typing import Literal
 
 
@@ -31,7 +31,7 @@ class ModelSettings:
     """The tool choice to use when calling the model."""
 
     parallel_tool_calls: bool | None = None
-    """Whether to use parallel tool calls when calling the model. 
+    """Whether to use parallel tool calls when calling the model.
     Defaults to False if not provided."""
 
     truncation: Literal["auto", "disabled"] | None = None
@@ -41,7 +41,7 @@ class ModelSettings:
     """The maximum number of output tokens to generate."""
 
     store: bool | None = None
-    """Whether to store the generated model response for later retrieval. 
+    """Whether to store the generated model response for later retrieval.
     Defaults to True if not provided."""
 
     def resolve(self, override: ModelSettings | None) -> ModelSettings:

tests/test_agent_runner.py

@@ -14,12 +14,14 @@
     InputGuardrail,
     InputGuardrailTripwireTriggered,
     ModelBehaviorError,
+    ModelSettings,
     OutputGuardrail,
     OutputGuardrailTripwireTriggered,
+    RunConfig,
     RunContextWrapper,
     Runner,
     UserError,
-    handoff, ModelSettings, RunConfig,
+    handoff,
 )
 from agents.agent import ToolsToFinalOutputResult
 from agents.tool import FunctionToolResult, function_tool