fix lint and mypy errors

Author: seratch

src/agents/models/chatcmpl_converter.py

@@ -51,6 +51,8 @@
 from ..tool import FunctionTool, Tool
 from .fake_id import FAKE_RESPONSES_ID
 
+ResponseInputContentWithAudioParam = Union[ResponseInputContentParam, ResponseInputAudioParam]
+
 
 class Converter:
     @classmethod
@@ -248,7 +250,7 @@ def maybe_reasoning_message(cls, item: Any) -> ResponseReasoningItemParam | None
 
     @classmethod
     def extract_text_content(
-        cls, content: str | Iterable[ResponseInputContentParam]
+        cls, content: str | Iterable[ResponseInputContentWithAudioParam]
     ) -> str | list[ChatCompletionContentPartTextParam]:
         all_content = cls.extract_all_content(content)
         if isinstance(all_content, str):
@@ -261,7 +263,7 @@ def extract_text_content(
 
     @classmethod
     def extract_all_content(
-        cls, content: str | Iterable[ResponseInputContentParam]
+        cls, content: str | Iterable[ResponseInputContentWithAudioParam]
     ) -> str | list[ChatCompletionContentPartParam]:
         if isinstance(content, str):
             return content
@@ -537,7 +539,7 @@ def ensure_assistant_message() -> ChatCompletionAssistantMessageParam:
             elif func_output := cls.maybe_function_tool_call_output(item):
                 flush_assistant_message()
                 output_content = cast(
-                    Union[str, Iterable[ResponseInputContentParam]], func_output["output"]
+                    Union[str, Iterable[ResponseInputContentWithAudioParam]], func_output["output"]
                 )
                 msg: ChatCompletionToolMessageParam = {
                     "role": "tool",

tests/extensions/memory/test_sqlalchemy_session.py

@@ -36,6 +36,7 @@ def _make_message_item(item_id: str, text_value: str) -> TResponseInputItem:
         "type": "output_text",
         "text": text_value,
         "annotations": [],
+        "logprobs": [],
     }
     message: ResponseOutputMessageParam = {
         "id": item_id,

tests/test_items_helpers.py

@@ -321,13 +321,14 @@ def test_input_to_new_input_list_copies_the_ones_produced_by_pydantic() -> None:
     # Given a list of message dictionaries, ensure the returned list is a deep copy.
     original = ResponseOutputMessageParam(
         id="a75654dc-7492-4d1c-bce0-89e8312fbdd7",
-        content=[
-            ResponseOutputTextParam(
-                type="output_text",
-                text="Hey, what's up?",
-                annotations=[],
-            )
-        ],
+            content=[
+                ResponseOutputTextParam(
+                    type="output_text",
+                    text="Hey, what's up?",
+                    annotations=[],
+                    logprobs=[],
+                )
+            ],
         role="assistant",
         status="completed",
         type="message",

tests/utils/test_json.py

@@ -16,6 +16,7 @@ def test_to_dump_compatible():
                     type="output_text",
                     text="Hey, what's up?",
                     annotations=[],
+                    logprobs=[],
                 )
             ].__iter__(),
             role="assistant",
@@ -28,5 +29,5 @@ def test_to_dump_compatible():
     result = json.dumps(_to_dump_compatible(input_iter))
     assert (
         result
-        == """[{"id": "a75654dc-7492-4d1c-bce0-89e8312fbdd7", "content": [{"type": "output_text", "text": "Hey, what's up?", "annotations": []}], "role": "assistant", "status": "completed", "type": "message"}]"""  # noqa: E501
+        == """[{"id": "a75654dc-7492-4d1c-bce0-89e8312fbdd7", "content": [{"type": "output_text", "text": "Hey, what's up?", "annotations": [], "logprobs": []}], "role": "assistant", "status": "completed", "type": "message"}]"""  # noqa: E501
     )