mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
4cb5f4c353
This PR should make it easier for linters to do type checking and for IDEs to jump to definition of code. See #20050 as a template for this PR. - As a byproduct: Added 3 missed `test_imports`. - Added missed `SolarChat` in to __init___.py Added it into test_import ut. - Added `# type: ignore` to fix linting. It is not clear, why linting errors appear after ^ changes. --------- Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
from typing import Iterator
|
|
|
|
from langchain_core.documents import Document
|
|
|
|
from langchain_community.document_loaders.base import BaseBlobParser
|
|
from langchain_community.document_loaders.blob_loaders import Blob
|
|
|
|
|
|
class MsWordParser(BaseBlobParser):
|
|
"""Parse the Microsoft Word documents from a blob."""
|
|
|
|
def lazy_parse(self, blob: Blob) -> Iterator[Document]: # type: ignore[valid-type]
|
|
"""Parse a Microsoft Word document into the Document iterator.
|
|
|
|
Args:
|
|
blob: The blob to parse.
|
|
|
|
Returns: An iterator of Documents.
|
|
|
|
"""
|
|
try:
|
|
from unstructured.partition.doc import partition_doc
|
|
from unstructured.partition.docx import partition_docx
|
|
except ImportError as e:
|
|
raise ImportError(
|
|
"Could not import unstructured, please install with `pip install "
|
|
"unstructured`."
|
|
) from e
|
|
|
|
mime_type_parser = {
|
|
"application/msword": partition_doc,
|
|
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": (
|
|
partition_docx
|
|
),
|
|
}
|
|
if blob.mimetype not in ( # type: ignore[attr-defined]
|
|
"application/msword",
|
|
"application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
|
):
|
|
raise ValueError("This blob type is not supported for this parser.")
|
|
with blob.as_bytes_io() as word_document: # type: ignore[attr-defined]
|
|
elements = mime_type_parser[blob.mimetype](file=word_document) # type: ignore[attr-defined]
|
|
text = "\n\n".join([str(el) for el in elements])
|
|
metadata = {"source": blob.source} # type: ignore[attr-defined]
|
|
yield Document(page_content=text, metadata=metadata)
|