Merge branch 'main' into nqn/session-bug

log10-io · May 23, 2024 · 6d0de17 · 6d0de17
2 parents 26f7880 + ebf8334
commit 6d0de17
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."