nia-mcp-server 1.0.17__tar.gz → 1.0.18__tar.gz

{nia_mcp_server-1.0.17 → nia_mcp_server-1.0.18}/PKG-INFO

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

{nia_mcp_server-1.0.17 → nia_mcp_server-1.0.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "nia-mcp-server"
-version = "1.0.17"
+version = "1.0.18"
 description = "Nia Knowledge Agent"
 readme = "README.md"
 requires-python = ">=3.8"

{nia_mcp_server-1.0.17 → nia_mcp_server-1.0.18}/src/nia_mcp_server/__init__.py

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

{nia_mcp_server-1.0.17 → nia_mcp_server-1.0.18}/src/nia_mcp_server/server.py

@@ -225,6 +225,7 @@ async def search_codebase(
         # Stream the response using unified query
         response_parts = []
         sources_parts = []
+        follow_up_questions = []
         
         async for chunk in client.query_unified(
             messages=messages,
@@ -243,6 +244,10 @@ async def search_codebase(
                 if "sources" in data and data["sources"]:
                     logger.debug(f"Received sources data: {type(data['sources'])}, count: {len(data['sources'])}")
                     sources_parts.extend(data["sources"])
+                
+                if "follow_up_questions" in data and data["follow_up_questions"]:
+                    follow_up_questions = data["follow_up_questions"]
+                    logger.debug(f"Received {len(follow_up_questions)} follow-up questions")
                     
             except json.JSONDecodeError as e:
                 logger.warning(f"Failed to parse JSON chunk: {chunk}, error: {e}")
@@ -294,6 +299,13 @@ async def search_codebase(
             response_text += "\n💡 **Need more details from a source?**\n\n"
             response_text += "If you need more information from the source links provided above, use the `read_source_content` tool from the available tools provided by Nia to get full context about that particular source.\n"
         
+        # Add follow-up questions if available
+        if follow_up_questions:
+            response_text += "\n\n## 🔍 Suggested Follow-up Questions\n\n"
+            for i, question in enumerate(follow_up_questions, 1):
+                response_text += f"{i}. {question}\n"
+            response_text += "\n*These questions are based on the search results and can help you explore deeper insights.*\n"
+        
         return [TextContent(type="text", text=response_text)]
         
     except APIError as e:
@@ -367,6 +379,7 @@ async def search_documentation(
         # Stream the response using unified query
         response_parts = []
         sources_parts = []
+        follow_up_questions = []
         
         async for chunk in client.query_unified(
             messages=messages,
@@ -385,6 +398,10 @@ async def search_documentation(
                 if "sources" in data and data["sources"]:
                     logger.debug(f"Received doc sources data: {type(data['sources'])}, count: {len(data['sources'])}")
                     sources_parts.extend(data["sources"])
+                
+                if "follow_up_questions" in data and data["follow_up_questions"]:
+                    follow_up_questions = data["follow_up_questions"]
+                    logger.debug(f"Received {len(follow_up_questions)} follow-up questions for documentation")
                     
             except json.JSONDecodeError as e:
                 logger.warning(f"Failed to parse JSON chunk in documentation search: {chunk}, error: {e}")
@@ -432,6 +449,13 @@ async def search_documentation(
             response_text += "\n💡 **Need more details from a source?**\n\n"
             response_text += "If you need more information from the source links provided above, use the `read_source_content` tool from the available tools provided by Nia to get full context about that particular source.\n"
         
+        # Add follow-up questions if available
+        if follow_up_questions:
+            response_text += "\n\n## 🔍 Suggested Follow-up Questions\n\n"
+            for i, question in enumerate(follow_up_questions, 1):
+                response_text += f"{i}. {question}\n"
+            response_text += "\n*These questions are based on the documentation and can help you explore related topics.*\n"
+        
         return [TextContent(type="text", text=response_text)]
         
     except APIError as e:
@@ -497,20 +521,7 @@ async def list_repositories() -> List[TextContent]:
             if repo.get("status") == "completed":
                 lines.append(f"- **Usage:** `search_codebase(query, [\"{repo_name}\"])`")
         
-        # Add general usage instructions at the end
-        lines.extend([
-            "\n---",
-            "\n## Usage Tips",
-            "- To search all repositories: `search_codebase(\"your query\")`",
-            "- To search specific repository: `search_codebase(\"your query\", [\"owner/repo\"])`"
-        ])
-        
-        if has_folder_repos:
-            lines.extend([
-                "- For folder-indexed repositories: Use the EXACT repository path shown above",
-                "  Example: `search_codebase(\"query\", [\"owner/repo/tree/branch/folder\"])`"
-            ])
-        
+        # Return without usage tips
         return [TextContent(type="text", text="\n".join(lines))]
         
     except APIError as e: