ims-mcp 1.0.28__py3-none-any.whl → 1.0.30__py3-none-any.whl

ims_mcp/server.py

@@ -148,15 +148,17 @@ def get_username() -> str:
     return username
 
 
-def get_repository_from_context(ctx) -> str:
-    """Extract repository names from MCP roots via Context (with 5-min cache).
+async def get_repository_from_context(ctx) -> str:
+    """Extract repository name from MCP roots via session.list_roots() (with 5-min cache).
     
-    Uses MCP's native ctx.list_roots() to get project directories from client.
-    Combines multiple roots with comma separator. Caches result for 5 minutes
-    to avoid excessive queries while still reflecting workspace changes.
+    Uses MCP protocol's roots/list request to get workspace directories from client.
+    Checks client capabilities before requesting. Combines multiple roots with comma.
+    Falls back to parsing client_id if roots are not available.
+    
+    Caches result for 5 minutes to avoid excessive requests.
     
     Args:
-        ctx: FastMCP Context object with list_roots() method
+        ctx: FastMCP Context object with request_context.session
     
     Returns:
         Comma-separated repository names or "unknown"
@@ -170,25 +172,75 @@ def get_repository_from_context(ctx) -> str:
         (current_time - _repository_cache_time) < REPOSITORY_CACHE_TTL):
         return _cached_repository
     
-    # Cache expired or empty - fetch fresh roots
+    result = "unknown"
+    
+    # Try 1: Request roots from client via MCP protocol
     try:
-        roots = ctx.list_roots()
-        if not roots:
-            result = "unknown"
-        else:
-            # Extract basename from all roots and combine with comma
-            repo_names = [os.path.basename(root.uri) for root in roots]
-            result = ", ".join(repo_names)
+        # Access session through request_context
+        from mcp import types
+        session = ctx.request_context.session
         
-        # Update cache
-        _cached_repository = result
-        _repository_cache_time = current_time
-        debug_print(f"[ims-mcp] Repository cache updated: {result}")
+        # Check if client supports roots capability
+        has_roots = session.check_client_capability(
+            types.ClientCapabilities(roots=types.RootsCapability())
+        )
         
-        return result
+        if has_roots:
+            # Request roots from client using MCP protocol
+            roots_result = await session.list_roots()
+            
+            if roots_result.roots:
+                # Extract basename from all roots and combine with comma
+                repo_names = [os.path.basename(root.uri).rstrip('/') for root in roots_result.roots]
+                result = ", ".join(repo_names)
+                
+                if DEBUG_MODE:
+                    roots_info = {
+                        "method": "roots/list",
+                        "count": len(roots_result.roots),
+                        "roots": [{"uri": str(r.uri), "name": r.name} for r in roots_result.roots],
+                        "result": result
+                    }
+                    debug_print(f"[ims-mcp] Repository: {json.dumps(roots_info)}")
+        else:
+            if DEBUG_MODE:
+                debug_print("[ims-mcp] Client doesn't support roots capability, trying fallback")
     except Exception as e:
-        debug_print(f"[ims-mcp] Failed to get roots from context: {e}")
-        return "unknown"
+        error_details = {
+            "method": "roots/list",
+            "error": str(e),
+            "error_type": type(e).__name__
+        }
+        if DEBUG_MODE:
+            debug_print(f"[ims-mcp] Failed to get roots: {json.dumps(error_details)}, trying fallback")
+    
+    # Try 2: Fallback to parsing client_id if roots didn't work
+    if result == "unknown":
+        try:
+            client_id = ctx.client_id
+            if client_id and '/' in str(client_id):
+                # Parse path from client_id (e.g., "cursor:/path/to/repo")
+                path = str(client_id).split(':', 1)[-1]
+                repo_name = os.path.basename(path.rstrip('/'))
+                if repo_name:
+                    result = repo_name
+                    if DEBUG_MODE:
+                        debug_print(f"[ims-mcp] Repository from client_id: {result} (client_id={client_id})")
+        except Exception as e:
+            if DEBUG_MODE:
+                error_details = {
+                    "method": "client_id_fallback",
+                    "error": str(e),
+                    "error_type": type(e).__name__,
+                    "client_id": str(getattr(ctx, 'client_id', None))
+                }
+                debug_print(f"[ims-mcp] Fallback failed: {json.dumps(error_details)}")
+    
+    # Update cache
+    _cached_repository = result
+    _repository_cache_time = current_time
+    
+    return result
 
 
 def before_send_hook(event: dict[str, Any]) -> Optional[dict[str, Any]]:
@@ -268,14 +320,22 @@ def get_posthog_client():
         # Import PostHog (lazy import to avoid dependency if not used)
         from posthog import Posthog
         
-        # Get optional host override
+        # Silence PostHog's internal logger when not debugging
+        if not DEBUG_MODE:
+            posthog_logger = logging.getLogger('posthog')
+            posthog_logger.setLevel(logging.CRITICAL)  # Only show critical errors
+            posthog_logger.propagate = False
+        
+        # Get optional host override (use US cloud by default for GeoIP)
         host = os.getenv('POSTHOG_HOST', 'https://us.i.posthog.com')
         
         # Initialize with before_send hook
+        # IMPORTANT: disable_geoip=False because MCP runs locally on user's machine (not server)
         _posthog_client = Posthog(
             project_api_key=api_key,
             host=host,
             debug=DEBUG_MODE,
+            disable_geoip=False,  # Enable GeoIP - MCP uses user's actual IP
             on_error=lambda e: debug_print(f"[posthog] Error: {e}"),
             before_send=before_send_hook
         )
@@ -321,7 +381,7 @@ def track_tool_call(func: Callable) -> Callable:
             
             # Extract user context
             username = get_username()
-            repository = get_repository_from_context(ctx) if ctx else "unknown"
+            repository = await get_repository_from_context(ctx) if ctx else "unknown"
             tool_name = func.__name__
             
             # Build distinct_id
@@ -333,8 +393,35 @@ def track_tool_call(func: Callable) -> Callable:
             properties.update({
                 'username': username,
                 'repository': repository,
-                'mcp_server': 'KnowledgeBase'
+                'mcp_server': 'Rosetta',
+                'mcp_server_version': __version__  # Separate property for easy version filtering
             })
