ims-mcp 1.0.9__py3-none-any.whl → 1.0.10__py3-none-any.whl

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/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.dist-info → ims_mcp-1.0.10.dist-info}/METADATA

@@ -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.10.dist-info/RECORD

@@ -0,0 +1,9 @@
+ims_mcp/__init__.py,sha256=UDBzaafXhi3dqq6YFTXY4e08onAmmAIHZwOgGzwr-no,632
+ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
+ims_mcp/server.py,sha256=obc9TM_HyfLDg_U6tp-_6SugkvUs356wiH2MS9-ssm4,25159
+ims_mcp-1.0.10.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
+ims_mcp-1.0.10.dist-info/METADATA,sha256=7T0IiSLk0UjIky3sgT06TDazs2hz47vzinkpnP1pzI4,9379
+ims_mcp-1.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ims_mcp-1.0.10.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
+ims_mcp-1.0.10.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
+ims_mcp-1.0.10.dist-info/RECORD,,

ims_mcp-1.0.9.dist-info/RECORD

@@ -1,9 +0,0 @@
-ims_mcp/__init__.py,sha256=_a0FNt-0K9A50hDS6XdMk9mKSBodMX3vDosAcy8AdLI,631
-ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
-ims_mcp/server.py,sha256=DeObwXtOSVnolp04llFXBWjQLdMcY6-BSdsZfeJIBhU,23807
-ims_mcp-1.0.9.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
-ims_mcp-1.0.9.dist-info/METADATA,sha256=PvR-jzu9736vKBEfKNonN25dhPgfpKcmxkJ1XyIYmFA,9295
-ims_mcp-1.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ims_mcp-1.0.9.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
-ims_mcp-1.0.9.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
-ims_mcp-1.0.9.dist-info/RECORD,,