nia-mcp-server 1.0.23__py3-none-any.whl → 1.0.25__py3-none-any.whl

nia_mcp_server/__init__.py

@@ -3,4 +3,4 @@ NIA MCP Server - Proxy server for NIA Knowledge Agent
 """
 
 
-__version__ = "1.0.23"
+__version__ = "1.0.25"

nia_mcp_server/api_client.py

@@ -28,7 +28,7 @@ class NIAApiClient:
         self.client = httpx.AsyncClient(
             headers={
                 "Authorization": f"Bearer {api_key}",
-                "User-Agent": "nia-mcp-server/1.0.23",
+                "User-Agent": "nia-mcp-server/1.0.25",
                 "Content-Type": "application/json"
             },
             timeout=720.0  # 12 minute timeout for deep research operations
@@ -926,7 +926,9 @@ class NIAApiClient:
         content: str,
         agent_source: str,
         tags: List[str] = None,
-        metadata: Dict[str, Any] = None
+        metadata: Dict[str, Any] = None,
+        nia_references: Optional[Dict[str, Any]] = None,
+        edited_files: Optional[List[Dict[str, Any]]] = None
     ) -> Dict[str, Any]:
         """Save a conversation context for cross-agent sharing."""
         try:
@@ -939,6 +941,12 @@ class NIAApiClient:
                 "metadata": metadata or {}
             }
 
+            # Add new structured fields if provided
+            if nia_references is not None:
+                payload["nia_references"] = nia_references
+            if edited_files is not None:
+                payload["edited_files"] = edited_files
+
             response = await self.client.post(
                 f"{self.base_url}/v2/contexts",
                 json=payload

nia_mcp_server/server.py

@@ -2826,160 +2826,6 @@ async def nia_package_search_read_file(
                  f"- The line range is valid (1-based, max 200 lines)"
         )]
 
-# @mcp.tool()
-async def visualize_codebase(
-    repository: str
-) -> List[TextContent]:
-    """
-    Open the graph visualization for an indexed repository in a browser.
-    
-    This tool launches a browser with the interactive graph visualization
-    that shows the code structure, relationships, and dependencies of
-    the indexed codebase.
-    
-    Args:
-        repository: Repository in owner/repo format (e.g., "facebook/react")
-        
-    Returns:
-        Status message with the URL that was opened
-        
-    Examples:
-        - visualize_codebase("facebook/react")
-        - visualize_codebase("langchain-ai/langchain")
-    """
-    try:
-        client = await ensure_api_client()
-        
-        logger.info(f"Looking up repository: {repository}")
-        
-        # List all repositories to find the matching one
-        repositories = await client.list_repositories()
-        
-        # Find the repository by name
-        matching_repo = None
-        for repo in repositories:
-            if repo.get("repository") == repository:
-                matching_repo = repo
-                break
-        
-        if not matching_repo:
-            # Try case-insensitive match as fallback
-            repository_lower = repository.lower()
-            for repo in repositories:
-                if repo.get("repository", "").lower() == repository_lower:
-                    matching_repo = repo
-                    break
-        
-        if not matching_repo:
-            return [TextContent(
-                type="text",
-                text=f"❌ Repository '{repository}' not found.\n\n"
-                     f"Available repositories:\n" +
-                     "\n".join(f"- {r.get('repository')}" for r in repositories if r.get('repository')) +
-                     "\n\nUse `list_repositories` to see all indexed repositories."
-            )]
-        
-        # Check if the repository is fully indexed
-        status = matching_repo.get("status", "unknown")
-        # Use the actual project ID if available, fall back to repository_id
-        repository_id = matching_repo.get("id") or matching_repo.get("repository_id")
-        
-        if not repository_id:
-            return [TextContent(
-                type="text",
-                text=f"❌ No repository ID found for '{repository}'. This may be a data inconsistency."
-            )]
-        
-        if status != "completed":
-            warning_msg = f"⚠️ Note: Repository '{repository}' is currently {status}.\n"
-            if status == "indexing":
-                warning_msg += "The visualization may show incomplete data.\n\n"
-            elif status == "error":
-                error_msg = matching_repo.get("error", "Unknown error")
-                warning_msg += f"Error: {error_msg}\n\n"
-            else:
-                warning_msg += "The visualization may not be available.\n\n"
-        else:
-            warning_msg = ""
-        
-        # Determine the base URL based on the API URL
-        api_base_url = client.base_url
-        if "localhost" in api_base_url or "127.0.0.1" in api_base_url:
-            # Local development
-            app_base_url = "http://localhost:3000"
-        else:
-            # Production
-            app_base_url = "https://app.trynia.ai"
-        
-        # Construct the visualization URL
-        visualization_url = f"{app_base_url}/visualize/{repository_id}"
-        
-        # Try to open the browser
-        try:
-            webbrowser.open(visualization_url)
-            browser_opened = True
-            open_msg = "✅ Opening graph visualization in your default browser..."
-        except Exception as e:
-            logger.warning(f"Failed to open browser: {e}")
-            browser_opened = False
-            open_msg = "⚠️ Could not automatically open browser."
-        
-        # Format the response
-        response_lines = [
-            f"# Graph Visualization: {repository}",
-            "",
-            warning_msg if warning_msg else "",
-            open_msg,
-            "",
-            f"**URL:** {visualization_url}",
-            "",
-        ]
-        
-        if matching_repo.get("display_name"):
-            response_lines.append(f"**Display Name:** {matching_repo['display_name']}")
-        
-        response_lines.extend([
-            f"**Branch:** {matching_repo.get('branch', 'main')}",
-            f"**Status:** {status}",
-            "",
-            "## Features Available:",
-            "- 🔍 Interactive force-directed graph",
-            "- 🎨 Color-coded node types (functions, classes, files, etc.)",
-            "- 🔗 Relationship visualization (calls, imports, inherits, etc.)",
-            "- 💬 Click on any node to chat with that specific code element",
-            "- 🔎 Search and filter capabilities",
-            "- 📊 Graph statistics and insights"
-        ])
-        
-        if not browser_opened:
-            response_lines.extend([
-                "",
-                "**Manual Access:**",
-                f"Copy and paste this URL into your browser: {visualization_url}"
-            ])
-        
-        return [TextContent(
-            type="text",
-            text="\n".join(response_lines)
-        )]
-        
-    except APIError as e:
-        logger.error(f"API Error in visualize_codebase: {e}")
-        if e.status_code == 403 or "free tier limit" in str(e).lower():
-            return [TextContent(
-                type="text",
-                text=f"❌ {str(e)}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited access."
-            )]
-        else:
-            return [TextContent(type="text", text=f"❌ {str(e)}")]
-    except Exception as e:
-        logger.error(f"Error in visualize_codebase: {e}")
-        return [TextContent(
-            type="text",
-            text=f"❌ Error opening visualization: {str(e)}"
-        )]
-
-
 @mcp.tool()
 async def nia_bug_report(
     description: str,
@@ -3094,7 +2940,9 @@ async def save_context(
     content: str,
     agent_source: str,
     tags: Optional[List[str]] = None,
-    metadata: Optional[dict] = None
+    metadata: Optional[dict] = None,
+    nia_references: Optional[dict] = None,
+    edited_files: Optional[List[dict]] = None
 ) -> List[TextContent]:
     """
     Save a conversation context for cross-agent sharing.
