open-webui/backend/open_webui/apps/rag/vector/main.py
Timothy J. Baek 939bfd153e refac
2024-09-13 01:18:20 -04: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]]]