Skip to main content
Nia is a search and index API that continuously provides context from docs, research papers, datasets, codebases, and more—so agents never rely on stale data. Scalable, 5x cheaper, and reliable.

​
Overview

​
Integration details

ClassPackageSerializableJS supportVersion
NiaToolkitlangchain-nia❌❌PyPI - Version

​
Tool features

Returns artifactNative asyncToolkitNumber of toolsPricing
❌✅✅20Free tier available

​
Setup

The integration lives in the langchain-nia package. 
pip install -U langchain-nia

​
Credentials

Sign up at trynia.ai to get an API key. 
import getpass
import os

if not os.environ.get("NIA_API_KEY"):
    os.environ["NIA_API_KEY"] = getpass.getpass("Nia API key:\n")
It’s also helpful (but not needed) to set up LangSmith for best-in-class observability/ of your tool calls. To enable automated tracing, set your LangSmith API key: 
os.environ["LANGSMITH_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
os.environ["LANGSMITH_TRACING"] = "true"

​
Instantiation

​
Using the toolkit

The NiaToolkit provides all 20 Nia tools with a shared API wrapper. Use include_* flags to control which tool groups are available: 
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()

​
Using individual tools

You can also use tools directly: 
from langchain_nia import NiaSearch

tool = NiaSearch()

​
Invocation

​
Search across indexed sources

from langchain_nia import NiaSearch

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

​
Search the web

from langchain_nia import NiaWebSearch

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

​
Read files from indexed sources

from langchain_nia import NiaRead

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

​
Within an agent

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"}]}
)

​
Available tools

​
Search tools

  • NiaSearch - Semantic search across indexed repos, docs, datasets, and more
  • NiaWebSearch - Web search with category filtering and date range
  • NiaDeepResearch - Multi-step comprehensive research
  • NiaUniversalSearch - Search all sources simultaneously
  • NiaAdvisor - Analyze code against indexed documentation

​
Source management tools

  • NiaIndex - Index new sources (repos, docs, papers, datasets)
  • NiaSourceList - List indexed sources with filtering
  • NiaSourceSubscribe - Subscribe to pre-indexed public sources
  • NiaSourceSync - Re-sync sources to pull latest changes
  • NiaRead - Read files/pages from indexed sources
  • NiaGrep - Regex search within indexed sources
  • NiaExplore - Browse file tree of indexed sources

​
GitHub tools

  • NiaGitHubSearch - Search code in GitHub repositories
  • NiaGitHubRead - Read files from GitHub repos
  • NiaGitHubGlob - Find files matching glob patterns
  • NiaGitHubTree - Browse repo file tree structure

​
Context and memory tools

  • NiaContextSave - Save context for cross-agent sharing
  • NiaContextSearch - Semantic search over saved contexts

​
Dependency tools

  • NiaDependencySubscribe - Auto-subscribe to docs for project dependencies
  • NiaDependencyAnalyze - Preview what would be indexed from a manifest

​
API reference

For detailed documentation of all Nia tools and configurations, see the langchain-nia GitHub repository.
Edit this page on GitHub or file an issue.
Connect these docs to Claude, VSCode, and more via MCP for real-time answers.