From d9748fdb65fd82b6c51aedc08f80c24ba4236983 Mon Sep 17 00:00:00 2001
From: ccurme <chester.curme@gmail.com>
Date: Mon, 18 Mar 2024 10:56:02 -0400
Subject: [PATCH] fix integration tests (#20)

---
 backend/tests/integration_tests/test_extraction.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/backend/tests/integration_tests/test_extraction.py b/backend/tests/integration_tests/test_extraction.py
index fb3b713..86ddb59 100644
--- 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,
                 }