chore(tests): fix pydantic v1 tests

openai · Aug 12, 2024 · f1e1060 · f1e1060
1 parent 22bbd4a
commit f1e1060
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/lib/chat/test_completions.py b/tests/lib/chat/test_completions.py
@@ -15,6 +15,7 @@
 import openai
 from openai import OpenAI, AsyncOpenAI
 from openai._utils import assert_signatures_in_sync
+from openai._compat import PYDANTIC_V2
 
 from ._utils import print_obj
 from ...conftest import base_url
@@ -147,6 +148,9 @@ class ColorDetection(BaseModel):
         color: Color
         hex_color_code: str = Field(description="The hex color code of the detected color")
 
+    if not PYDANTIC_V2:
+        ColorDetection.update_forward_refs(**locals())  # type: ignore
+
     completion = _make_snapshot_request(
         lambda c: c.beta.chat.completions.parse(
             model="gpt-4o-2024-08-06",