Skip to main content
Open In ColabOpen on GitHub

Arcee

Arcee 帮助开发 SLMs——小型、专业化、安全且可扩展的语言模型。

这个笔记本演示了如何使用ArceeRetriever类来检索与Arcee的Domain Adapted Language Models(DALMs)相关的文档。

设置​

在使用ArceeRetriever之前，请确保设置了Arcee API密钥作为ARCEE_API_KEY环境变量。您也可以将API密钥作为命名参数进行传递。

from langchain_community.retrievers import ArceeRetriever

retriever = ArceeRetriever(
    model="DALM-PubMed",
    # arcee_api_key="ARCEE-API-KEY" # if not already set in the environment
)
API 参考:Arcee 检索器

附加配置​

您可以根据需要配置ArceeRetriever的参数，如arcee_api_url、arcee_app_url和model_kwargs。 在对象初始化时设置model_kwargs将使用过滤器和大小作为后续检索的默认值。

retriever = ArceeRetriever(
    model="DALM-PubMed",
    # arcee_api_key="ARCEE-API-KEY", # if not already set in the environment
    arcee_api_url="https://custom-api.arcee.ai",  # default is https://api.arcee.ai
    arcee_app_url="https://custom-app.arcee.ai",  # default is https://app.arcee.ai
    model_kwargs={
        "size": 5,
        "filters": [
            {
                "field_name": "document",
                "filter_type": "fuzzy_search",
                "value": "Einstein",
            }
        ],
    },
)

检索文档​

可以提供查询以从上传的上下文中检索相关文档。以下是一个示例：

query = "Can AI-driven music therapy contribute to the rehabilitation of patients with disorders of consciousness?"
documents = retriever.invoke(query)

附加参数​

Arcee 允许你应用 filters 并设置 size（以数量为单位）检索的文档数。过滤器有助于缩小结果范围。这是如何使用这些参数的方法：

# Define filters
filters = [
    {"field_name": "document", "filter_type": "fuzzy_search", "value": "Music"},
    {"field_name": "year", "filter_type": "strict_search", "value": "1905"},
]

# Retrieve documents with filters and size params
documents = retriever.invoke(query, size=5, filters=filters)