nia-mcp-server 1.0.17__py3-none-any.whl → 1.0.18__py3-none-any.whl

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/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:

{nia_mcp_server-1.0.17.dist-info → nia_mcp_server-1.0.18.dist-info}/METADATA

@@ -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.dist-info → nia_mcp_server-1.0.18.dist-info}/RECORD

@@ -1,19 +1,19 @@
-nia_mcp_server/__init__.py,sha256=Vey2MprReAMtBucCOwDlS0ZmshGrAJ0IJ1RV6BTFT50,85
+nia_mcp_server/__init__.py,sha256=_QdlhBenOh3kbV3EalKuqZxEBxmPq3i71u8R03FMYSs,85
 nia_mcp_server/__main__.py,sha256=YQSpFtDeKp18r8mKr084cHnRFV4416_EKCu9FTM8_ik,394
 nia_mcp_server/api_client.py,sha256=L9xf7rz6i2aPKKLfYOwj8Fs0_SYqJHHZgTki0y3UtcY,30415
 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=hdn81qwTR_zPvOsO4wnIs7jI3HzYHvmzgKKs9QX_7-Y,94731
+nia_mcp_server/server.py,sha256=oGNcJKU87_5GlsoAGmr7Bqn6RWQb2Vy8onkJ0HeB6oQ,95601
 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.17.dist-info/METADATA,sha256=FRAGfMUlSLv3AOkry3a6Blb0LffzGze_R9nfnJq54R4,1324
-nia_mcp_server-1.0.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.17.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.17.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.17.dist-info/RECORD,,
+nia_mcp_server-1.0.18.dist-info/METADATA,sha256=Ptsenk9rz-XEmqygQHZUEYAkDEZkhqhdR3isWhMx7GQ,1324
+nia_mcp_server-1.0.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.18.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.18.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
+nia_mcp_server-1.0.18.dist-info/RECORD,,