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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.0
+Version: 1.0.3
 Summary: NIA Knowledge Agent - MCP server for intelligent codebase search
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai
@@ -90,6 +90,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 +130,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.3}/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.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "nia-mcp-server"
-version = "1.0.0"
+version = "1.0.3"
 description = "NIA Knowledge Agent - MCP server for intelligent codebase search"
 readme = "README.md"
 requires-python = ">=3.8"

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.3}/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.3"

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.3}/src/nia_mcp_server/api_client.py

@@ -31,7 +31,7 @@ class NIAApiClient:
                 "User-Agent": "nia-mcp-server/1.0.0",
                 "Content-Type": "application/json"
             },
-            timeout=300.0  # 5 minute timeout for long operations
+            timeout=720.0  # 12 minute timeout for deep research operations
         )
     
     async def close(self):
@@ -474,4 +474,65 @@ class NIAApiClient:
         except httpx.HTTPStatusError as e:
             raise self._handle_api_error(e)
         except Exception as e:
-            raise APIError(f"Query failed: {str(e)}")
+            raise APIError(f"Query failed: {str(e)}")
+    
+    async def web_search(
+        self,
+        query: str,
+        num_results: int = 5,
+        category: Optional[str] = None,
+        days_back: Optional[int] = None,
+        find_similar_to: Optional[str] = None
+    ) -> Dict[str, Any]:
+        """Perform AI-powered web search."""
+        try:
+            payload = {
+                "query": query,
+                "num_results": min(num_results, 10),
+            }
+            
+            # Add optional parameters
+            if category:
+                payload["category"] = category
+            if days_back:
+                payload["days_back"] = days_back
+            if find_similar_to:
+                payload["find_similar_to"] = find_similar_to
+            
+            response = await self.client.post(
+                f"{self.base_url}/v2/web-search",
+                json=payload
+            )
+            response.raise_for_status()
+            return response.json()
+            
+        except httpx.HTTPStatusError as e:
+            raise self._handle_api_error(e)
+        except Exception as e:
+            raise APIError(f"Web search failed: {str(e)}")
+    
+    async def deep_research(
+        self,
+        query: str,
+        output_format: Optional[str] = None
+    ) -> Dict[str, Any]:
+        """Perform deep research using AI agent."""
+        try:
+            payload = {
+                "query": query,
+            }
+            
+            if output_format:
+                payload["output_format"] = output_format
+            
+            response = await self.client.post(
+                f"{self.base_url}/v2/deep-research",
+                json=payload
+            )
+            response.raise_for_status()
+            return response.json()
+            
+        except httpx.HTTPStatusError as e:
+            raise self._handle_api_error(e)
+        except Exception as e:
+            raise APIError(f"Deep research failed: {str(e)}")

