Source code for langchain_community.document_loaders.psychic

from typing import Iterator, Optional

from langchain_core.documents import Document

from langchain_community.document_loaders.base import BaseLoader


class PsychicLoader(BaseLoader):
    """从`Psychic.dev`加载。"""

    def __init__(
        self, api_key: str, account_id: str, connector_id: Optional[str] = None
    ):
        """使用API密钥、连接器ID和账户ID进行初始化。

参数:
    api_key: Psychic API密钥。
    account_id: Psychic账户ID。
    connector_id: Psychic连接器ID。
"""

        try:
            from psychicapi import ConnectorId, Psychic
        except ImportError:
            raise ImportError(
                "`psychicapi` package not found, please run `pip install psychicapi`"
            )
        self.psychic = Psychic(secret_key=api_key)
        self.connector_id = ConnectorId(connector_id)
        self.account_id = account_id

    def lazy_load(self) -> Iterator[Document]:
        psychic_docs = self.psychic.get_documents(
            connector_id=self.connector_id, account_id=self.account_id
        )
        for doc in psychic_docs.documents:
            yield Document(
                page_content=doc["content"],
                metadata={"title": doc["title"], "source": doc["uri"]},
            )