Spaces:

priaansh
/

Open-WebUI

Paused

Open-WebUI / backend /open_webui /retrieval /vector /main.py

Upload 807 files

9de5911 verified over 1 year ago

404 Bytes

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]]]

	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]]]