Skip to main content
Nia 是一个搜索和索引 API，它持续从文档、研究论文、数据集、代码库等中提供上下文，因此代理程序永远不会依赖过时的数据。可扩展、成本降低 5 倍且可靠。

​
概述

​
集成详情

类包可序列化JS 支持版本
NiaToolkitlangchain-nia❌❌PyPI - Version

​
工具特性

返回工件原生异步工具包工具数量定价
❌✅✅20提供免费层级

​
设置

集成位于 langchain-nia 包中。 
pip install -U langchain-nia

​
凭证

在 trynia.ai 注册以获取 API 密钥。 
import getpass
import os

if not os.environ.get("NIA_API_KEY"):
    os.environ["NIA_API_KEY"] = getpass.getpass("Nia API key:\n")
设置 LangSmith 以获得最佳的可观察性/ 也很有帮助（但不是必需的）。要启用自动追踪，请设置您的 LangSmith API 密钥： 
os.environ["LANGSMITH_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
os.environ["LANGSMITH_TRACING"] = "true"

​
实例化

​
使用工具包

NiaToolkit 提供所有 20 个 Nia 工具，并附带共享的 API 包装器。使用 include_* 标志来控制哪些工具组可用： 
from langchain_nia import NiaToolkit

toolkit = NiaToolkit(
    include_search=True,        # NiaSearch, NiaWebSearch, NiaDeepResearch, NiaUniversalSearch, NiaAdvisor
    include_sources=True,       # NiaIndex, NiaSourceList, NiaSourceSubscribe, NiaSourceSync, NiaRead, NiaGrep, NiaExplore
    include_github=True,        # NiaGitHubSearch, NiaGitHubRead, NiaGitHubGlob, NiaGitHubTree
    include_contexts=True,      # NiaContextSave, NiaContextSearch
    include_dependencies=True,  # NiaDependencySubscribe, NiaDependencyAnalyze
)
tools = toolkit.get_tools()

​
使用单个工具

您也可以直接使用工具： 
from langchain_nia import NiaSearch

tool = NiaSearch()

​
调用

​
跨索引源搜索

from langchain_nia import NiaSearch

tool = NiaSearch()
tool.invoke({"query": "how to use React hooks"})

​
搜索网络

from langchain_nia import NiaWebSearch

tool = NiaWebSearch()
tool.invoke({"query": "latest Python release", "num_results": 5})

​
从索引源读取文件

from langchain_nia import NiaRead

tool = NiaRead()
tool.invoke({"source_id": "your-source-id", "path": "README.md"})

​
在代理内使用

from langchain_nia import NiaToolkit

toolkit = NiaToolkit(include_search=True, include_sources=False, include_github=False, include_contexts=False, include_dependencies=False)
tools = toolkit.get_tools()

# pip install -qU "langchain[anthropic]"
from langchain.agents import create_agent

agent = create_agent(
    model="claude-sonnet-4-6",
    tools=tools,
)

agent.invoke(
    {"messages": [{"role": "user", "content": "Search for React hooks best practices"}]}
)

​
可用工具

​
搜索工具

  • NiaSearch - 对索引的仓库、文档、数据集等进行语义搜索
  • NiaWebSearch - 带有类别过滤和日期范围的网络搜索
  • NiaDeepResearch - 多步骤综合研究
  • NiaUniversalSearch - 同时搜索所有源
  • NiaAdvisor - 根据索引的文档分析代码

​
源管理工具

  • NiaIndex - 索引新源（仓库、文档、论文、数据集）
  • NiaSourceList - 列出索引的源并进行过滤
  • NiaSourceSubscribe - 订阅预索引的公共源
  • NiaSourceSync - 重新同步源以拉取最新更改
  • NiaRead - 从索引源读取文件/页面
  • NiaGrep - 在索引源内进行正则表达式搜索
  • NiaExplore - 浏览索引源的文件树

​
GitHub 工具

  • NiaGitHubSearch - 在 GitHub 仓库中搜索代码
  • NiaGitHubRead - 从 GitHub 仓库读取文件
  • NiaGitHubGlob - 查找匹配 glob 模式的文件
  • NiaGitHubTree - 浏览仓库文件树结构

​
上下文和内存工具

  • NiaContextSave - 保存上下文以供跨代理共享
  • NiaContextSearch - 对保存的上下文进行语义搜索

​
依赖工具

  • NiaDependencySubscribe - 自动订阅项目依赖的文档
  • NiaDependencyAnalyze - 预览将从清单中索引的内容

​
API 参考

有关所有 Nia 工具和配置的详细文档，请参阅 langchain-nia GitHub 仓库。
在 GitHub 上编辑此页面 或 提交问题。
通过 MCP 将这些文档连接到 Claude、VSCode 等 以获取实时答案。