ims-mcp 1.0.9__tar.gz → 1.0.10__tar.gz

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.9
+Version: 1.0.10
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -84,6 +84,7 @@ The server automatically reads configuration from environment variables:
 | `R2R_API_KEY` | API key for authentication | None |
 | `R2R_EMAIL` | Email for authentication (requires R2R_PASSWORD) | None |
 | `R2R_PASSWORD` | Password for authentication (requires R2R_EMAIL) | None |
+| `IMS_DEBUG` | Enable debug logging to stderr (1/true/yes/on) | None (disabled) |
 
 **Authentication Priority:**
 1. If `R2R_API_KEY` is set, it will be used

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/README.md

@@ -57,6 +57,7 @@ The server automatically reads configuration from environment variables:
 | `R2R_API_KEY` | API key for authentication | None |
 | `R2R_EMAIL` | Email for authentication (requires R2R_PASSWORD) | None |
 | `R2R_PASSWORD` | Password for authentication (requires R2R_EMAIL) | None |
+| `IMS_DEBUG` | Enable debug logging to stderr (1/true/yes/on) | None (disabled) |
 
 **Authentication Priority:**
 1. If `R2R_API_KEY` is set, it will be used

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/ims_mcp/__init__.py

@@ -11,7 +11,7 @@ Environment Variables:
 Note: Environment variables use R2R_ prefix for compatibility with underlying R2R SDK.
 """
 
-__version__ = "1.0.9"
+__version__ = "1.0.10"
 __author__ = "Igor Solomatov"
 
 from ims_mcp.server import mcp

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/ims_mcp/server.py

@@ -15,15 +15,60 @@ configuration is needed when running via uvx or other launchers.
 """
 
 import functools
+import logging
 import os
+import signal
 import sys
 import uuid
 from r2r import R2RClient, R2RException
 
+# Debug mode controlled by environment variable
+DEBUG_MODE = os.getenv('IMS_DEBUG', '').lower() in ('1', 'true', 'yes', 'on')
+
+# Configure logging based on debug mode
+if DEBUG_MODE:
+    logging.basicConfig(level=logging.DEBUG)
+else:
+    # Suppress all logging output from R2R and other libraries
+    logging.basicConfig(level=logging.CRITICAL)
+    # Specifically suppress httpx and httpcore which R2R uses
+    logging.getLogger('httpx').setLevel(logging.CRITICAL)
+    logging.getLogger('httpcore').setLevel(logging.CRITICAL)
+    logging.getLogger('r2r').setLevel(logging.CRITICAL)
+
 # Global client instance with authentication
 _authenticated_client = None
 
 
+def debug_print(msg: str):
+    """Print debug message to stderr if debug mode enabled."""
+    if DEBUG_MODE:
+        print(msg, file=sys.stderr)
+        sys.stderr.flush()
+
+
+def cleanup_and_exit(signum=None, frame=None):
+    """Gracefully shutdown the server on termination signals."""
+    global _authenticated_client
+    
+    debug_print(f"[ims-mcp] Shutting down gracefully...")
+    
+    # Cleanup authenticated client if exists
+    if _authenticated_client is not None:
+        try:
+            # R2R client cleanup if needed
+            _authenticated_client = None
+        except Exception:
+            pass  # Ignore errors during shutdown
+    
+    debug_print(f"[ims-mcp] Shutdown complete")
+    sys.exit(0)
+
+
+# Register signal handlers for graceful shutdown
+signal.signal(signal.SIGTERM, cleanup_and_exit)
+signal.signal(signal.SIGINT, cleanup_and_exit)
+
 def get_authenticated_client() -> R2RClient:
     """Get or create an authenticated R2R client.
     
@@ -48,13 +93,12 @@ def get_authenticated_client() -> R2RClient:
     email = os.getenv("R2R_EMAIL")
     password = os.getenv("R2R_PASSWORD")
     
-    print(f"[ims-mcp v{__version__}]", file=sys.stderr)
-    print(f"  server={base_url}", file=sys.stderr)
-    print(f"  collection={collection}", file=sys.stderr)
-    print(f"  api_key={api_key[:3] + '...' if api_key else 'none'}", file=sys.stderr)
-    print(f"  email={email if email else 'none'}", file=sys.stderr)
-    print(f"  password={password[:3] + '...' if password else 'none'}", file=sys.stderr)
-    sys.stderr.flush()
+    debug_print(f"[ims-mcp v{__version__}]")
+    debug_print(f"  server={base_url}")
+    debug_print(f"  collection={collection}")
+    debug_print(f"  api_key={api_key[:3] + '...' if api_key else 'none'}")
+    debug_print(f"  email={email if email else 'none'}")
+    debug_print(f"  password={password[:3] + '...' if password else 'none'}")
     
     # Create new client
     client = R2RClient()
@@ -67,9 +111,9 @@ def get_authenticated_client() -> R2RClient:
         try:
             # Login - R2RClient automatically handles token internally
             client.users.login(email=email, password=password)
-            print(f"[ims-mcp] Login successful", file=sys.stderr)
+            debug_print(f"[ims-mcp] Login successful")
         except Exception as e:
-            print(f"[ims-mcp] Login failed: {e}", file=sys.stderr)
+            debug_print(f"[ims-mcp] Login failed: {e}")
             # If login fails, continue without authentication (might work for local servers)
             pass
     
@@ -97,7 +141,7 @@ def retry_on_auth_error(func):
         except R2RException as e:
             # Check if this is an authentication error (token expired)
             if hasattr(e, 'status_code') and e.status_code in [401, 403]:
-                print(f"[ims-mcp] Token expired, re-authenticating...", file=sys.stderr)
+                debug_print(f"[ims-mcp] Token expired, re-authenticating...")
                 invalidate_client()
                 # Retry once with fresh authentication
                 return await func(*args, **kwargs)

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/ims_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.9
+Version: 1.0.10
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -84,6 +84,7 @@ The server automatically reads configuration from environment variables:
 | `R2R_API_KEY` | API key for authentication | None |
 | `R2R_EMAIL` | Email for authentication (requires R2R_PASSWORD) | None |
 | `R2R_PASSWORD` | Password for authentication (requires R2R_EMAIL) | None |
+| `IMS_DEBUG` | Enable debug logging to stderr (1/true/yes/on) | None (disabled) |
 
 **Authentication Priority:**
 1. If `R2R_API_KEY` is set, it will be used

{ims_mcp-1.0.9 → ims_mcp-1.0.10}/pyproject.toml

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