diff --git a/libs/databricks/tests/integration_tests/test_chat_models.py b/libs/databricks/tests/integration_tests/test_chat_models.py index 18f0e7b..e57d643 100644 --- a/libs/databricks/tests/integration_tests/test_chat_models.py +++ b/libs/databricks/tests/integration_tests/test_chat_models.py @@ -36,6 +36,7 @@ _TEST_ENDPOINT = "databricks-meta-llama-3-1-70b-instruct" + def test_chat_databricks_invoke(): chat = ChatDatabricks( endpoint=_TEST_ENDPOINT, temperature=0, max_tokens=10, stop=["Java"]