Add test case

Kiln-AI · Sep 9, 2024 · fbf155f · fbf155f
1 parent f82a84d
commit fbf155f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libs/core/kiln_ai/adapters/test_langchain_adapter.py b/libs/core/kiln_ai/adapters/test_langchain_adapter.py
@@ -18,3 +18,8 @@ def test_langchain_adapter_munge_response(tmp_path):
     munged = lca._munge_response(response)
     assert munged["setup"] == "Why did the cow join a band?"
     assert munged["punchline"] == "Because she wanted to be a moo-sician!"
+
+    # non mistral format should continue to work
+    munged = lca._munge_response(response["arguments"])
+    assert munged["setup"] == "Why did the cow join a band?"
+    assert munged["punchline"] == "Because she wanted to be a moo-sician!"