Skip to content

Commit

Permalink
Fix test_chat_databricks_invoke with llama 3.3
Browse files Browse the repository at this point in the history
Signed-off-by: Prithvi Kannan <[email protected]>
  • Loading branch information
prithvikannan committed Dec 12, 2024
1 parent efbfc54 commit 7cb69dd
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,9 @@ def test_chat_databricks_invoke():
response = chat.invoke("How to learn Java? Start the response by 'To learn Java,'")
assert isinstance(response, AIMessage)
assert response.content == "To learn "
assert response.response_metadata["prompt_tokens"] == 24
assert response.response_metadata["prompt_tokens"] == 25
assert response.response_metadata["completion_tokens"] == 3
assert response.response_metadata["total_tokens"] == 27
assert response.response_metadata["total_tokens"] == 28

response = chat.invoke("How to learn Python? Start the response by 'To learn Python,'")
assert response.content.startswith("To learn Python,")
Expand Down

0 comments on commit 7cb69dd

Please sign in to comment.