@@ -3108,8 +2956,16 @@ async def save_context(
         summary: Brief summary of the conversation (10-1000 characters)
         content: Full conversation context - the agent should compact the conversation history but keep all important parts togethers, as well as code snippets. No excuses.
         agent_source: Which agent is creating this context (e.g., "cursor", "claude-code", "windsurf")
-        tags: Optional list of searchable tags (up to 10 tags)
+        tags: Optional list of searchable tags
         metadata: Optional metadata like file paths, repositories discussed, etc.
+        nia_references: Structured data about NIA resources used during conversation
+            Format: {
+                "indexed_resources": [{"identifier": "owner/repo", "resource_type": "repository", "purpose": "Used for authentication patterns"}],
+                "search_queries": [{"query": "JWT implementation", "query_type": "codebase", "resources_searched": ["owner/repo"], "key_findings": "Found JWT utils in auth folder"}],
+                "session_summary": "Used NIA to explore authentication patterns and API design"
+            }
+        edited_files: List of files that were modified during conversation
+            Format: [{"file_path": "src/auth.ts", "operation": "modified", "changes_description": "Added JWT validation", "key_changes": ["Added validate() function"]}]
 
     Returns:
         Confirmation of successful context save with context ID
@@ -3121,7 +2977,11 @@ async def save_context(
             content="User asked about implementing streaming... [agent should include conversation]",
             agent_source="cursor",
             tags=["streaming", "ai-sdk", "implementation"],
-            metadata={"files": ["src/api/chat.ts"], "repos": ["myproject"]}
+            nia_references={
+                "indexed_resources": [{"identifier": "vercel/ai", "resource_type": "repository", "purpose": "Reference for streaming implementation"}],
+                "search_queries": [{"query": "streaming API", "query_type": "documentation", "key_findings": "Found useChat hook with streaming"}]
+            },
+            edited_files=[{"file_path": "src/chat.ts", "operation": "created", "changes_description": "Added streaming chat component"}]
         )
     """
     try:
@@ -3144,9 +3004,6 @@ async def save_context(
         if not agent_source or not agent_source.strip():
             return [TextContent(type="text", text="❌ Error: Agent source is required")]
 
-        if tags and len(tags) > 10:
-            return [TextContent(type="text", text="❌ Error: Maximum 10 tags allowed")]
-
         client = await ensure_api_client()
 
         logger.info(f"Saving context: title='{title}', agent={agent_source}, content_length={len(content)}")
@@ -3157,7 +3014,9 @@ async def save_context(
             content=content,
             agent_source=agent_source.strip(),
             tags=tags or [],
-            metadata=metadata or {}
+            metadata=metadata or {},
+            nia_references=nia_references,
+            edited_files=edited_files or []
         )
 
         context_id = result.get("id")
@@ -3360,10 +3219,81 @@ async def retrieve_context(context_id: str) -> List[TextContent]:
 
         response += f"\n📝 **Summary:**\n{context['summary']}\n\n"
 
+        # Add NIA References - CRITICAL for context handoffs
+        # Use 'or {}' to handle cases where nia_references is None (not just missing)
+        nia_references = context.get('nia_references') or {}
+        if nia_references:
+            response += "🧠 **NIA RESOURCES USED - RECOMMENDED ACTIONS:**\n"
+
+            indexed_resources = nia_references.get('indexed_resources', [])
+            if indexed_resources:
+                response += "**📦 Re-index these resources:**\n"
+                for resource in indexed_resources:
+                    identifier = resource.get('identifier', 'Unknown')
+                    resource_type = resource.get('resource_type', 'unknown')
+                    purpose = resource.get('purpose', 'No purpose specified')
+
+                    if resource_type == 'repository':
+                        response += f"• `Index {identifier}` - {purpose}\n"
+                    elif resource_type == 'documentation':
+                        response += f"• `Index documentation {identifier}` - {purpose}\n"
+                    else:
+                        response += f"• `Index {identifier}` ({resource_type}) - {purpose}\n"
+                response += "\n"
+
+            search_queries = nia_references.get('search_queries', [])
+            if search_queries:
+                response += "**🔍 Useful search queries to re-run:**\n"
+                for query in search_queries:
+                    query_text = query.get('query', 'Unknown query')
+                    query_type = query.get('query_type', 'search')
+                    key_findings = query.get('key_findings', 'No findings specified')
+                    resources_searched = query.get('resources_searched', [])
+
+                    response += f"• **Query:** `{query_text}` ({query_type})\n"
+                    if resources_searched:
+                        response += f"  **Resources:** {', '.join(resources_searched)}\n"
+                    response += f"  **Key Findings:** {key_findings}\n"
+                response += "\n"
+
+            session_summary = nia_references.get('session_summary')
+            if session_summary:
+                response += f"**📋 NIA Session Summary:** {session_summary}\n\n"
+
+        # Add Edited Files - CRITICAL for code handoffs
+        # Use 'or []' to handle cases where edited_files is None (not just missing)
+        edited_files = context.get('edited_files') or []
+        if edited_files:
+            response += "📝 **FILES MODIFIED - READ THESE TO GET UP TO SPEED:**\n"
+            for file_info in edited_files:
+                file_path = file_info.get('file_path', 'Unknown file')
+                operation = file_info.get('operation', 'modified')
+                changes_desc = file_info.get('changes_description', 'No description')
+                key_changes = file_info.get('key_changes', [])
+                language = file_info.get('language', '')
+
+                operation_emoji = {
+                    'created': '🆕',
+                    'modified': '✏️',
+                    'deleted': '🗑️'
+                }.get(operation, '📄')
+
+                response += f"• {operation_emoji} **`{file_path}`** ({operation})\n"
+                response += f"  **Changes:** {changes_desc}\n"
+
+                if key_changes:
+                    response += f"  **Key Changes:** {', '.join(key_changes)}\n"
+                if language:
+                    response += f"  **Language:** {language}\n"
+
+                response += f"  **💡 Action:** Read this file with: `Read {file_path}`\n"
+            response += "\n"
+
         # Add metadata if available
-        metadata = context.get('metadata', {})
+        # Use 'or {}' to handle cases where metadata is None (not just missing)
+        metadata = context.get('metadata') or {}
         if metadata:
-            response += f"📊 **Metadata:**\n"
+            response += f"📊 **Additional Metadata:**\n"
             for key, value in metadata.items():
                 if isinstance(value, list):
                     response += f"• **{key}:** {', '.join(map(str, value))}\n"
@@ -3374,10 +3304,15 @@ async def retrieve_context(context_id: str) -> List[TextContent]:
         response += f"📄 **Full Context:**\n\n{context['content']}\n\n"
 
         response += f"---\n"
-        response += f"💡 **Tips:**\n"
-        response += f"• This context was created by {context['agent_source']}\n"
-        response += f"• Use this information to understand the strategic planning\n"
-        response += f"• You can now implement based on this context!"
+        response += f"🚀 **NEXT STEPS FOR SEAMLESS HANDOFF:**\n"
+        response += f"• This context was created by **{context['agent_source']}**\n"
+
+        if nia_references.get('search_queries'):
+            response += f"• **RECOMMENDED:** Re-run the search queries to get the same insights\n"
+        if edited_files:
+            response += f"• **ESSENTIAL:** Read the modified files above to understand code changes\n"
+
+        response += f"• Use the summary and full context to understand the strategic planning\n"
 
         return [TextContent(type="text", text=response)]
 

{nia_mcp_server-1.0.23.dist-info → nia_mcp_server-1.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.23
+Version: 1.0.25
 Summary: Nia Knowledge Agent
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai

{nia_mcp_server-1.0.23.dist-info → nia_mcp_server-1.0.25.dist-info}/RECORD

@@ -1,19 +1,19 @@
-nia_mcp_server/__init__.py,sha256=fRGM5mMWuUp59DtNCXNq3BQKxw30WI3zptbKFGTkt_A,87
+nia_mcp_server/__init__.py,sha256=_IbmZ23vCb-dFXG91RhQmFhM7IbSunVkuUJH0sPLZDg,87
 nia_mcp_server/__main__.py,sha256=YQSpFtDeKp18r8mKr084cHnRFV4416_EKCu9FTM8_ik,394
-nia_mcp_server/api_client.py,sha256=LOFKnadDmtwRLm5QhUkKn8X7Be5GlZf0WaGnahsQ9tQ,41177
+nia_mcp_server/api_client.py,sha256=zXeJQAbvwVsAWgyeEko_5kEm8VIilcpcnGg5qMKFBnU,41546
 nia_mcp_server/cli.py,sha256=32VSPNIocXtDgVBDZNZsxvj3kytBn54_a1pIE84vOdY,1834
 nia_mcp_server/profiles.py,sha256=2DD8PFRr5Ij4IK4sPUz0mH8aKjkrEtkKLC1R0iki2bA,7221
 nia_mcp_server/project_init.py,sha256=T0-ziJhofL4L8APwnM43BLhxtlmOHaYH-V9PF2yXLw4,7138
 nia_mcp_server/rule_transformer.py,sha256=wCxoQ1Kl_rI9mUFnh9kG5iCXYU4QInrmFQOReZfAFVo,11000
-nia_mcp_server/server.py,sha256=0y0feDisTv-hhQeqWDByqo2yaDJMvTILXZFYZULs248,157233
+nia_mcp_server/server.py,sha256=oyFSz0xqMl57m-3I7oL0dzGjRAtBzB0dWa0VcqRy5hw,156571
 nia_mcp_server/setup.py,sha256=nJXVY8NHGtWROtoH8DW-3uOgyuPs4F9dW0cNhcbCLrM,5355
 nia_mcp_server/assets/rules/claude_rules.md,sha256=HNL5GJMUbFxSpNbIAJUQWqAywjMl4lf530I1in69aNY,7380
 nia_mcp_server/assets/rules/cursor_rules.md,sha256=hd6lhzNrK1ULQUYIEVeOnyKnuLKq4hmwZPbMqGUI1Lk,1720
 nia_mcp_server/assets/rules/nia_rules.md,sha256=l6sx000uqoczoHYqOPp4hnNgyfpnhvO9NyT0fVx5nU0,8059
 nia_mcp_server/assets/rules/vscode_rules.md,sha256=fqn4aJO_bhftaCGkVoquruQHf3EaREQJQWHXq6a4FOk,6967
 nia_mcp_server/assets/rules/windsurf_rules.md,sha256=PzU2as5gaiVsV6PAzg8T_-GR7VCyRQGMjAHcSzYF_ms,3354
-nia_mcp_server-1.0.23.dist-info/METADATA,sha256=UB-GuIDpM_9P32kz5bG98aEfHEb51rA3rPliKMtI9fs,1324
-nia_mcp_server-1.0.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.23.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.23.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.23.dist-info/RECORD,,
+nia_mcp_server-1.0.25.dist-info/METADATA,sha256=tsMmAI_cehFBGhWmw1FeXco_WFFJziTnBnI57HG7Tdk,1324
+nia_mcp_server-1.0.25.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.25.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.25.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
+nia_mcp_server-1.0.25.dist-info/RECORD,,