Add type_to_response_format_param()

Author: RussellLuo

coagent/agents/util.py

@@ -1,13 +1,22 @@
 import asyncio
 import functools
-from typing import AsyncIterator
+from typing import AsyncIterator, Type
 
 from coagent.core import logger
+from pydantic import BaseModel
 
 from .messages import ChatMessage
 from .model_client import default_model_client, ModelClient
 
 
+def type_to_response_format_param(
+    response_format: Type[BaseModel] | dict | None,
+) -> dict | None:
+    import litellm.utils
+
+    return litellm.utils.type_to_response_format_param(response_format)
+
+
 async def chat(
     messages: list[ChatMessage], client: ModelClient = default_model_client
 ) -> ChatMessage: