ims-mcp 1.0.29__tar.gz → 1.0.31__tar.gz

{ims_mcp-1.0.29 → ims_mcp-1.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.29
+Version: 1.0.31
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -399,8 +399,8 @@ 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 `roots/list` protocol request, comma-separated if multiple; fallback to `client_id` parsing; 5-min cache)
-- Library: "Rosetta" with version number
-- GeoIP enabled for location tracking
+- 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.29 → ims_mcp-1.0.31}/README.md

@@ -371,8 +371,8 @@ 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 `roots/list` protocol request, comma-separated if multiple; fallback to `client_id` parsing; 5-min cache)
-- Library: "Rosetta" with version number
-- GeoIP enabled for location tracking
+- 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.29 → ims_mcp-1.0.31}/ims_mcp/server.py

@@ -189,19 +189,21 @@ async def get_repository_from_context(ctx) -> str:
             # Request roots from client using MCP protocol
             roots_result = await session.list_roots()
             
+            # Debug: serialize entire object to see what we received
+            if DEBUG_MODE:
+                try:
+                    # Serialize the entire roots_result object
+                    serialized = json.dumps(roots_result.model_dump(), default=str)
+                    debug_print(f"[ims-mcp] Roots received: {serialized}")
+                except Exception as e:
+                    debug_print(f"[ims-mcp] Failed to serialize roots_result: {e}")
+            
             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]
+                # root.uri is FileUrl object, convert to string first
+                repo_names = [os.path.basename(str(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)}")
+                debug_print(f"[ims-mcp] Repository: {result}")
         else:
             if DEBUG_MODE:
                 debug_print("[ims-mcp] Client doesn't support roots capability, trying fallback")
@@ -320,14 +322,22 @@ def get_posthog_client():
         # Import PostHog (lazy import to avoid dependency if not used)
         from posthog import Posthog
         
+        # 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
         )
@@ -371,6 +381,16 @@ def track_tool_call(func: Callable) -> Callable:
             # Extract context from kwargs (FastMCP injects it)
             ctx = kwargs.get('ctx')
             
+            # Debug: serialize entire context recursively
+            if DEBUG_MODE and ctx:
+                try:
+                    ctx_dump = json.dumps(ctx.__dict__, default=str)
+                    debug_print(f"[ims-mcp] Full Context: {ctx_dump}")
+                except RecursionError:
+                    debug_print("[ims-mcp] Context has circular references")
+                except Exception as e:
+                    debug_print(f"[ims-mcp] Failed to serialize context: {e}")
+            
             # Extract user context
             username = get_username()
             repository = await get_repository_from_context(ctx) if ctx else "unknown"
@@ -386,16 +406,52 @@ def track_tool_call(func: Callable) -> Callable:
                 'username': username,
                 'repository': repository,
                 'mcp_server': 'Rosetta',
-                '$lib': 'Rosetta',
-                '$lib_version': __version__,
-                '$geoip_disable': False  # Enable GeoIP
+                'mcp_server_version': __version__
             })
+            # Note: GeoIP is enabled via disable_geoip=False in client initialization
             
-            # Add screen_name if we have document context
-            if 'title' in kwargs and kwargs['title']:
-                properties['$screen_name'] = kwargs['title']
+            # Add MCP session_id for PostHog session tracking (from debug output)
+            try:
+                if ctx and hasattr(ctx, 'request_context') and ctx.request_context:
+                    req_ctx = ctx.request_context
+                    # Check if session has an ID attribute
+                    if hasattr(req_ctx, 'session') and req_ctx.session:
+                        session = req_ctx.session
+                        # Try various possible session ID attributes
+                        session_id = (
+                            getattr(session, 'session_id', None) or
+                            getattr(session, 'id', None) or
+                            getattr(session, '_session_id', None)
+                        )
+                        if session_id:
+                            properties['$session_id'] = str(session_id)
+            except Exception:
+                pass  # Session ID not critical
+            
+            # 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']:
-                properties['$screen_name'] = f"doc:{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 = ', '.join(tags[:5])
+            elif 'filters' in kwargs and kwargs['filters']:
+                filters = kwargs['filters']
+                if isinstance(filters, dict) and filters:
+                    screen_name = ', '.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.29 → ims_mcp-1.0.31}/ims_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.29
+Version: 1.0.31
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -399,8 +399,8 @@ 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 `roots/list` protocol request, comma-separated if multiple; fallback to `client_id` parsing; 5-min cache)
-- Library: "Rosetta" with version number
-- GeoIP enabled for location tracking
+- 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.29 → ims_mcp-1.0.31}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ims-mcp"
-version = "1.0.29"
+version = "1.0.31"
 description = "Model Context Protocol server for IMS (Instruction Management Systems)"
 readme = "README.md"
 requires-python = ">=3.10"