mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 09:32:22 +03:00
be7acf052b
* 🌱 list files in storage & generate signed URL * ✨ add knowledge router * 🗃️ add knowledge tables * ✨ add knowledge during upload * 🚧 add knowledge a brain_knowledge models and repo * 🔥 remove brain_knowledge * ✨ add upload to knowledge table * ✨ add crawl to knowledge table * ✏️ fixes
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
import json
|
|
from multiprocessing import get_logger
|
|
|
|
from langchain.pydantic_v1 import Field
|
|
from langchain.schema import Document
|
|
from models import get_supabase_client
|
|
from supabase.client import Client
|
|
|
|
logger = get_logger()
|
|
|
|
|
|
def upload_file_storage(file, file_identifier: str):
|
|
supabase_client: Client = get_supabase_client()
|
|
# res = supabase_client.storage.create_bucket("quivr")
|
|
response = None
|
|
|
|
try:
|
|
response = supabase_client.storage.from_("quivr").upload(file_identifier, file)
|
|
return response
|
|
except Exception as e:
|
|
logger.error(e)
|
|
raise e
|
|
|
|
|
|
class DocumentSerializable(Document):
|
|
"""Class for storing a piece of text and associated metadata."""
|
|
|
|
page_content: str
|
|
metadata: dict = Field(default_factory=dict)
|
|
|
|
@property
|
|
def lc_serializable(self) -> bool:
|
|
return True
|
|
|
|
def __repr__(self):
|
|
return f"Document(page_content='{self.page_content[:50]}...', metadata={self.metadata})"
|
|
|
|
def __str__(self):
|
|
return self.__repr__()
|
|
|
|
def to_json(self) -> str:
|
|
"""Convert the Document object to a JSON string."""
|
|
return json.dumps(
|
|
{
|
|
"page_content": self.page_content,
|
|
"metadata": self.metadata,
|
|
}
|
|
)
|
|
|
|
@classmethod
|
|
def from_json(cls, json_str: str):
|
|
"""Create a Document object from a JSON string."""
|
|
data = json.loads(json_str)
|
|
return cls(page_content=data["page_content"], metadata=data["metadata"])
|