kennethreitz/instructor
1
0
Fork 0
mirror of https://github.com/kennethreitz/instructor.git synced 2026-06-05 22:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add md_mode

Browse Source
This commit is contained in:
Jason Liu
2023-12-12 22:48:34 -05:00
parent 11664ec09c
commit 329aad023b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/openai/test_patch.py
+4 -4
View File
@@ -15,7 +15,7 @@ class UserExtract(BaseModel):
age: int
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.TOOLS])
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.TOOLS, Mode.MD_JSON])
def test_runmodel(mode):
client = instructor.patch(OpenAI(), mode=mode)
model = client.chat.completions.create(
@@ -34,7 +34,7 @@ def test_runmodel(mode):
), "The raw response should be available from OpenAI"
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.TOOLS])
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.TOOLS, Mode.MD_JSON])
@pytest.mark.asyncio
async def test_runmodel_async(mode):
aclient = instructor.patch(AsyncOpenAI(), mode=mode)
@@ -66,7 +66,7 @@ class UserExtractValidated(BaseModel):
return v
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON])
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.MD_JSON])
def test_runmodel_validator(mode):
client = instructor.patch(OpenAI(), mode=mode)
model = client.chat.completions.create(
@@ -84,7 +84,7 @@ def test_runmodel_validator(mode):
), "The raw response should be available from OpenAI"
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON])
@pytest.mark.parametrize("mode", [Mode.FUNCTIONS, Mode.JSON, Mode.MD_JSON])
@pytest.mark.asyncio
async def test_runmodel_async_validator(mode):
aclient = instructor.patch(AsyncOpenAI(), mode=mode)