anthropic[patch]: fix argument integration test (langchain-ai#18605)

thebhulawat · Mar 6, 2024 · 8b4ea65 · 8b4ea65
1 parent 22a1879
commit 8b4ea65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/partners/anthropic/tests/integration_tests/test_experimental.py b/libs/partners/anthropic/tests/integration_tests/test_experimental.py
@@ -1,5 +1,7 @@
 """Test ChatAnthropic chat model."""
 
+import json
+
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.pydantic_v1 import BaseModel
 
@@ -114,7 +116,7 @@ class Person(BaseModel):
     assert tool_call["type"] == "function"
     function = tool_call["function"]
     assert function["name"] == "Person"
-    assert function["arguments"] == {"name": "Erick", "age": "27"}
+    assert json.loads(function["arguments"]) == {"name": "Erick", "age": "27"}
 
 
 def test_with_structured_output() -> None: