nia-mcp-server 1.0.9__py3-none-any.whl → 1.0.10__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.9"
+__version__ = "1.0.10"

nia_mcp_server/api_client.py

@@ -402,7 +402,9 @@ class NIAApiClient:
         url_patterns: List[str] = None,
         exclude_patterns: List[str] = None,
         max_age: int = None,
-        only_main_content: bool = True
+        only_main_content: bool = True,
+        wait_for: int = None,
+        include_screenshot: bool = None
     ) -> Dict[str, Any]:
         """Create a new documentation/web data source."""
         try:
@@ -415,10 +417,14 @@ class NIAApiClient:
             # Add optional parameters
             if max_age is not None:
                 payload["max_age"] = max_age
-            # Always use markdown for documentation scraping
-            payload["formats"] = ["markdown"]
+            # Don't hardcode formats - let backend defaults apply
+            # This allows screenshots to be captured by default
             if only_main_content is not None:
                 payload["only_main_content"] = only_main_content
+            if wait_for is not None:
+                payload["wait_for"] = wait_for
+            if include_screenshot is not None:
+                payload["include_screenshot"] = include_screenshot
             
             response = await self.client.post(
                 f"{self.base_url}/v2/data-sources",

nia_mcp_server/server.py

@@ -506,7 +506,9 @@ async def index_documentation(
     url_patterns: Optional[List[str]] = None,
     exclude_patterns: Optional[List[str]] = None,
     max_age: Optional[int] = None,
-    only_main_content: Optional[bool] = True
+    only_main_content: Optional[bool] = True,
+    wait_for: Optional[int] = None,
+    include_screenshot: Optional[bool] = None
 ) -> List[TextContent]:
     """
     Index documentation or website for intelligent search.
@@ -517,6 +519,8 @@ async def index_documentation(
         exclude_patterns: Optional list of URL patterns to exclude from crawling (e.g., ["/blog/*", "/changelog/*"])
         max_age: Maximum age of cached content in seconds (for fast scraping mode)
         only_main_content: Extract only main content (removes navigation, ads, etc.)
+        wait_for: Time to wait for page to load in milliseconds (defaults to backend setting)
+        include_screenshot: Whether to capture full page screenshots (defaults to backend setting)
         
     Returns:
         Status of the indexing operation
@@ -525,6 +529,7 @@ async def index_documentation(
         - When started indexing, prompt users to either use check_documentation_status tool or go to app.trynia.ai to check the status.
         - By default, crawls the entire domain (up to 10,000 pages)
         - Use exclude_patterns to filter out unwanted sections like blogs, changelogs, etc.
+        - Screenshots are captured by default to provide visual context
     """
     try:
         client = await ensure_api_client()
@@ -536,7 +541,9 @@ async def index_documentation(
             url_patterns=url_patterns,
             exclude_patterns=exclude_patterns,
             max_age=max_age,
-            only_main_content=only_main_content
+            only_main_content=only_main_content,
+            wait_for=wait_for,
+            include_screenshot=include_screenshot
         )
         
         source_id = result.get("id")

{nia_mcp_server-1.0.9.dist-info → nia_mcp_server-1.0.10.dist-info}/METADATA

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

{nia_mcp_server-1.0.9.dist-info → nia_mcp_server-1.0.10.dist-info}/RECORD

@@ -1,17 +1,17 @@
-nia_mcp_server/__init__.py,sha256=65ET2rdy2p2-auZWPBpQBp1VoU64W83jRGx82wLPyao,84
+nia_mcp_server/__init__.py,sha256=KyI9fMELt-32jL_i_v49z4TF3p3kOf7D_pfv62ZoxEU,85
 nia_mcp_server/__main__.py,sha256=XY11ESL4hctu-BBgtPATFZyd1o-O7wE7y-UOSoNs-hw,152
-nia_mcp_server/api_client.py,sha256=zcTWTDkG9KYj5xunx8A8t1JDLctJJqNqB_ZiGZc9H1M,24362
+nia_mcp_server/api_client.py,sha256=_6Oh3ZTqNVCOQsQcV8ep9mHZ0giNbE_ht93pamU-PlI,24654
 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=m8tvape27a72NyQEho8_tT9z0plRGBR0kcEB6gSrqsw,65816
+nia_mcp_server/server.py,sha256=W3yyKiAsumVqVpS9rMiKtPhav8cXiw8VqG4DGuUOi0k,66246
 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=mvdYrkoiRgxeROhtnRXCV53TX5B9wqLiCJ6oYTqSPfY,6345
 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.9.dist-info/METADATA,sha256=aV26HPJjZ14aZX4OJykkFrJ_7B-m3iP_FX4p7bB_2qw,6766
-nia_mcp_server-1.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.9.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.9.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.9.dist-info/RECORD,,
+nia_mcp_server-1.0.10.dist-info/METADATA,sha256=0AYn0fZjCz0tuB_vuCCpVuyRbr4Bi2nCgQkgXaZ77RY,6767
+nia_mcp_server-1.0.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.10.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.10.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
+nia_mcp_server-1.0.10.dist-info/RECORD,,