nia-mcp-server 1.0.0__tar.gz

nia_mcp_server-1.0.0/.gitignore

@@ -0,0 +1,54 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Virtual environments
+venv/
+ENV/
+env/
+.venv
+
+# IDEs
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Environment variables
+.env
+.env.local
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+.nox/
+
+# Distribution
+*.whl
+*.tar.gz

nia_mcp_server-1.0.0/ARCHITECTURE.md

@@ -0,0 +1,101 @@
+# NIA MCP Server Architecture
+
+## Overview
+
+The NIA MCP Server is a lightweight proxy that enables any MCP-compatible AI assistant to interact with NIA's production API. Users only need their NIA API key to get started.
+
+## Design Principles
+
+1. **Zero Infrastructure**: Users don't need MongoDB, vector stores, or any backend setup
+2. **API Key Only**: Single configuration requirement - just the NIA API key
+3. **Stateless**: All state is managed by NIA's backend
+4. **Lightweight**: Minimal dependencies (mcp, httpx, pydantic)
+5. **Universal**: Works with any MCP client (Claude Desktop, Continue, etc.)
+
+## Architecture
+
+```
+┌─────────────────┐     ┌──────────────────┐     ┌────────────────┐
+│   MCP Client    │────▶│  NIA MCP Server  │────▶│   NIA API      │
+│ (Claude, etc.)  │◀────│   (Local Proxy)  │◀────│  (Production)  │
+└─────────────────┘     └──────────────────┘     └────────────────┘
+        MCP                    HTTPS                   Backend
+     Protocol                Requests               Infrastructure
+```
+
+## Components
+
+### 1. MCP Server (`server.py`)
+- Implements MCP protocol
+- Exposes tools and resources
+- Handles async operations
+- Manages API client lifecycle
+
+### 2. API Client (`api_client.py`)
+- HTTP client for NIA's V2 API
+- Handles authentication
+- Streaming response support
+- Error handling and retries
+
+### 3. Tools
+- `index_repository` - Start indexing a GitHub repo
+- `search_codebase` - Search with natural language
+- `list_repositories` - List indexed repos
+- `check_repository_status` - Check indexing progress
+- `delete_repository` - Remove indexed repo
+
+### 4. Resources
+- Exposes indexed repositories as MCP resources
+- Provides metadata about each repository
+- Enables AI assistants to understand available codebases
+
+## API Endpoints Used
+
+The proxy communicates with NIA's V2 API:
+
+- `GET /v2/repositories` - List repositories
+- `POST /v2/repositories` - Index new repository
+- `GET /v2/repositories/{owner/repo}` - Get repository status
+- `DELETE /v2/repositories/{owner/repo}` - Delete repository
+- `POST /v2/query` - Search repositories (streaming)
+
+## Authentication
+
+All requests include the API key in the Authorization header:
+```
+Authorization: Bearer YOUR-API-KEY
+```
+
+## Error Handling
+
+1. **Invalid API Key**: Clear error message with link to get key
+2. **Rate Limiting**: Passes through API rate limit errors
+3. **Network Errors**: Retries with exponential backoff
+4. **Indexing Failures**: Reports detailed error messages
+
+## Distribution
+
+### PyPI Package
+```bash
+pip install nia-mcp-server
+```
+
+### NPM Wrapper (Optional)
+```bash
+npx @trynia/mcp-server
+```
+
+## Security
+
+- API keys are never logged or stored
+- All communication uses HTTPS
+- No data is stored locally
+- User isolation handled by NIA backend
+
+## Future Enhancements
+
+1. **Caching**: Local caching of repository lists
+2. **Offline Mode**: Queue operations when offline
+3. **Advanced Search**: More search options
+4. **Web/Text Sources**: Support for non-code content
+5. **Batch Operations**: Index multiple repos at once

nia_mcp_server-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 NIA (trynia.ai)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

nia_mcp_server-1.0.0/PKG-INFO

@@ -0,0 +1,200 @@
+Metadata-Version: 2.4
+Name: nia-mcp-server
+Version: 1.0.0
+Summary: NIA Knowledge Agent - MCP server for intelligent codebase search
+Project-URL: Homepage, https://trynia.ai
+Project-URL: Documentation, https://docs.trynia.ai
+Project-URL: Repository, https://github.com/trynia/nia-mcp-server
+Project-URL: Issues, https://github.com/trynia/nia-mcp-server/issues
+Author-email: NIA Team <support@trynia.ai>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,codebase,mcp,nia,search
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: mcp>=0.1.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Description-Content-Type: text/markdown
+
+# NIA MCP Server
+
+The NIA MCP Server enables AI assistants like Claude to search and understand your indexed codebases through the Model Context Protocol (MCP).
+
+## Quick Start
+
+### 1. Get your NIA API Key
+
+Sign up and get your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
+
+### 2. Install via pip
+
+```bash
+pip install nia-mcp-server
+```
+
+### 3. Configure with Claude Desktop
+
+Add to your Claude Desktop configuration (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
+
+```json
+{
+  "mcpServers": {
+    "nia": {
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+### 4. Restart Claude Desktop
+
+That's it! You can now ask Claude to index and search codebases.
+
+## Usage Examples
+
+### Index a repository
+```
+Claude, please index https://github.com/facebook/react
+```
+
+### Index documentation
+```
+Index the documentation at https://docs.python.org
+```
+
+### Search across everything
+```
+How does async/await work? Search both my code and documentation.
+```
+
+### Search only repositories
+```
+Find the authentication logic in my repositories
+```
+
+### Search only documentation
+```
+What are the best practices for error handling according to the docs?
+```
+
+### List your resources
+```
+Show me all my indexed repositories and documentation
+```
+
+## Available Tools
+
+### Repository Management
+- **`index_repository`** - Index a GitHub repository
+- **`list_repositories`** - List all indexed repositories
+- **`check_repository_status`** - Check repository indexing progress
+- **`delete_repository`** - Remove an indexed repository
+
+### Documentation Management
+- **`index_documentation`** - Index documentation or any website
+- **`list_documentation`** - List all indexed documentation sources
+- **`check_documentation_status`** - Check documentation indexing progress
+- **`delete_documentation`** - Remove indexed documentation
+
+### Unified Search
+- **`search_codebase`** - Search across repositories and/or documentation
+  - `search_mode`: "repositories", "sources", or "unified" (default)
+  - Automatically searches all indexed content if not specified
+
+## Other MCP Clients
+
+### Continue.dev
+
+Add to your `~/.continue/config.json`:
+
+```json
+{
+  "models": [...],
+  "mcpServers": [
+    {
+      "name": "nia",
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  ]
+}
+```
+
+### VS Code Cline
+
+Add to your Cline settings:
+
+```json
+{
+  "mcpServers": {
+    "nia": {
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+## Environment Variables
+
+- `NIA_API_KEY` (required) - Your NIA API key
+- `NIA_API_URL` (optional) - API endpoint (defaults to https://api.trynia.ai)
+
+## Pricing
+
+NIA offers simple, transparent pricing:
+
+- **Free Tier**: Limited usage, public repos only
+- **Pro**: Unlimited API calls, private repos, advanced features
+
+See [https://trynia.ai/pricing](https://trynia.ai/pricing) for details.
+
+## Features
+
+- 🔍 **Unified Search** - Search across code AND documentation seamlessly
+- 📚 **Documentation Indexing** - Index any website or documentation
+- 🚀 **Fast Indexing** - Index repositories and websites quickly
+- 🔒 **Private Repos** - Support for private repositories (Pro)
+- 📊 **Smart Understanding** - AI-powered code and content comprehension
+- 🌐 **Works Everywhere** - Any MCP-compatible client
+
+## Troubleshooting
+
+### "No API key provided"
+Make sure `NIA_API_KEY` is set in your MCP client configuration.
+
+### "Invalid API key"
+Check your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
+
+### "Rate limit exceeded"
+You've hit your monthly limit. Upgrade at [https://trynia.ai/billing](https://trynia.ai/billing)
+
+### Repository not indexing
+Large repositories can take a few minutes. Use `check_repository_status` to monitor progress.
+
+## Support
+
+- Documentation: [https://docs.trynia.ai](https://docs.trynia.ai)
+- Discord: [https://discord.gg/trynia](https://discord.gg/trynia)
+- Email: support@trynia.ai
+
+## License
+
+MIT License - see LICENSE file for details.

nia_mcp_server-1.0.0/README.md

@@ -0,0 +1,172 @@
+# NIA MCP Server
+
+The NIA MCP Server enables AI assistants like Claude to search and understand your indexed codebases through the Model Context Protocol (MCP).
+
+## Quick Start
+
+### 1. Get your NIA API Key
+
+Sign up and get your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
+
+### 2. Install via pip
+
+```bash
+pip install nia-mcp-server
+```
+
+### 3. Configure with Claude Desktop
+
+Add to your Claude Desktop configuration (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
+
+```json
+{
+  "mcpServers": {
+    "nia": {
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+### 4. Restart Claude Desktop
+
+That's it! You can now ask Claude to index and search codebases.
+
+## Usage Examples
+
+### Index a repository
+```
+Claude, please index https://github.com/facebook/react
+```
+
+### Index documentation
+```
+Index the documentation at https://docs.python.org
+```
+
+### Search across everything
+```
+How does async/await work? Search both my code and documentation.
+```
+
+### Search only repositories
+```
+Find the authentication logic in my repositories
+```
+
+### Search only documentation
+```
+What are the best practices for error handling according to the docs?
+```
+
+### List your resources
+```
+Show me all my indexed repositories and documentation
+```
+
+## Available Tools
+
+### Repository Management
+- **`index_repository`** - Index a GitHub repository
+- **`list_repositories`** - List all indexed repositories
+- **`check_repository_status`** - Check repository indexing progress
+- **`delete_repository`** - Remove an indexed repository
+
+### Documentation Management
+- **`index_documentation`** - Index documentation or any website
+- **`list_documentation`** - List all indexed documentation sources
+- **`check_documentation_status`** - Check documentation indexing progress
+- **`delete_documentation`** - Remove indexed documentation
+
+### Unified Search
+- **`search_codebase`** - Search across repositories and/or documentation
+  - `search_mode`: "repositories", "sources", or "unified" (default)
+  - Automatically searches all indexed content if not specified
+
+## Other MCP Clients
+
+### Continue.dev
+
+Add to your `~/.continue/config.json`:
+
+```json
+{
+  "models": [...],
+  "mcpServers": [
+    {
+      "name": "nia",
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  ]
+}
+```
+
+### VS Code Cline
+
+Add to your Cline settings:
+
+```json
+{
+  "mcpServers": {
+    "nia": {
+      "command": "nia-mcp-server",
+      "env": {
+        "NIA_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+## Environment Variables
+
+- `NIA_API_KEY` (required) - Your NIA API key
+- `NIA_API_URL` (optional) - API endpoint (defaults to https://api.trynia.ai)
+
+## Pricing
+
+NIA offers simple, transparent pricing:
+
+- **Free Tier**: Limited usage, public repos only
+- **Pro**: Unlimited API calls, private repos, advanced features
+
+See [https://trynia.ai/pricing](https://trynia.ai/pricing) for details.
+
+## Features
+
+- 🔍 **Unified Search** - Search across code AND documentation seamlessly
+- 📚 **Documentation Indexing** - Index any website or documentation
+- 🚀 **Fast Indexing** - Index repositories and websites quickly
+- 🔒 **Private Repos** - Support for private repositories (Pro)
+- 📊 **Smart Understanding** - AI-powered code and content comprehension
+- 🌐 **Works Everywhere** - Any MCP-compatible client
+
+## Troubleshooting
+
+### "No API key provided"
+Make sure `NIA_API_KEY` is set in your MCP client configuration.
+
+### "Invalid API key"
+Check your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
+
+### "Rate limit exceeded"
+You've hit your monthly limit. Upgrade at [https://trynia.ai/billing](https://trynia.ai/billing)
+
+### Repository not indexing
+Large repositories can take a few minutes. Use `check_repository_status` to monitor progress.
+
+## Support
+
+- Documentation: [https://docs.trynia.ai](https://docs.trynia.ai)
+- Discord: [https://discord.gg/trynia](https://discord.gg/trynia)
+- Email: support@trynia.ai
+
+## License
+
+MIT License - see LICENSE file for details.

nia_mcp_server-1.0.0/install_dev.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+
+# Development installation script
+
+echo "Installing NIA MCP Server for development..."
+
+# Create virtual environment
+python -m venv venv
+
+# Activate virtual environment
+source venv/bin/activate
+
+# Install in editable mode
+pip install -e .
+
+echo ""
+echo "✅ Installation complete!"
+echo ""
+echo "To use the server:"
+echo "1. Set your API key: export NIA_API_KEY=your-api-key-here"
+echo "2. Run: ./run_local.sh"
+echo ""
+echo "To test connection: python test_connection.py"

nia_mcp_server-1.0.0/pyproject.toml

@@ -0,0 +1,45 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "nia-mcp-server"
+version = "1.0.0"
+description = "NIA Knowledge Agent - MCP server for intelligent codebase search"
+readme = "README.md"
+requires-python = ">=3.8"
+license = "MIT"
+authors = [
+    { name = "NIA Team", email = "support@trynia.ai" }
+]
+keywords = ["mcp", "ai", "codebase", "search", "nia"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+
+dependencies = [
+    "mcp>=0.1.0",
+    "httpx>=0.24.0",
+    "pydantic>=2.0.0",
+    "python-dotenv>=1.0.0",
+]
+
+[project.urls]
+Homepage = "https://trynia.ai"
+Documentation = "https://docs.trynia.ai"
+Repository = "https://github.com/trynia/nia-mcp-server"
+Issues = "https://github.com/trynia/nia-mcp-server/issues"
+
+[project.scripts]
+nia-mcp-server = "nia_mcp_server.__main__:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/nia_mcp_server"]

nia_mcp_server-1.0.0/run_local.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+
+# Local development runner for NIA MCP Server
+
+# Colors
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+RED='\033[0;31m'
+NC='\033[0m'
+
+echo -e "${GREEN}NIA MCP Server - Local Development${NC}"
+echo "===================================="
+
+# Check for API key
+if [ -z "$NIA_API_KEY" ]; then
+    echo -e "${YELLOW}Warning: NIA_API_KEY not set${NC}"
+    echo "Set it with: export NIA_API_KEY=your-api-key-here"
+    echo "Get your API key at: https://trynia.ai/api-keys"
+    exit 1
+fi
+
+# Run the server
+echo -e "${GREEN}Starting MCP server...${NC}"
+python -m nia_mcp_server

nia_mcp_server-1.0.0/src/nia_mcp_server/__init__.py

@@ -0,0 +1,5 @@
+"""
+NIA MCP Server - Proxy server for NIA Knowledge Agent
+"""
+
+__version__ = "1.0.0"

nia_mcp_server-1.0.0/src/nia_mcp_server/__main__.py

@@ -0,0 +1,11 @@
+"""
+Entry point for NIA MCP Server
+"""
+from .server import run
+
+def main():
+    """Main entry point."""
+    run()
+
+if __name__ == "__main__":
+    main()