DocsGPT/application/vectorstore/document_class.py

9 lines
308 B
Python
Raw Normal View History

2024-01-05 17:16:16 +00:00
class Document(str):
"""Class for storing a piece of text and associated metadata."""
def __new__(cls, page_content: str, metadata: dict):
instance = super().__new__(cls, page_content)
instance.page_content = page_content
instance.metadata = metadata
return instance