chore: remove vcr marker to fix tests (#69)

langchain-ai · Jul 10, 2024 · a04b787 · a04b787
1 parent a5da828
commit a04b787
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 146 deletions.
diff --git a/libs/cohere/tests/integration_tests/cassettes/test_batch.yaml b/libs/cohere/tests/integration_tests/cassettes/test_batch.yaml
diff --git a/libs/cohere/tests/integration_tests/test_chat_models.py b/libs/cohere/tests/integration_tests/test_chat_models.py
@@ -85,10 +85,9 @@ async def test_abatch_tags() -> None:
         assert isinstance(token.content, str)
 
 
-@pytest.mark.vcr()
 def test_batch() -> None:
     """Test batch tokens from ChatCohere."""
-    llm = ChatCohere(model=DEFAULT_MODEL)
+    llm = ChatCohere()
 
     result = llm.batch(["I'm Pickle Rick", "I'm not Pickle Rick"])
     for token in result: