mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-04 10:42:55 +00:00
add cassette for parsed
This commit is contained in:
parent
3768bb1d58
commit
36dbb20da7
Binary file not shown.
@ -181,8 +181,13 @@ class FooDict(TypedDict):
|
|||||||
response: str
|
response: str
|
||||||
|
|
||||||
|
|
||||||
def test_parsed_pydantic_schema() -> None:
|
@pytest.mark.default_cassette("test_parsed_pydantic_schema.yaml.gz")
|
||||||
llm = ChatOpenAI(model=MODEL_NAME, use_responses_api=True)
|
@pytest.mark.vcr
|
||||||
|
@pytest.mark.parametrize("output_version", ["v0", "responses/v1", "v1"])
|
||||||
|
def test_parsed_pydantic_schema(output_version: Literal["v0", "responses/v1", "v1"]) -> None:
|
||||||
|
llm = ChatOpenAI(
|
||||||
|
model=MODEL_NAME, use_responses_api=True, output_version=output_version
|
||||||
|
)
|
||||||
response = llm.invoke("how are ya", response_format=Foo)
|
response = llm.invoke("how are ya", response_format=Foo)
|
||||||
parsed = Foo(**json.loads(response.text()))
|
parsed = Foo(**json.loads(response.text()))
|
||||||
assert parsed == response.additional_kwargs["parsed"]
|
assert parsed == response.additional_kwargs["parsed"]
|
||||||
|
Loading…
Reference in New Issue
Block a user