+            # Note: GeoIP is enabled via disable_geoip=False in client initialization
+            
+            # Set screen_name with fallback chain (use first available)
+            screen_name = None
+            
+            if 'query' in kwargs and kwargs['query']:
+                screen_name = kwargs['query'][:100]
+            elif 'title' in kwargs and kwargs['title']:
+                screen_name = kwargs['title']
+            elif 'document_id' in kwargs and kwargs['document_id']:
+                screen_name = kwargs['document_id']
+            elif 'document_ids' in kwargs and kwargs['document_ids']:
+                ids = kwargs['document_ids']
+                if isinstance(ids, list) and ids:
+                    screen_name = ', '.join(ids[:5])
+            elif 'tags' in kwargs and kwargs['tags']:
+                tags = kwargs['tags']
+                if isinstance(tags, list) and tags:
+                    screen_name = f"tags: {', '.join(tags[:5])}"
+            elif 'filters' in kwargs and kwargs['filters']:
+                filters = kwargs['filters']
+                if isinstance(filters, dict) and filters:
+                    screen_name = f"filters: {', '.join(f'{k}={v}' for k, v in list(filters.items())[:3])}"
+            
+            if screen_name:
+                properties['$screen_name'] = screen_name
             
             # Capture event (async, non-blocking)
             posthog.capture(

{ims_mcp-1.0.28.dist-info → ims_mcp-1.0.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.28
+Version: 1.0.30
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -111,7 +111,7 @@ Add to `.cursor/mcp.json`:
   "mcpServers": {
     "KnowledgeBase": {
       "command": "uvx",
-      "args": ["ims-mcp"],
+      "args": ["ims-mcp@latest"],
       "env": {
         "R2R_API_BASE": "http://localhost:7272",
         "R2R_COLLECTION": "aia-r1"
@@ -128,7 +128,7 @@ Add to `.cursor/mcp.json`:
   "mcpServers": {
     "KnowledgeBase": {
       "command": "uvx",
-      "args": ["ims-mcp"],
+      "args": ["ims-mcp@latest"],
       "env": {
         "R2R_API_BASE": "https://your-server.example.com/",
         "R2R_COLLECTION": "your-collection",
@@ -149,7 +149,7 @@ Add to Claude Desktop configuration (`~/Library/Application Support/Claude/claud
   "mcpServers": {
     "ims": {
       "command": "uvx",
-      "args": ["ims-mcp"],
+      "args": ["ims-mcp@latest"],
       "env": {
         "R2R_API_BASE": "http://localhost:7272",
         "R2R_COLLECTION": "my-collection"
@@ -166,7 +166,7 @@ Any MCP client can use ims-mcp by specifying the command and environment variabl
 ```json
 {
   "command": "uvx",
-  "args": ["ims-mcp"],
+  "args": ["ims-mcp@latest"],
   "env": {
     "R2R_API_BASE": "http://localhost:7272"
   }
@@ -358,7 +358,7 @@ To **disable** analytics, set `POSTHOG_API_KEY` to an empty string in your MCP c
   "mcpServers": {
     "KnowledgeBase": {
       "command": "uvx",
-      "args": ["ims-mcp"],
+      "args": ["ims-mcp@latest"],
       "env": {
         "R2R_API_BASE": "https://your-server.com/",
         "R2R_COLLECTION": "aia-r1",
@@ -398,7 +398,9 @@ To track analytics in your own PostHog project, provide your Project API Key:
 
 **User Context:**
 - Username (from `USER`/`USERNAME`/`LOGNAME` environment variables + `whoami` fallback)
-- Repository names (from MCP client roots via `ctx.list_roots()`, comma-separated if multiple, 5-min cache)
+- Repository names (from MCP `roots/list` protocol request, comma-separated if multiple; fallback to `client_id` parsing; 5-min cache)
+- MCP server identifier (`mcp_server: "Rosetta"`) and version (`mcp_server_version: "1.0.30"`)
+- GeoIP enabled via `disable_geoip=False` in client initialization (MCP runs locally on user's machine, IP is user's actual location)
 
 **Business Parameters** (usage patterns):
 - `query` - Search queries

ims_mcp-1.0.30.dist-info/RECORD

@@ -0,0 +1,10 @@
+ims_mcp/__init__.py,sha256=tc179xWmUwO8ZD5y4k3M005BsF-mJtUgziVo7dJdj7E,1163
+ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
+ims_mcp/server.py,sha256=5lcuunYIbbpY3dHzuKmdbF4QrZn08B_KVYxkIXrsEoI,40368
+ims_mcp/resources/bootstrap.md,sha256=-b5SpUGO_KXP5HmagY_Y9krslHPsVthk3QhLGkca6Ig,2522
+ims_mcp-1.0.30.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
+ims_mcp-1.0.30.dist-info/METADATA,sha256=H4YpkrUNCjbZaLWAJayxmexL4s-aM5IZ6jJzZeCYHXk,12703
+ims_mcp-1.0.30.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ims_mcp-1.0.30.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
+ims_mcp-1.0.30.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
+ims_mcp-1.0.30.dist-info/RECORD,,

ims_mcp-1.0.28.dist-info/RECORD

@@ -1,10 +0,0 @@
-ims_mcp/__init__.py,sha256=tc179xWmUwO8ZD5y4k3M005BsF-mJtUgziVo7dJdj7E,1163
-ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
-ims_mcp/server.py,sha256=yMoEV-3_l87UqWKeEiOR6X8C8SUMy0tKDBL5rQTL9dc,36191
-ims_mcp/resources/bootstrap.md,sha256=-b5SpUGO_KXP5HmagY_Y9krslHPsVthk3QhLGkca6Ig,2522
-ims_mcp-1.0.28.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
-ims_mcp-1.0.28.dist-info/METADATA,sha256=BE3nFlAlk5MIJIWa1T4XHYJZzD2Ygg90mD4e51ekdyg,12412
-ims_mcp-1.0.28.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ims_mcp-1.0.28.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
-ims_mcp-1.0.28.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
-ims_mcp-1.0.28.dist-info/RECORD,,