CI fix

ag2ai · Jan 15, 2025 · fe61a34 · fe61a34
1 parent 7bcf258
commit fe61a34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autogen/oai/client.py b/autogen/oai/client.py
@@ -13,7 +13,7 @@
 import warnings
 from typing import Any, Callable, Optional, Protocol, Union
 
-from pydantic import BaseModel, TypeAdapter
+from pydantic import BaseModel, schema_json_of
 
 from ..cache import Cache
 from ..exception_utils import ModelToolNotSupportedError
@@ -873,7 +873,7 @@ def yes_or_no_filter(context, response):
                 with cache_client as cache:
                     # Try to get the response from cache
                     key = get_key(
-                        {**params, **{"response_format": TypeAdapter.json_schema(params["response_format"])}}
+                        {**params, **{"response_format": schema_json_of(params["response_format"])}}
                         if "response_format" in params
                         else params
                     )