{nia_mcp_server-1.0.0 → nia_mcp_server-1.0.3}/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,354 @@ 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:
+        client = await ensure_api_client()
+        
+        logger.info(f"Searching content for query: {query}")
+        
+        # Use the API client method instead of direct HTTP call
+        result = await client.web_search(
+            query=query,
+            num_results=num_results,
+            category=category,
+            days_back=days_back,
+            find_similar_to=find_similar_to
+        )
+        
+        # Extract results
+        github_repos = result.get("github_repos", [])
+        documentation = result.get("documentation", [])
+        other_content = result.get("other_content", [])
+        
+        # Format response to naturally guide next actions
+        response_text = f"## 🔍 NIA Web Search Results for: \"{query}\"\n\n"
+        
+        if days_back:
+            response_text += f"*Showing results from the last {days_back} days*\n\n"
+        
+        if find_similar_to:
+            response_text += f"*Finding content similar to: {find_similar_to}*\n\n"
+        
+        # GitHub Repositories Section
+        if github_repos:
+            response_text += f"### 📦 GitHub Repositories ({len(github_repos)} found)\n\n"
+            
+            for i, repo in enumerate(github_repos[:num_results], 1):
+                response_text += f"**{i}. {repo['title']}**\n"
+                response_text += f"   📍 `{repo['url']}`\n"
+                if repo.get('published_date'):
+                    response_text += f"   📅 Updated: {repo['published_date']}\n"
+                if repo['summary']:
+                    response_text += f"   📝 {repo['summary']}...\n"
+                if repo['highlights']:
+                    response_text += f"   ✨ Key features: {', '.join(repo['highlights'])}\n"
+                response_text += "\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_text += "**🚀 RECOMMENDED ACTION - Index this repository with NIA:**\n"
+                response_text += f"```\nIndex {github_repos[0]['owner_repo']}\n```\n"
+                response_text += "✨ This will enable AI-powered code search, understanding, and analysis!\n\n"
+            else:
+                response_text += "**🚀 Make these repositories searchable with NIA's AI:**\n"
+                response_text += f"- **Quick start:** Say \"Index {github_repos[0]['owner_repo']}\"\n"
+                response_text += "- **Index multiple:** Say \"Index all repositories\"\n"
+                response_text += "- **Benefits:** AI-powered code search, architecture understanding, implementation details\n\n"
+        
+        # Documentation Section
+        if documentation:
+            response_text += f"### 📚 Documentation ({len(documentation)} found)\n\n"
+            
+            for i, doc in enumerate(documentation[:num_results], 1):
+                response_text += f"**{i}. {doc['title']}**\n"
+                response_text += f"   📍 `{doc['url']}`\n"
+                if doc['summary']:
+                    response_text += f"   📝 {doc['summary']}...\n"
+                if doc.get('highlights'):
+                    response_text += f"   ✨ Key topics: {', '.join(doc['highlights'])}\n"
+                response_text += "\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_text += "**📖 RECOMMENDED ACTION - Index this documentation with NIA:**\n"
+                response_text += f"```\nIndex documentation {documentation[0]['url']}\n```\n"
+                response_text += "✨ NIA will make this fully searchable with AI-powered Q&A!\n\n"
+            else:
+                response_text += "**📖 Make this documentation AI-searchable with NIA:**\n"
+                response_text += f"- **Quick start:** Say \"Index documentation {documentation[0]['url']}\"\n"
+                response_text += "- **Index all:** Say \"Index all documentation\"\n"
+                response_text += "- **Benefits:** Instant answers, smart search, code examples extraction\n\n"
+        
+        # Other Content Section
+        if other_content and not github_repos and not documentation:
+            response_text += f"### 🌐 Other Content ({len(other_content)} found)\n\n"
+            
+            for i, content in enumerate(other_content[:num_results], 1):
+                response_text += f"**{i}. {content['title']}**\n"
+                response_text += f"   📍 `{content['url']}`\n"
+                if content['summary']:
+                    response_text += f"   📝 {content['summary']}...\n"
+                response_text += "\n"
+        
+        # No results found
+        if not github_repos and not documentation and not other_content:
+            response_text = f"No results found for '{query}'. Try:\n"
+            response_text += "- Using different keywords\n"
+            response_text += "- Being more specific (e.g., 'Python RAG implementation')\n"
+            response_text += "- Including technology names (e.g., 'LangChain', 'TypeScript')\n"
+        
+        # Add prominent call-to-action if we found indexable content
+        if github_repos or documentation:
+            response_text += "\n## 🎯 **Ready to unlock NIA's AI capabilities?**\n"
+            response_text += "The repositories and documentation above can be indexed for:\n"
+            response_text += "- 🤖 AI-powered code understanding and search\n"
+            response_text += "- 💡 Instant answers to technical questions\n"
+            response_text += "- 🔍 Deep architectural insights\n"
+            response_text += "- 📚 Smart documentation Q&A\n\n"
+            response_text += "**Just copy and paste the index commands above!**\n"
+        
+        # Add search metadata
+        response_text += f"\n---\n"
+        response_text += f"*Searched {result.get('total_results', 0)} sources using NIA Web Search*"
+        
+        return [TextContent(type="text", text=response_text)]
+        
+    except APIError as e:
+        logger.error(f"API Error in web search: {e}")
+        if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
+            if e.detail and "25 free API requests" in e.detail:
+                return [TextContent(
+                    type="text", 
+                    text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
+                )]
+            else:
+                return [TextContent(
+                    type="text",
+                    text=f"❌ {str(e)}\n\n💡 Tip: You've reached the free tier limit. Upgrade to Pro for unlimited access."
+                )]
+        else:
+            return [TextContent(type="text", text=f"❌ {str(e)}")]
+    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:
+        client = await ensure_api_client()
+        
+        logger.info(f"Starting deep research for: {query}")
+        
+        # Use the API client method with proper timeout handling
+        try:
+            result = await asyncio.wait_for(
+                client.deep_research(query=query, output_format=output_format),
+                timeout=720.0  # 12 minutes to allow for longer research tasks
+            )
+        except asyncio.TimeoutError:
+            logger.error(f"Deep research timed out after 12 minutes for query: {query}")
+            return [TextContent(
+                type="text",
+                text="❌ Research timed out. The query may be too complex. Try:\n"
+                     "- Breaking it into smaller questions\n"  
+                     "- Using more specific keywords\n"
+                     "- Trying the nia_web_search tool for simpler queries"
+            )]
+        
+        # Format the research results
+        response_text = f"## 🔬 NIA Deep Research Agent Results\n\n"
+        response_text += f"**Query:** {query}\n\n"
+        
+        if result.get("data"):
+            response_text += "### 📊 Research Findings:\n\n"
+            
+            # Pretty print the JSON data
+            import json
+            formatted_data = json.dumps(result["data"], indent=2)
+            response_text += f"```json\n{formatted_data}\n```\n\n"
+            
+            # Add citations if available
+            if result.get("citations"):
+                response_text += "### 📚 Sources & Citations:\n\n"
+                citation_num = 1
+                for field, citations in result["citations"].items():
+                    if citations:
+                        response_text += f"**{field}:**\n"
+                        for citation in citations[:3]:  # Limit to 3 citations per field
+                            response_text += f"{citation_num}. [{citation.get('title', 'Source')}]({citation.get('url', '#')})\n"
+                            if citation.get('snippet'):
+                                response_text += f"   > {citation['snippet'][:150]}...\n"
+                            citation_num += 1
+                        response_text += "\n"
+            
+            response_text += "### 💡 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_text += "**🚀 DISCOVERED REPOSITORIES - Index with NIA for deep analysis:**\n"
+                for repo in repos_found:
+                    response_text += f"```\nIndex {repo}\n```\n"
+                response_text += "✨ Enable AI-powered code search and architecture understanding!\n\n"
+            
+            if docs_found:
+                response_text += "**📖 DISCOVERED DOCUMENTATION - Index with NIA for smart search:**\n"
+                for doc in docs_found[:2]:  # Limit to 2 for readability
+                    response_text += f"```\nIndex documentation {doc}\n```\n"
+                response_text += "✨ Make documentation instantly searchable with AI Q&A!\n\n"
+            
+            if not repos_found and not docs_found:
+                response_text += "**🔍 Manual indexing options:**\n"
+                response_text += "- If you see any GitHub repos mentioned: Say \"Index [owner/repo]\"\n"
+                response_text += "- If you see any documentation sites: Say \"Index documentation [url]\"\n"
+                response_text += "- These will unlock NIA's powerful AI search capabilities!\n\n"
+            
+            response_text += "**📊 Other actions:**\n"
+            response_text += "- Ask follow-up questions about the research\n"
+            response_text += "- Request a different analysis format\n"
+            response_text += "- Search for more specific information\n"
+        else:
+            response_text += "No structured data returned. The research may need a more specific query."
+        
+        return [TextContent(type="text", text=response_text)]
+        
+    except APIError as e:
+        logger.error(f"API Error in deep research: {e}")
+        if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
+            if e.detail and "25 free API requests" in e.detail:
+                return [TextContent(
+                    type="text", 
+                    text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
+                )]
+            else:
+                return [TextContent(
+                    type="text",
+                    text=f"❌ {str(e)}\n\n💡 Tip: You've reached the free tier limit. Upgrade to Pro for unlimited access."
+                )]
+        else:
+            return [TextContent(type="text", text=f"❌ {str(e)}")]
+    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())