nia-mcp-server 1.0.0__tar.gz → 1.0.1__tar.gz

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.0
+Version: 1.0.1
 Summary: NIA Knowledge Agent - MCP server for intelligent codebase search
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai
@@ -20,6 +20,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.8
+Requires-Dist: exa-py>=1.0.8
 Requires-Dist: httpx>=0.24.0
 Requires-Dist: mcp>=0.1.0
 Requires-Dist: pydantic>=2.0.0
@@ -90,6 +91,39 @@ Find the authentication logic in my repositories
 What are the best practices for error handling according to the docs?
 ```
 
+### Search and index new content
+```
+Find the best RAG implementations out there
+```
+Claude will:
+1. Use the `nia_web_search` tool to find trending RAG repos
+2. Show you the results with summaries
+3. Prompt you to index the ones you want
+4. You say "Index the first two" and it indexes them!
+
+```
+What are the hottest new Rust web frameworks this week?
+```
+Claude searches trending repos and guides you through indexing them.
+
+Advanced search examples:
+```
+Find GitHub repos similar to langchain/langchain
+
+Search for AI papers published in the last 30 days
+
+What are the trending machine learning frameworks this month?
+```
+
+### Deep research questions
+```
+Compare the top 3 vector databases for RAG applications
+
+What are the pros and cons of different LLM orchestration frameworks?
+
+Research the latest developments in AI agent architectures
+```
+
 ### List your resources
 ```
 Show me all my indexed repositories and documentation
@@ -97,6 +131,21 @@ Show me all my indexed repositories and documentation
 
 ## Available Tools
 
+### Search & Research
+- **`nia_web_search`** - AI-powered search of repositories, documentation, and content
+  - Finds trending GitHub repos, relevant documentation, and more
+  - Returns structured results that guide you to index the best content
+  - Advanced options:
+    - `category`: Filter by type (github, company, research paper, news, etc.)
+    - `days_back`: Find content from the last N days (great for trending)
+    - `find_similar_to`: Search for content similar to a given URL
+  - Built into NIA's advanced search capabilities
+
+- **`nia_deep_research_agent`** - Multi-step AI research for complex questions
+  - Best for comparative analysis, comprehensive overviews
+  - Returns structured data with citations
+  - Examples: "Compare top RAG frameworks", "Analyze trends in AI safety"
+
 ### Repository Management
 - **`index_repository`** - Index a GitHub repository
 - **`list_repositories`** - List all indexed repositories

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.1}/README.md

@@ -62,6 +62,39 @@ Find the authentication logic in my repositories
 What are the best practices for error handling according to the docs?
 ```
 
+### Search and index new content
+```
+Find the best RAG implementations out there
+```
+Claude will:
+1. Use the `nia_web_search` tool to find trending RAG repos
+2. Show you the results with summaries
+3. Prompt you to index the ones you want
+4. You say "Index the first two" and it indexes them!
+
+```
+What are the hottest new Rust web frameworks this week?
+```
+Claude searches trending repos and guides you through indexing them.
+
+Advanced search examples:
+```
+Find GitHub repos similar to langchain/langchain
+
+Search for AI papers published in the last 30 days
+
+What are the trending machine learning frameworks this month?
+```
+
+### Deep research questions
+```
+Compare the top 3 vector databases for RAG applications
+
+What are the pros and cons of different LLM orchestration frameworks?
+
+Research the latest developments in AI agent architectures
+```
+
 ### List your resources
 ```
 Show me all my indexed repositories and documentation
@@ -69,6 +102,21 @@ Show me all my indexed repositories and documentation
 
 ## Available Tools
 
+### Search & Research
+- **`nia_web_search`** - AI-powered search of repositories, documentation, and content
+  - Finds trending GitHub repos, relevant documentation, and more
+  - Returns structured results that guide you to index the best content
+  - Advanced options:
+    - `category`: Filter by type (github, company, research paper, news, etc.)
+    - `days_back`: Find content from the last N days (great for trending)
+    - `find_similar_to`: Search for content similar to a given URL
+  - Built into NIA's advanced search capabilities
+
+- **`nia_deep_research_agent`** - Multi-step AI research for complex questions
+  - Best for comparative analysis, comprehensive overviews
+  - Returns structured data with citations
+  - Examples: "Compare top RAG frameworks", "Analyze trends in AI safety"
+
 ### Repository Management
 - **`index_repository`** - Index a GitHub repository
 - **`list_repositories`** - List all indexed repositories

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "nia-mcp-server"
-version = "1.0.0"
+version = "1.0.1"
 description = "NIA Knowledge Agent - MCP server for intelligent codebase search"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -30,6 +30,7 @@ dependencies = [
     "httpx>=0.24.0",
     "pydantic>=2.0.0",
     "python-dotenv>=1.0.0",
+    "exa-py>=1.0.8",
 ]
 
 [project.urls]

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.1}/src/nia_mcp_server/__init__.py

