Merge pull request #160 from log10-io/kt/fix-langchain-test

Update langchain test assertion
log10-io · May 23, 2024 · ebf8334 · ebf8334
2 parents db36cbe + d2c5ffa
commit ebf8334
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_langchain.py b/tests/test_langchain.py
@@ -28,5 +28,6 @@ def test_chat_anthropic_messages(anthropic_legacy_model):
     messages = [SystemMessage(content="You are a ping pong machine"), HumanMessage(content="Ping?")]
     completion = llm.predict_messages(messages)
 
-    assert isinstance(completion.content, str)
-    assert "pong" in completion.content.lower()
+    content = completion.content
+    assert isinstance(content, str)
+    assert content, "No output from the model."