fix integration tests (#20)

langchain-ai · Mar 18, 2024 · d9748fd · d9748fd
1 parent 22a9eee
commit d9748fd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/backend/tests/integration_tests/test_extraction.py b/backend/tests/integration_tests/test_extraction.py
@@ -50,7 +50,6 @@ class Person(BaseModel):
         )
         assert result.status_code == 200, result.text
         response_data = result.json()
-        assert response_data == {}
         assert isinstance(response_data["output"]["data"], list)
 
         # Test with instructions
@@ -79,7 +78,7 @@ class Person(BaseModel):
             json={
                 "input": {
                     "text": text,
-                    "schema": Person(),
+                    "schema": Person.schema(),
                     "instructions": "Redact all names using the characters `######`",
                     "examples": examples,
                 }