Source code for langchain_community.document_loaders.xorbits
from typing import Any
from langchain_community.document_loaders.dataframe import BaseDataFrameLoader
[docs]class XorbitsLoader(BaseDataFrameLoader):
"""加载 `Xorbits` 数据框。"""
[docs] def __init__(self, data_frame: Any, page_content_column: str = "text"):
"""使用数据框对象进行初始化。
要求:
必须安装 xorbits。您可以使用 `pip install xorbits` 进行安装。
参数:
data_frame:Xorbits 数据框对象。
page_content_column:包含页面内容的列的名称。默认为 "text"。
"""
try:
import xorbits.pandas as pd
except ImportError as e:
raise ImportError(
"Cannot import xorbits, please install with 'pip install xorbits'."
) from e
if not isinstance(data_frame, pd.DataFrame):
raise ValueError(
f"Expected data_frame to be a xorbits.pandas.DataFrame, \
got {type(data_frame)}"
)
super().__init__(data_frame, page_content_column=page_content_column)