core[minor]: BaseChatModel with_structured_output implementation (#22859)

This commit is contained in:
Brace Sproul
2024-06-21 08:14:03 -07:00
committed by GitHub
parent 360a70c8a8
commit abe7566d7d
4 changed files with 142 additions and 29 deletions

View File

@@ -135,6 +135,7 @@ class TestOllamaFunctions(unittest.TestCase):
structured_llm = model.with_structured_output(Joke, include_raw=True)
res = structured_llm.invoke("Tell me a joke about cars")
assert isinstance(res, dict)
assert "raw" in res
assert "parsed" in res
assert isinstance(res["raw"], AIMessage)