以编程方式使用文档

这将帮助您开始使用 Zotero 检索器。有关所有 ZoteroRetriever 功能和配置的详细文档,请前往 GitHub 页面.

设置

如果您想从单个查询获取自动化追踪,还可以设置您的 LangSmith API 密钥,只需取消下方注释即可:

os.environ["LANGSMITH_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
os.environ["LANGSMITH_TRACING"] = "true"

安装

此检索器位于 langchain-zotero-retriever 包中。我们还需要 pyzotero dependency:

pip install -qU langchain-zotero-retriever pyzotero

实例化

ZoteroRetriever 参数包括:

  • - k:要包含的结果数量(默认值:50)
  • - type:要执行的搜索类型。"Top" 检索顶级 Zotero 库项目,"items" 返回任何 Zotero 库项目。(默认值:top)
  • - get_fulltext:如果库中的项目附加了全文,则检索全文。如果为 False,或者没有附加文本,则返回空字符串作为页面_内容。(默认值:True)
  • - library_id:要搜索的 Zotero 库的 ID。连接库时必填。
  • - library_type:要搜索的库类型。"user" 表示个人库,"group" 表示共享组库。(默认值:user)
  • - api_key:如果未设置为环境变量,则使用 Zotero API 密钥。可选,但访问非公开组库或个人库时需要。如果将 ZOTERO_API_KEY 设置为环境变量,则会自动获取。
from langchain_zotero_retriever.retrievers import ZoteroRetriever

retriever = ZoteroRetriever(
    k=10,
    library_id="2319375",  # a public group library that does not require an API key for access
    library_type="group",  # set this to "user" if you are using a personal library. Personal libraries require an API key
)

用法

query外,检索器还提供以下附加搜索参数:

  • - itemType:要搜索的项目类型(例如 "book" 或 "journalArticle")
  • - tag:用于搜索附加到库项目的标签(请参阅组合多个标签的搜索语法)
  • - qmode:要使用的搜索模式。更改查询搜索的内容。"everything" 包含全文内容。"titleCreatorYear" 搜索标题、作者和年份。
  • - since:仅返回在指定库版本之后修改的对象。默认为返回所有内容。

搜索语法请参阅 Zotero API 文档: www.zotero.org/support/dev/web_api/v3/basics#search_语法

有关完整的 API 架构(包括可用的 itemTypes),请参阅: github.com/zotero/zotero-schema

query = "Zuboff"

retriever.invoke(query)
tags = [
    "Surveillance",
    "Digital Capitalism",
]  # note that providing tags as a list will result in a logical AND operation

retriever.invoke("", tag=tags)

在链中使用

由于 Zotero API 搜索的工作方式,直接将用户问题传递给 ZoteroRetriever 通常不会返回令人满意的结果。对于在链或代理框架中使用,建议将 ZoteroRetriever 转换为 工具。这样,LLM 就可以将用户查询转换为更简洁的 API 搜索查询。此外,这还允许 LLM 填充其他搜索参数,如标签或项目类型。

from typing import List, Optional, Union

from langchain_core.output_parsers import PydanticToolsParser
from langchain.tools import StructuredTool, tool
from langchain_openai import ChatOpenAI


def retrieve(
    query: str,
    itemType: Optional[str],
    tag: Optional[Union[str, List[str]]],
    qmode: str = "everything",
    since: Optional[int] = None,
):
    retrieved_docs = retriever.invoke(
        query, itemType=itemType, tag=tag, qmode=qmode, since=since
    )
    serialized_docs = "\n\n".join(
        (
            f"Metadata: { {key: doc.metadata[key] for key in doc.metadata if key != 'abstractNote'} }\n"
            f"Abstract: {doc.metadata['abstractNote']}\n"
        )
        for doc in retrieved_docs
    )

    return serialized_docs, retrieved_docs


description = """Search and return relevant documents from a Zotero library. The following search parameters can be used:

    Args:
        query: str: The search query to be used. Try to keep this specific and short, e.g. a specific topic or author name
        itemType: Optional. Type of item to search for (e.g. "book" or "journalArticle"). Multiple types can be passed as a string separated by "||", e.g. "book || journalArticle". Defaults to all types.
        tag: Optional. For searching over tags attached to library items. If documents tagged with multiple tags are to be retrieved, pass them as a list. If documents with any of the tags are to be retrieved, pass them as a string separated by "||", e.g. "tag1 || tag2"
        qmode: Search mode to use. Changes what the query searches over. "everything" includes full-text content. "titleCreatorYear" to search over title, authors and year. Defaults to "everything".
        since: Return only objects modified after the specified library version. Defaults to return everything.
    """

retriever_tool = StructuredTool.from_function(
    func=retrieve,
    name="retrieve",
    description=description,
    return_direct=True,
)


llm = ChatOpenAI(model="gpt-5.4-mini")

llm_with_tools = llm.bind_tools([retrieve])

q = "What journal articles do I have on Surveillance in the zotero library?"

chain = llm_with_tools | PydanticToolsParser(tools=[retrieve])

chain.invoke(q)

API 参考

有关所有 ZoteroRetriever 功能和配置的详细文档,请前往 GitHub 页面.

有关 Zotero API 的详细文档,请前往 Zotero API 参考.