This commit is contained in:
Timothy J. Baek
2024-09-10 04:37:06 +01:00
parent d5f13dd9e0
commit 522afbb0a0
7 changed files with 240 additions and 127 deletions

View File

@@ -0,0 +1,16 @@
from pydantic import BaseModel
from typing import Optional, List, Any
class VectorItem(BaseModel):
id: str
text: str
vector: List[float | int]
metadata: Any
class QueryResult(BaseModel):
ids: Optional[List[List[str]]]
distances: Optional[List[List[float | int]]]
documents: Optional[List[List[str]]]
metadatas: Optional[List[List[Any]]]