open-webui/backend/open_webui/retrieval/vector/main.py
Timothy Jaeryang Baek d3d161f723 wip
2024-12-10 00:54:13 -08:00

20 lines
404 B
Python

from pydantic import BaseModel
from typing import Optional, List, Any
class VectorItem(BaseModel):
id: str
text: str
vector: List[float | int]
metadata: Any
class GetResult(BaseModel):
ids: Optional[List[List[str]]]
documents: Optional[List[List[str]]]
metadatas: Optional[List[List[Any]]]
class SearchResult(GetResult):
distances: Optional[List[List[float | int]]]