Fix TResponseInputItem

Author: RussellLuo

coagent/agents/react_agent/types.py

@@ -18,7 +18,29 @@
 TResponse = Response
 """A type alias for the Response type from the OpenAI SDK."""
 
-TResponseInputItem = ResponseInputItemParam
+
+class FunctionToolCallProgressItem(BaseModel):
+    call_id: str
+    """The unique ID of the function tool call generated by the model."""
+
+    progress: float = 0
+    """The current progress value of the function call. (e.g. 24)"""
+
+    total: float = 0
+    """The total value of the function call. (e.g. 100)"""
+
+    message: str = ""
+    """The human readable progress information. (e.g. Starting render...)"""
+
+    type: Literal["function_call_progress"]
+    """The type of the function tool call progress. Always `function_call_progress`."""
+
+
+TResponseInputItem: TypeAlias = Union[
+    ResponseInputItemParam,
+    ResponseFunctionToolCallOutputItem,
+    FunctionToolCallProgressItem,
+]
 """A type alias for the ResponseInputItemParam type from the OpenAI SDK."""
 
 TResponseOutputItem = ResponseOutputItem
@@ -50,27 +72,6 @@ class FunctionTool:
 """A type alias for the FunctionTool class."""
 
 
-class FunctionToolCallProgressItem(BaseModel):
-    call_id: str
-    """The unique ID of the function tool call generated by the model."""
-
-    progress: float = 0
-    """
-    The current progress value of the function call. (e.g. 24)
-            total: Optional total value e.g. 100
-            message: Optional message e.g. Starting render...A float between 0 and 1, where 0 means the function
-    """
-
-    total: float = 0
-    """The total value of the function call. (e.g. 100)"""
-
-    message: str = ""
-    """The human readable progress information. (e.g. Starting render...)"""
-
-    type: Literal["function_call_progress"]
-    """The type of the function tool call progress. Always `function_call_progress`."""
-
-
 @dataclass
 class MessageOutputItem:
     """Represents a message from the LLM."""