Spaces:
Sleeping
Sleeping
File size: 643 Bytes
7fdb8e9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from pydantic import UUID4, Field
from rag_demo.preprocessing.base import VectorBaseDocument
class Query(VectorBaseDocument):
content: str
metadata: dict = Field(default_factory=dict)
class Config:
category = "query"
@classmethod
def from_str(cls, query: str) -> "Query":
return Query(content=query.strip("\n "))
def replace_content(self, new_content: str) -> "Query":
return Query(
id=self.id,
content=new_content,
metadata=self.metadata,
)
class EmbeddedQuery(Query):
embedding: list[float]
class Config:
category = "query"
|