fix: pydantic error

superagent-ai · Apr 26, 2024 · 12d72f2 · 12d72f2
1 parent 371f171
commit 12d72f2
Showing 1 changed file with 2 additions and 21 deletions.
diff --git a/models/query.py b/models/query.py
@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, RootModel
 from typing import List, Optional, Union, Any
 
 from models.document import BaseDocumentChunk
@@ -7,26 +7,7 @@
 from qdrant_client.http.models import Filter as QdrantFilter
 
 
-class PineconeFilter(BaseModel):
-    __root__: dict[str, Union[str, float, int, bool, List, dict]]
-
-
-class AstraFilter(BaseModel):
-    __root__: dict[str, Any]
-
-
-class WeaviateFilter(BaseModel):
-    __root__: dict
-
-
-class PgVectorFilter(BaseModel):
-    __root__: dict
-
-
-class Filter(BaseModel):
-    __root__: Union[
-        PineconeFilter, QdrantFilter, WeaviateFilter, AstraFilter, PgVectorFilter
-    ]
+Filter = Union[QdrantFilter, dict]
 
 
 class RequestPayload(BaseModel):