fix: updated failing tests

instructor-ai · Nov 8, 2024 · 89d62f2 · 89d62f2
1 parent a353547
commit 89d62f2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/dsl/test_partial.py b/tests/dsl/test_partial.py
@@ -1,6 +1,6 @@
 # type: ignore[all]
 from pydantic import BaseModel, Field
-from instructor.dsl.partial import Partial
+from instructor.dsl.partial import Partial, LiteralPartialMixin
 import pytest
 import instructor
 from openai import OpenAI, AsyncOpenAI
@@ -116,7 +116,7 @@ async def async_generator():
 
 
 def test_summary_extraction():
-    class Summary(BaseModel):
+    class Summary(BaseModel, LiteralPartialMixin):
         summary: str = Field(description="A detailed summary")
 
     client = OpenAI()
@@ -143,7 +143,7 @@ class Summary(BaseModel):
 
 @pytest.mark.asyncio
 async def test_summary_extraction_async():
-    class Summary(BaseModel):
+    class Summary(BaseModel, LiteralPartialMixin):
         summary: str = Field(description="A detailed summary")
 
     client = AsyncOpenAI()