以编程方式使用文档

Nia 是一个搜索和索引 API,持续从文档、研究论文、数据集、代码库等提供上下文——因此智能体永远不会依赖过时数据。可扩展、便宜 5 倍、可靠。

概述

集成详情

ClassPackagePY 支持Version
NiaToolkit@nozomioai/langchain-nia!NPM - Version

设置

集成位于 @nozomioai/langchain-nia package:

npm install @nozomioai/langchain-nia @langchain/core
yarn add @nozomioai/langchain-nia @langchain/core
pnpm add @nozomioai/langchain-nia @langchain/core

凭证

在以下网址注册 trynia.ai 获取 API 密钥。

process.env.NIA_API_KEY = "nk_...";

设置起来也很有帮助(但不是必需的) LangSmith 用于一流的可观察性:

process.env.LANGSMITH_TRACING = "true";
process.env.LANGSMITH_API_KEY = "your-api-key";

实例化

使用工具包

const toolkit = new NiaToolkit({
  includeSearch: true,        // 5 search tools
  includeSources: true,       // 7 source management tools
  includeGithub: true,        // 4 GitHub tools
  includeContexts: true,      // 2 context/memory tools
  includeDependencies: true,  // 2 dependency tools
});

const tools = toolkit.getTools(); // 20 tools

使用单个工具

const tool = new NiaSearch();

调用

搜索索引源

const tool = new NiaSearch();
const result = await tool.invoke({ query: "how to use React hooks" });

网络搜索

const tool = new NiaWebSearch();
const result = await tool.invoke({ query: "latest Node.js release", numResults: 5 });

在智能体内

const toolkit = new NiaToolkit({ includeSearch: true });
const tools = toolkit.getTools();

const llm = new ChatOpenAI({ model: "gpt-4o" });
const llmWithTools = llm.bindTools(tools);

const result = await llmWithTools.invoke("Search for React hooks best practices");

可用工具

ToolDescription
NiaSearch跨索引仓库、文档和数据源的语义搜索
NiaWebSearch带类别过滤和日期范围的网络搜索
NiaDeepResearch多步骤综合研究
NiaUniversalSearch同时搜索所有源
NiaAdvisor针对索引文档分析代码
NiaIndex为新源建立索引(仓库、文档、论文、数据集)
NiaSourceList列出带过滤条件的索引源
NiaSourceSubscribe订阅预索引公共源
NiaSourceSync重新同步源以获取最新变更
NiaReadRead files/pages from indexed sources
NiaGrep在索引源中进行正则表达式搜索
NiaExplore浏览索引源的文件树
NiaGitHubSearch在 GitHub 仓库中搜索代码
NiaGitHubRead读取 GitHub 仓库中的文件
NiaGitHubGlob查找匹配 glob 模式的文件
NiaGitHubTree浏览仓库文件树
NiaContextSave保存上下文以供跨智能体共享
NiaContextSearch对保存的上下文进行语义搜索
NiaDependencySubscribe自动订阅项目依赖项的文档
NiaDependencyAnalyze从清单预览可索引的依赖项

API 参考

有关详细文档,请参阅 langchain-nia GitHub 仓库.