Merge branch 'nqn/session-bug' of github.com:log10-io/log10 into nqn/…

…session-bug
log10-io · May 23, 2024 · c02f954 · c02f954
2 parents c8336d6 + 6d0de17
commit c02f954
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."