chore: Redefine TextQuery as BaseModel with coerce

argilla-io · Nov 5, 2024 · ff30ee1 · ff30ee1
1 parent fb16594
commit ff30ee1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/argilla-server/.env.dev b/argilla-server/.env.dev
@@ -1,2 +1,2 @@
 ALEMBIC_CONFIG=src/argilla_server/alembic.ini
-ARGILLA_DATABASE_URL=sqlite+aiosqlite:///${HOME}/.argilla/argilla.db?check_same_thread=False
+ARGILLA_DATABASE_URL=sqlite+aiosqlite:///${HOME}/.argilla/argilla.dev.db?check_same_thread=False
diff --git a/argilla-server/src/argilla_server/search_engine/base.py b/argilla-server/src/argilla_server/search_engine/base.py
@@ -114,11 +114,12 @@ class Order:
     order: SortOrder
 
 
-@dataclasses.dataclass
-class TextQuery:
+class TextQuery(BaseModel):
     q: str
     field: Optional[str] = None
 
+    model_config = ConfigDict(coerce_numbers_to_str=True)
+
 
 class UserResponseStatusFilter(BaseModel):
     statuses: List[ResponseStatusFilter]