Put rebuild back

Author: tconley1428

temporalio/contrib/openai_agents/__init__.py

@@ -7,6 +7,13 @@
     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 (
@@ -30,3 +37,22 @@
     "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,6 @@
 
 import enum
 import json
-from pydantic.dataclasses import dataclass
 from typing import Any, Optional, Union, cast
 
 from agents import (
@@ -24,6 +23,7 @@
     WebSearchTool,
 )
 from agents.models.multi_provider import MultiProvider
+from pydantic.dataclasses import dataclass
 from typing_extensions import Required, TypedDict
 
 from temporalio import activity

tests/contrib/openai_agents/test_openai.py

@@ -1754,4 +1754,4 @@ async def test_response_serialization():
         usage=Usage(),
         response_id="",
     )
-    encoded = await pydantic_data_converter.encode([model_response])
+    encoded = await pydantic_data_converter.encode([model_response])