From 89d62f2dd3f5cdf35cc4b7853fbedc7ddaaacc50 Mon Sep 17 00:00:00 2001 From: Ivan Leo Date: Fri, 8 Nov 2024 13:03:49 +0800 Subject: [PATCH] fix: updated failing tests --- tests/dsl/test_partial.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/dsl/test_partial.py b/tests/dsl/test_partial.py index 733736564..b18bfc3e9 100644 --- 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()