anthropic: timeout in tests (10s) (#28488)

langchain-ai · Dec 5, 2024 · c5acedd · c5acedd
1 parent f459754
commit c5acedd
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 3 deletions.
diff --git a/libs/partners/anthropic/Makefile b/libs/partners/anthropic/Makefile
@@ -8,7 +8,7 @@ TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
 test tests integration_test integration_tests:
-	poetry run pytest $(TEST_FILE)
+	poetry run pytest -vvv --timeout 10 $(TEST_FILE)
 
 test_watch:
 	poetry run ptw --snapshot-update --now . -- -vv $(TEST_FILE)

diff --git a/libs/partners/anthropic/poetry.lock b/libs/partners/anthropic/poetry.lock
diff --git a/libs/partners/anthropic/pyproject.toml b/libs/partners/anthropic/pyproject.toml
@@ -65,6 +65,7 @@ syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
 defusedxml = "^0.7.1"
+pytest-timeout = "^2.3.1"
 
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"

diff --git a/libs/partners/anthropic/tests/integration_tests/test_chat_models.py b/libs/partners/anthropic/tests/integration_tests/test_chat_models.py
@@ -22,7 +22,7 @@
 from langchain_anthropic import ChatAnthropic, ChatAnthropicMessages
 from tests.unit_tests._utils import FakeCallbackHandler
 
-MODEL_NAME = "claude-3-sonnet-20240229"
+MODEL_NAME = "claude-3-5-sonnet-20240620"
 
 
 def test_stream() -> None: