tests: wip

aurelio-labs · Sep 26, 2024 · c085fa8 · c085fa8
1 parent ea7b7a6
commit c085fa8
Show file tree

Hide file tree

Showing 4 changed files with 292 additions and 182 deletions.
diff --git a/aurelio_sdk/schema.py b/aurelio_sdk/schema.py
@@ -40,7 +40,7 @@ class Usage(BaseModel):
     pages: Optional[int] = None
     seconds: Optional[int] = None
 
-    class Config:
+    class ConfigDict:
         exclude_none = True
 
 
@@ -107,6 +107,6 @@ class ExtractResponse(BaseModel):
     )
     document: ResponseDocument = Field(..., description="Processed document")
 
-    class Config:
+    class ConfigDict:
         arbitrary_types_allowed = True
         exclude_none = True
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,6 +24,9 @@ black = {extras = ["jupyter"], version = "^24.8.0"}
 types-requests = "^2.32.0.20240914"
 datasets = "^3.0.0"
 mypy = "^1.11.2"
+pytest = "^8.3.3"
+pytest-xdist = "^3.6.1"
+pytest-asyncio = "^0.24.0"
 
 [build-system]
 requires = ["poetry-core"]