anthropic[patch]: integration test update (#18823)

langchain-ai · Mar 8, 2024 · a8de6d1 · a8de6d1
1 parent d1f5bc4
commit a8de6d1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/libs/partners/anthropic/tests/integration_tests/test_experimental.py b/libs/partners/anthropic/tests/integration_tests/test_experimental.py
@@ -10,6 +10,7 @@
 from langchain_anthropic.experimental import ChatAnthropicTools
 
 MODEL_NAME = "claude-3-sonnet-20240229"
+BIG_MODEL_NAME = "claude-3-opus-20240229"
 
 #####################################
 ### Test Basic features, no tools ###
@@ -108,7 +109,9 @@ class Person(BaseModel):
         name: str
         age: int
 
-    llm = ChatAnthropicTools(model_name=MODEL_NAME).bind_tools([Person])
+    llm = ChatAnthropicTools(model_name=BIG_MODEL_NAME, temperature=0).bind_tools(
+        [Person]
+    )
     result = llm.invoke("Erick is 27 years old")
     assert result.content == "", f"content should be empty, not {result.content}"
     assert "tool_calls" in result.additional_kwargs
@@ -126,7 +129,9 @@ class Person(BaseModel):
         name: str
         age: int
 
-    chain = ChatAnthropicTools(model_name=MODEL_NAME).with_structured_output(Person)
+    chain = ChatAnthropicTools(
+        model_name=BIG_MODEL_NAME, temperature=0
+    ).with_structured_output(Person)
     result = chain.invoke("Erick is 27 years old")
     assert isinstance(result, Person)
     assert result.name == "Erick"
@@ -167,7 +172,7 @@ class Email(BaseModel):
         ]
     )
 
-    llm = ChatAnthropicTools(temperature=0, model_name="claude-3-sonnet-20240229")
+    llm = ChatAnthropicTools(temperature=0, model_name=BIG_MODEL_NAME)
 
     extraction_chain = prompt | llm.with_structured_output(Email)