update pydantic

cleanlab · Feb 7, 2025 · 89647dc · 89647dc
1 parent 6b6f07a
commit 89647dc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ classifiers = [
 ]
 dependencies = [
   "codex-sdk==0.1.0a9",
-  "pydantic>=1.9.0, <3",
+  "pydantic==2.10",
 ]
 
 [project.urls]

diff --git a/src/cleanlab_codex/internal/organization.py b/src/cleanlab_codex/internal/organization.py
@@ -9,6 +9,4 @@
 
 
 def list_organizations(client: _Codex) -> list[Organization]:
-    return [
-        Organization.model_validate(org.model_dump()) for org in client.users.myself.organizations.list().organizations
-    ]
+    return [Organization.model_validate(org) for org in client.users.myself.organizations.list().organizations]
diff --git a/src/cleanlab_codex/internal/project.py b/src/cleanlab_codex/internal/project.py
@@ -25,16 +25,14 @@ def query_project(
 ) -> tuple[Optional[str], Optional[Entry]]:
     maybe_entry = client.projects.entries.query(project_id, question=question)
     if maybe_entry is not None:
-        entry = Entry.model_validate(maybe_entry.model_dump())
+        entry = Entry.model_validate(maybe_entry)
         if entry.answer is not None:
             return entry.answer, entry
 
         return fallback_answer, entry
 
     if not read_only:
-        created_entry = Entry.model_validate(
-            client.projects.entries.add_question(project_id, question=question).model_dump()
-        )
+        created_entry = Entry.model_validate(client.projects.entries.add_question(project_id, question=question))
         return fallback_answer, created_entry
 
     return fallback_answer, None