Strip down to minimum test

Author: tconley1428

temporalio/contrib/openai_agents/__init__.py

@@ -7,13 +7,6 @@
     This module is experimental and may change in future versions.
     Use with caution in production environments.
 """
-
-import importlib
-import inspect
-import pkgutil
-
-from pydantic import BaseModel
-
 from temporalio.contrib.openai_agents._invoke_model_activity import ModelActivity
 from temporalio.contrib.openai_agents._model_parameters import ModelActivityParameters
 from temporalio.contrib.openai_agents._trace_interceptor import (
@@ -37,22 +30,3 @@
     "TestModelProvider",
 ]
 
-
-def _reload_models(module_name: str) -> None:
-    """Recursively walk through modules and rebuild BaseModel classes."""
-    module = importlib.import_module(module_name)
-
-    # Process classes in the current module
-    for _, obj in inspect.getmembers(module, inspect.isclass):
-        if issubclass(obj, BaseModel) and obj is not BaseModel:
-            obj.model_rebuild()
-
-    # Recursively process submodules
-    if hasattr(module, "__path__"):
-        for _, submodule_name, _ in pkgutil.iter_modules(module.__path__):
-            full_submodule_name = f"{module_name}.{submodule_name}"
-            _reload_models(full_submodule_name)
-
-
-# Recursively call model_rebuild() on all BaseModel classes in openai.types
-_reload_models("openai.types")

temporalio/contrib/openai_agents/_invoke_model_activity.py

@@ -5,7 +5,7 @@
 
 import enum
 import json
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from typing import Any, Optional, Union, cast
 
 from agents import (

temporalio/contrib/pydantic.py

@@ -15,6 +15,7 @@
 
 from typing import Any, Optional, Type
 
+import pydantic.dataclasses
 from pydantic import TypeAdapter
 from pydantic_core import to_json
 

tests/contrib/openai_agents/test_openai.py

@@ -1747,4 +1747,11 @@ async def test_response_serialization():
         b'{"id": "msg_68757ec43348819d86709f0fcb70316301a1194a3e05b38c","type": "image_generation_call","status": "completed"}'
     )
     call = TypeAdapter(ImageGenerationCall).validate_python(data)
-    encoded = await pydantic_data_converter.encode([call])
+    model_response = ModelResponse(
+        output=[
+            call,
+        ],
+        usage=Usage(),
+        response_id="",
+    )
+    encoded = await pydantic_data_converter.encode([model_response])