19 lines
610 B
Python
19 lines
610 B
Python
from langchain.document_loaders import Docx2txtLoader
|
|
|
|
from embedchain.helper_classes.json_serializable import register_deserializable
|
|
from embedchain.loaders.base_loader import BaseLoader
|
|
|
|
|
|
@register_deserializable
|
|
class DocxFileLoader(BaseLoader):
|
|
def load_data(self, url):
|
|
"""Load data from a .docx file."""
|
|
loader = Docx2txtLoader(url)
|
|
output = []
|
|
data = loader.load()
|
|
content = data[0].page_content
|
|
meta_data = data[0].metadata
|
|
meta_data["url"] = "local"
|
|
output.append({"content": content, "meta_data": meta_data})
|
|
return output
|