@@ -2,4 +2,4 @@
 NIA MCP Server - Proxy server for NIA Knowledge Agent
 """
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.1}/src/nia_mcp_server/server.py

@@ -7,10 +7,17 @@ import json
 import asyncio
 from typing import List, Optional, Dict, Any
 from datetime import datetime
+from urllib.parse import urlparse
 
 from mcp.server.fastmcp import FastMCP
 from mcp.types import TextContent, Resource
 from .api_client import NIAApiClient, APIError
+from dotenv import load_dotenv
+
+# Load .env from parent directory (nia-app/.env)
+from pathlib import Path
+env_path = Path(__file__).parent.parent.parent.parent / ".env"
+load_dotenv(env_path)
 
 # Configure logging
 logging.basicConfig(
@@ -19,6 +26,20 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# TOOL SELECTION GUIDE FOR AI ASSISTANTS:
+# 
+# Use 'nia_web_search' for:
+#   - "Find RAG libraries" → Simple search
+#   - "What's trending in Rust?" → Quick discovery
+#   - "Show me repos like LangChain" → Similarity search
+#
+# Use 'nia_deep_research_agent' for:
+#   - "Compare RAG vs GraphRAG approaches" → Comparative analysis
+#   - "What are the best vector databases for production?" → Evaluation needed
+#   - "Analyze the pros and cons of different LLM frameworks" → Structured analysis
+#
+# The AI should assess query complexity and choose accordingly.
+
 # Create the MCP server
 mcp = FastMCP("nia-knowledge-agent")
 
@@ -164,7 +185,7 @@ async def search_codebase(
             messages=messages,
             repositories=repositories,
             data_sources=[],  # No documentation sources
-            search_mode="unified",  # Use unified for full answers
+            search_mode="repositories",  # Use repositories mode to exclude external sources
             stream=True,
             include_sources=include_sources
         ):
@@ -699,6 +720,453 @@ async def delete_repository(repository: str) -> List[TextContent]:
             text=f"❌ Error deleting repository: {str(e)}"
         )]
 
+@mcp.tool()
+async def nia_web_search(
+    query: str,
+    num_results: int = 5,
+    category: Optional[str] = None,
+    days_back: Optional[int] = None,
+    find_similar_to: Optional[str] = None
+) -> List[TextContent]:
+    """
+    Search repositories, documentation, and other content using AI-powered search.
+    Returns results formatted to guide next actions.
+    
+    USE THIS TOOL WHEN:
+    - Finding specific repos/docs/content ("find X library", "trending Y frameworks")
+    - Looking for examples or implementations
+    - Searching for what's available on a topic
+    - Simple, direct searches that need quick results
+    - Finding similar content to a known URL
+    
+    DON'T USE THIS FOR:
+    - Comparative analysis (use nia_deep_research_agent instead)
+    - Complex multi-faceted questions (use nia_deep_research_agent instead)
+    - Questions requiring synthesis of multiple sources (use nia_deep_research_agent instead)
+    
+    Args:
+        query: Natural language search query (e.g., "best RAG implementations", "trending rust web frameworks")
+        num_results: Number of results to return (default: 5, max: 10)
+        category: Filter by category: "github", "company", "research paper", "news", "tweet", "pdf"
+        days_back: Only show results from the last N days (for trending content)
+        find_similar_to: URL to find similar content to
+        
+    Returns:
+        Search results with actionable next steps
+    """
+    try:
+        # Check for API key
+        api_key = os.getenv("EXA_API_KEY")
+        if not api_key:
+            return [TextContent(
+                type="text",
+                text="❌ NIA Web Search unavailable.\n\n"
+                     "This feature requires additional configuration. "
+                     "Please contact support for access to advanced search features."
+            )]
+        
+        # Import client
+        try:
+            from exa_py import Exa
+        except ImportError:
+            return [TextContent(
+                type="text",
+                text="❌ NIA Web Search unavailable. Please update the NIA MCP server."
+            )]
+        
+        # Initialize client
+        client = Exa(api_key)
+        
+        # Limit results to reasonable number
+        num_results = min(num_results, 10)
+        
+        logger.info(f"Searching content for query: {query}")
+        
+        # Build search parameters
+        search_params = {
+            "num_results": num_results * 2,  # Get more to filter
+            "type": "auto",  # Automatically choose best search type
+            "text": True,
+            "highlights": True
+        }
+        
+        # Add category filter if specified
+        if category:
+            # Map user-friendly categories to Exa categories
+            category_map = {
+                "github": "github",
+                "company": "company", 
+                "research": "research paper",
+                "news": "news",
+                "tweet": "tweet",
+                "pdf": "pdf",
+                "blog": "personal site"
+            }
+            if category.lower() in category_map:
+                search_params["category"] = category_map[category.lower()]
+        
+        # Add date filter for trending content
+        if days_back:
+            from datetime import datetime, timedelta
+            start_date = (datetime.now() - timedelta(days=days_back)).strftime("%Y-%m-%d")
+            search_params["start_published_date"] = start_date
+        
+        # Use similarity search if URL provided
+        if find_similar_to:
+            results = client.find_similar_and_contents(
+                find_similar_to,
+                **search_params
+            )
+        else:
+            results = client.search_and_contents(
+                query,
+                **search_params
+            )
+        
+        # Separate results by type
+        github_repos = []
+        documentation = []
+        other_content = []
+        
+        for result in results.results:
+            url = result.url
+            
+            # Categorize based on URL patterns
+            if "github.com" in url and "/tree/" not in url and "/blob/" not in url:
+                # It's a GitHub repo (not a specific file)
+                # Extract owner/repo from URL
+                try:
+                    parsed = urlparse(url)
+                    # Ensure we have a valid GitHub URL
+                    if parsed.hostname == "github.com":
+                        # Remove leading/trailing slashes and split the path
+                        path_parts = parsed.path.strip("/").split("/")
+                        # Verify we have at least owner and repo in the path
+                        if len(path_parts) >= 2 and path_parts[0] and path_parts[1]:
+                            # Extract only the owner and repo, ignoring any additional path components
+                            owner_repo = f"{path_parts[0]}/{path_parts[1]}"
+                            github_repos.append({
+                                "url": url,
+                                "owner_repo": owner_repo,
+                                "title": result.title or owner_repo,
+                                "summary": result.text[:200] if result.text else "",
+                                "highlights": result.highlights[:2] if result.highlights else [],
+                                "published_date": getattr(result, 'published_date', None)
+                            })
+                except:
+                    pass
+            elif any(doc_pattern in url for doc_pattern in ["docs.", "documentation", ".readthedocs.", "/docs/"]):
+                # It's documentation
+                documentation.append({
+                    "url": url,
+                    "title": result.title or "Documentation",
+                    "summary": result.text[:200] if result.text else "",
+                    "highlights": result.highlights[:2] if result.highlights else []
+                })
+            else:
+                # Other content
+                other_content.append({
+                    "url": url,
+                    "title": result.title or "Content",
+                    "summary": result.text[:200] if result.text else ""
+                })
+        
+        # Format response to naturally guide next actions
+        response = f"## 🔍 NIA Web Search Results for: \"{query}\"\n\n"
+        
+        if days_back:
+            response += f"*Showing results from the last {days_back} days*\n\n"
+        
+        if find_similar_to:
+            response += f"*Finding content similar to: {find_similar_to}*\n\n"
+        
+        # GitHub Repositories Section
+        if github_repos:
+            response += f"### 📦 GitHub Repositories ({len(github_repos)} found)\n\n"
+            
+            for i, repo in enumerate(github_repos[:num_results], 1):
+                response += f"**{i}. {repo['title']}**\n"
+                response += f"   📍 `{repo['url']}`\n"
+                if repo.get('published_date'):
+                    response += f"   📅 Updated: {repo['published_date']}\n"
+                if repo['summary']:
+                    response += f"   📝 {repo['summary']}...\n"
+                if repo['highlights']:
+                    response += f"   ✨ Key features: {', '.join(repo['highlights'])}\n"
+                response += "\n"
+            
+            # Be more aggressive based on query specificity
+            if len(github_repos) == 1 or any(specific_word in query.lower() for specific_word in ["specific", "exact", "particular", "find me", "looking for"]):
+                response += "**🚀 RECOMMENDED ACTION - Index this repository with NIA:**\n"
+                response += f"```\nIndex {github_repos[0]['owner_repo']}\n```\n"
+                response += "✨ This will enable AI-powered code search, understanding, and analysis!\n\n"
+            else:
+                response += "**🚀 Make these repositories searchable with NIA's AI:**\n"
+                response += f"- **Quick start:** Say \"Index {github_repos[0]['owner_repo']}\"\n"
+                response += "- **Index multiple:** Say \"Index all repositories\"\n"
+                response += "- **Benefits:** AI-powered code search, architecture understanding, implementation details\n\n"
+        
+        # Documentation Section
+        if documentation:
+            response += f"### 📚 Documentation ({len(documentation)} found)\n\n"
+            
+            for i, doc in enumerate(documentation[:num_results], 1):
+                response += f"**{i}. {doc['title']}**\n"
+                response += f"   📍 `{doc['url']}`\n"
+                if doc['summary']:
+                    response += f"   📝 {doc['summary']}...\n"
+                if doc.get('highlights'):
+                    response += f"   ✨ Key topics: {', '.join(doc['highlights'])}\n"
+                response += "\n"
+            
+            # Be more aggressive for documentation too
+            if len(documentation) == 1 or any(specific_word in query.lower() for specific_word in ["docs", "documentation", "guide", "tutorial", "reference"]):
+                response += "**📖 RECOMMENDED ACTION - Index this documentation with NIA:**\n"
+                response += f"```\nIndex documentation {documentation[0]['url']}\n```\n"
+                response += "✨ NIA will make this fully searchable with AI-powered Q&A!\n\n"
+            else:
+                response += "**📖 Make this documentation AI-searchable with NIA:**\n"
+                response += f"- **Quick start:** Say \"Index documentation {documentation[0]['url']}\"\n"
+                response += "- **Index all:** Say \"Index all documentation\"\n"
+                response += "- **Benefits:** Instant answers, smart search, code examples extraction\n\n"
+        
+        # Other Content Section
+        if other_content and not github_repos and not documentation:
+            response += f"### 🌐 Other Content ({len(other_content)} found)\n\n"
+            
+            for i, content in enumerate(other_content[:num_results], 1):
+                response += f"**{i}. {content['title']}**\n"
+                response += f"   📍 `{content['url']}`\n"
+                if content['summary']:
+                    response += f"   📝 {content['summary']}...\n"
+                response += "\n"
+        
+        # No results found
+        if not github_repos and not documentation and not other_content:
+            response = f"No results found for '{query}'. Try:\n"
+            response += "- Using different keywords\n"
+            response += "- Being more specific (e.g., 'Python RAG implementation')\n"
+            response += "- Including technology names (e.g., 'LangChain', 'TypeScript')\n"
+        
+        # Add prominent call-to-action if we found indexable content
+        if github_repos or documentation:
+            response += "\n## 🎯 **Ready to unlock NIA's AI capabilities?**\n"
+            response += "The repositories and documentation above can be indexed for:\n"
+            response += "- 🤖 AI-powered code understanding and search\n"
+            response += "- 💡 Instant answers to technical questions\n"
+            response += "- 🔍 Deep architectural insights\n"
+            response += "- 📚 Smart documentation Q&A\n\n"
+            response += "**Just copy and paste the index commands above!**\n"
+        
+        # Add search metadata
+        response += f"\n---\n"
+        response += f"*Searched {len(results.results)} sources using NIA Web Search*"
+        
+        return [TextContent(type="text", text=response)]
+        
+    except Exception as e:
+        logger.error(f"Error in NIA web search: {str(e)}")
+        return [TextContent(
+            type="text",
+            text=f"❌ NIA Web Search error: {str(e)}\n\n"
+                 "This might be due to:\n"
+                 "- Network connectivity issues\n"
+                 "- Service temporarily unavailable"
+        )]
+
+@mcp.tool()
+async def nia_deep_research_agent(
+    query: str,
+    output_format: Optional[str] = None
+) -> List[TextContent]:
+    """
+    Perform deep, multi-step research on a topic using advanced AI research capabilities.
+    Best for complex questions that need comprehensive analysis.
+    
+    USE THIS TOOL WHEN:
+    - Comparing multiple options ("compare X vs Y vs Z")
+    - Analyzing pros and cons
+    - Questions with "best", "top", "which is better"
+    - Needing structured analysis or synthesis
+    - Complex questions requiring multiple sources
+    - Questions about trends, patterns, or developments
+    - Requests for comprehensive overviews
+    
+    DON'T USE THIS FOR:
+    - Simple lookups (use nia_web_search instead)
+    - Finding a specific known item (use nia_web_search instead)
+    - Quick searches for repos/docs (use nia_web_search instead)
+    
+    COMPLEXITY INDICATORS:
+    - Words like: compare, analyze, evaluate, pros/cons, trade-offs
+    - Multiple criteria mentioned
+    - Asking for recommendations based on context
+    - Needing structured output (tables, lists, comparisons)
+    
+    Args:
+        query: Research question (e.g., "Compare top 3 RAG frameworks with pros/cons")
+        output_format: Optional structure hint (e.g., "comparison table", "pros and cons list")
+        
+    Returns:
+        Comprehensive research results with citations
+    """
+    try:
+        # Check for API key
+        api_key = os.getenv("EXA_API_KEY")
+        if not api_key:
+            return [TextContent(
+                type="text",
+                text="❌ Deep research unavailable. This advanced feature requires additional configuration."
+            )]
+        
+        # Import client
+        try:
+            from exa_py import Exa
+            import json
+        except ImportError:
+            return [TextContent(
+                type="text",
+                text="❌ Research service unavailable. Please update the NIA MCP server."
+            )]
+        
+        # Initialize client
+        client = Exa(api_key)
+        
+        logger.info(f"Starting deep research for: {query}")
+        
+        # Create a research task
+        try:
+            # Let the AI infer the schema based on the query
+            task = client.research.create_task(
+                instructions=query,
+                infer_schema=True,
+                model="exa-research-pro"  # Use the pro model
+            )
+            
+            logger.info(f"Research task created: {task.id}")
+            
+            # Poll for completion (max 2 minutes)
+            result = client.research.poll_task(
+                task.id,
+                poll_interval=3,
+                max_wait_time=120
+            )
+            
+            if result.status == "failed":
+                return [TextContent(
+                    type="text",
+                    text=f"❌ Research failed. Please try rephrasing your question."
+                )]
+            
+            # Format the research results
+            response = f"## 🔬 NIA Deep Research Agent Results\n\n"
+            response += f"**Query:** {query}\n\n"
+            
+            if result.data:
+                response += "### 📊 Research Findings:\n\n"
+                
+                # Pretty print the JSON data
+                formatted_data = json.dumps(result.data, indent=2)
+                response += f"```json\n{formatted_data}\n```\n\n"
+                
+                # Add citations if available
+                if result.citations:
+                    response += "### 📚 Sources & Citations:\n\n"
+                    citation_num = 1
+                    for field, citations in result.citations.items():
+                        if citations:
+                            response += f"**{field}:**\n"
+                            for citation in citations[:3]:  # Limit to 3 citations per field
+                                response += f"{citation_num}. [{citation.get('title', 'Source')}]({citation.get('url', '#')})\n"
+                                if citation.get('snippet'):
+                                    response += f"   > {citation['snippet'][:150]}...\n"
+                                citation_num += 1
+                            response += "\n"
+                
+                response += "### 💡 RECOMMENDED NEXT ACTIONS WITH NIA:\n\n"
+                
+                # Extract potential repos and docs from the research data
+                repos_found = []
+                docs_found = []
+                
+                # Helper function to extract URLs from nested data structures
+                def extract_urls_from_data(data, urls_list=None):
+                    if urls_list is None:
+                        urls_list = []
+                    
+                    if isinstance(data, dict):
+                        for value in data.values():
+                            extract_urls_from_data(value, urls_list)
+                    elif isinstance(data, list):
+                        for item in data:
+                            extract_urls_from_data(item, urls_list)
+                    elif isinstance(data, str):
+                        # Check if this string is a URL
+                        if data.startswith(('http://', 'https://')):
+                            urls_list.append(data)
+                    
+                    return urls_list
+                
+                # Extract all URLs from the data
+                all_urls = extract_urls_from_data(result.data)
+                
+                # Filter for GitHub repos and documentation
+                import re
+                github_pattern = r'github\.com/([a-zA-Z0-9-]+/[a-zA-Z0-9-_.]+)'
+                
+                for url in all_urls:
+                    # Check for GitHub repos
+                    github_match = re.search(github_pattern, url)
+                    if github_match and '/tree/' not in url and '/blob/' not in url:
+                        repos_found.append(github_match.group(1))
+                    # Check for documentation URLs
+                    elif any(doc_indicator in url.lower() for doc_indicator in ['docs', 'documentation', '.readthedocs.', '/guide', '/tutorial']):
+                        docs_found.append(url)
+                
+                # Remove duplicates and limit results
+                repos_found = list(set(repos_found))[:3]
+                docs_found = list(set(docs_found))[:3]
+                
+                if repos_found:
+                    response += "**🚀 DISCOVERED REPOSITORIES - Index with NIA for deep analysis:**\n"
+                    for repo in repos_found:
+                        response += f"```\nIndex {repo}\n```\n"
+                    response += "✨ Enable AI-powered code search and architecture understanding!\n\n"
+                
+                if docs_found:
+                    response += "**📖 DISCOVERED DOCUMENTATION - Index with NIA for smart search:**\n"
+                    for doc in docs_found[:2]:  # Limit to 2 for readability
+                        response += f"```\nIndex documentation {doc}\n```\n"
+                    response += "✨ Make documentation instantly searchable with AI Q&A!\n\n"
+                
+                if not repos_found and not docs_found:
+                    response += "**🔍 Manual indexing options:**\n"
+                    response += "- If you see any GitHub repos mentioned: Say \"Index [owner/repo]\"\n"
+                    response += "- If you see any documentation sites: Say \"Index documentation [url]\"\n"
+                    response += "- These will unlock NIA's powerful AI search capabilities!\n\n"
+                
+                response += "**📊 Other actions:**\n"
+                response += "- Ask follow-up questions about the research\n"
+                response += "- Request a different analysis format\n"
+                response += "- Search for more specific information\n"
+            else:
+                response += "No structured data returned. The research may need a more specific query."
+            
+            return [TextContent(type="text", text=response)]
+            
+        except Exception as e:
+            # Fallback to regular search if research API fails
+            logger.warning(f"Research API failed, falling back to search: {e}")
+            return await nia_web_search(query, num_results=10)
+            
+    except Exception as e:
+        logger.error(f"Error in deep research: {str(e)}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Research error: {str(e)}\n\n"
+                 "Try simplifying your question or using the regular nia_web_search tool."
+        )]
+
 # Resources
 
 # Note: FastMCP doesn't have list_resources or read_resource decorators

nia_mcp_server-1.0.0/install_dev.sh

@@ -1,23 +0,0 @@
-#!/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/run_local.sh

@@ -1,24 +0,0 @@
-#!/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/test_connection.py

@@ -1,64 +0,0 @@
-#!/usr/bin/env python3
-"""
-Test script to verify NIA MCP Server connection
-"""
-import os
-import asyncio
-from src.nia_mcp_server.api_client import NIAApiClient, APIError
-
-async def test_connection():
-    """Test the connection to NIA API."""
-    api_key = os.getenv("NIA_API_KEY")
-    if not api_key:
-        print("❌ NIA_API_KEY environment variable not set")
-        print("   Set it with: export NIA_API_KEY=your-api-key-here")
-        return
-    
-    api_url = os.getenv("NIA_API_URL", "https://api.trynia.ai")
-    print(f"🔄 Testing connection to NIA API at {api_url}...")
-    
-    try:
-        client = NIAApiClient(api_key, base_url=api_url)
-        
-        # Test API key validation
-        if await client.validate_api_key():
-            print("✅ API key is valid!")
-            
-            # Try to list repositories
-            try:
-                repos = await client.list_repositories()
-                print(f"\n📚 You have {len(repos)} indexed repositories")
-                
-                for repo in repos[:3]:  # Show first 3
-                    print(f"   - {repo['repository']} ({repo.get('status', 'unknown')})")
-                
-                if len(repos) > 3:
-                    print(f"   ... and {len(repos) - 3} more")
-                    
-            except APIError as e:
-                print(f"\n❌ {str(e)}")
-                if e.status_code == 403 and "lifetime limit" in str(e).lower():
-                    print("\n💡 You've reached the free tier limit of 25 API requests.")
-                    print("   Upgrade to Pro at https://trynia.ai/billing for unlimited access.")
-                
-        else:
-            print("❌ API key validation failed")
-            print("   This could be due to:")
-            print("   - Invalid API key")
-            print("   - Exceeded usage limits (free tier: 25 lifetime requests)")
-            print("   Check your API key at: https://trynia.ai/api-keys")
-            
-        await client.close()
-        
-    except APIError as e:
-        print(f"❌ {str(e)}")
-        if e.status_code == 403:
-            print("\n💡 Access forbidden. This usually means:")
-            print("   - You've exceeded your usage limits")
-            print("   - Your subscription has expired")
-            print("   Visit https://trynia.ai/billing to check your account")
-    except Exception as e:
-        print(f"❌ Error: {e}")
-
-if __name__ == "__main__":
-    asyncio.run(test_connection())

nia_mcp_server-1.0.0/test_unified_search.py

@@ -1,92 +0,0 @@
-#!/usr/bin/env python3
-"""
-Test script for unified search functionality
-"""
-import os
-import asyncio
-from src.nia_mcp_server.api_client import NIAApiClient
-
-async def test_unified_search():
-    """Test the unified search functionality."""
-    api_key = os.getenv("NIA_API_KEY")
-    if not api_key:
-        print("❌ NIA_API_KEY environment variable not set")
-        print("   Set it with: export NIA_API_KEY=your-api-key-here")
-        return
-    
-    api_url = os.getenv("NIA_API_URL", "https://api.trynia.ai")
-    print(f"🔄 Testing unified search at {api_url}...")
-    
-    try:
-        client = NIAApiClient(api_key, base_url=api_url)
-        
-        # First, list repositories and data sources
-        print("\n📚 Listing repositories...")
-        repos = await client.list_repositories()
-        print(f"Found {len(repos)} repositories")
-        for repo in repos[:3]:
-            print(f"  - {repo['repository']} ({repo.get('status', 'unknown')})")
-        
-        print("\n📄 Listing documentation sources...")
-        sources = await client.list_data_sources()
-        print(f"Found {len(sources)} documentation sources")
-        for source in sources[:3]:
-            print(f"  - {source.get('url', 'Unknown')} ({source.get('status', 'unknown')})")
-        
-        # Test unified search
-        if repos or sources:
-            print("\n🔍 Testing unified search...")
-            query = "authentication"
-            messages = [{"role": "user", "content": query}]
-            
-            # Get first completed repo and source
-            repo_list = [r["repository"] for r in repos if r.get("status") == "completed"][:1]
-            source_list = [s["id"] for s in sources if s.get("status") == "completed"][:1]
-            
-            if repo_list or source_list:
-                print(f"Searching for '{query}' across:")
-                if repo_list:
-                    print(f"  - Repository: {repo_list[0]}")
-                if source_list:
-                    print(f"  - Documentation: {sources[0].get('url', 'Unknown')}")
-                
-                # Perform unified search
-                response_text = ""
-                source_count = 0
-                
-                async for chunk in client.query_unified(
-                    messages=messages,
-                    repositories=repo_list,
-                    data_sources=source_list,
-                    search_mode="unified",
-                    stream=True,
-                    include_sources=True
-                ):
-                    try:
-                        import json
-                        data = json.loads(chunk)
-                        if "content" in data:
-                            response_text += data["content"]
-                        if "sources" in data:
-                            source_count += len(data["sources"])
-                    except:
-                        continue
-                
-                print(f"\n✅ Unified search successful!")
-                print(f"Response preview: {response_text[:200]}...")
-                print(f"Found {source_count} source references")
-            else:
-                print("⚠️  No completed repositories or documentation sources to search")
-        else:
-            print("\n⚠️  No repositories or documentation sources found")
-            print("   Index some content first using the MCP tools")
-        
-        await client.close()
-        
-    except Exception as e:
-        print(f"❌ Error: {e}")
-        import traceback
-        traceback.print_exc()
-
-if __name__ == "__main__":
-    asyncio.run(test_unified_search())