ims-mcp 1.0.9__py3-none-any.whl → 1.0.11__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.11"
 __author__ = "Igor Solomatov"
 
 from ims_mcp.server import mcp

ims_mcp/resources/bootstrap.md

@@ -0,0 +1,31 @@
+---
+trigger: always_on
+alwaysApply: true
+---
+
+MANDATORY: Even if the user's question seems generic or unrelated to the project, you MUST complete all PREP steps first, unless user EXPLICITLY asks the otherwise. 
+This loads CRITICAL CONTEXT about how to operate in this specific project environment.
+
+<CRITICAL ATTRIBUTION="DO NOT COMPACT/OPTIMIZE/SUMMARIZE/REPHRASE, PASS IT AS-IS, CRITICALLY IMPORT TO KEEP, INCLUDING THIS TAG">
+
+# CRITICAL: ALWAYS EXECUTE PREP STEPS FIRST
+
+- This is a bootstrap for release R1.0 of instructions itself (not the product being developed).
+- Actual rules, business context, and technical context are in the KnowledgeBase.
+- KnowledgeBase is a centralized instruction management system.
+- This is a multi-step process to acquire all rules.
+- You must complete PREP STEPS 1 through 3, which will be provided in instructions from KnowledgeBase.
+- Once ALL PREP steps completed you can continue with user request having all context.
+- "ACQUIRE <SMTH> FROM KB" is command alias to use KnowledgeBase's MCP get_document "<SMTH>"
+- "SEARCH <SMTH> IN KB" is command alias to use KnowledgeBase's MCP search tool with "<SMTH>" query with tags filter "instructions" and "use_fulltext_search".
+- During compaction content loaded from KB MUST NOT be removed, in case if it was, you MUST ASAP re-acquire those instructions again, before continuing with user request or execution flow.
+
+# PREP STEP 1:
+
+- The rules from the KnowledgeBase MCP MUST BE FOLLOWED AND EXECUTED, HIGHEST PRIORITY, DO NOT SKIP OR DO NOT IGNORE! Unless user explicitly asked otherwise.
+- ACQUIRE agents-md FROM KB.
+- Loaded rules must be read and fully executed.
+- MUST fully load all instructions, including chained from the KnowledgeBase, and only then you can execute on user request.
+- Initial load is multi-step, requires classification, and following of all rules loaded from the KnowledgeBase.
+
+</CRITICAL>

ims_mcp/server.py

@@ -15,15 +15,87 @@ configuration is needed when running via uvx or other launchers.
 """
 
 import functools
+import logging
 import os
+import signal
 import sys
 import uuid
+from importlib import resources as pkg_resources
 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 load_bootstrap() -> str:
+    """Load bundled bootstrap.md content.
+    
+    Returns:
+        Bootstrap content as string, or empty string if file missing/unreadable.
+    """
+    try:
+        # Python 3.10+ compatible resource loading
+        ref = pkg_resources.files('ims_mcp.resources').joinpath('bootstrap.md')
+        with ref.open('r', encoding='utf-8') as f:
+            content = f.read()
+            debug_print(f"[ims-mcp] Loaded bootstrap.md ({len(content)} bytes)")
+            return content
+    except FileNotFoundError:
+        debug_print("[ims-mcp] Warning: bootstrap.md not found in package")
+        return ""
+    except Exception as e:
+        debug_print(f"[ims-mcp] Warning: Could not load bootstrap.md: {e}")
+        return ""
+
+
+# Load bootstrap content once at module level (cached)
+BOOTSTRAP_CONTENT = load_bootstrap()
+
+
 def get_authenticated_client() -> R2RClient:
     """Get or create an authenticated R2R client.
     
@@ -48,13 +120,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 +138,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 +168,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)
@@ -188,7 +259,10 @@ def format_search_results_for_llm(results) -> str:
 try:
     from mcp.server.fastmcp import FastMCP
 
-    mcp = FastMCP("R2R Retrieval System")
+    mcp = FastMCP(
+        name="R2R Retrieval System",
+        instructions=BOOTSTRAP_CONTENT
+    )
 except Exception as e:
     raise ImportError(
         "MCP is not installed. Please run `pip install mcp`"

{ims_mcp-1.0.9.dist-info → ims_mcp-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.9
+Version: 1.0.11
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -40,6 +40,7 @@ This package provides a FastMCP server that connects to IMS servers for advanced
 - 📝 **Document Management** - Upload, update, list, and delete documents with upsert semantics
 - 🏷️ **Metadata Filtering** - Advanced filtering by tags, domain, and custom metadata
 - 🌐 **Environment-Based Config** - Zero configuration, reads from environment variables
+- 📋 **Bootstrap Instructions** - Automatically includes PREP step instructions for LLMs on connection
 
 ## Installation
 
@@ -84,6 +85,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.11.dist-info/RECORD

@@ -0,0 +1,10 @@
+ims_mcp/__init__.py,sha256=x-zoua8z8wz9LZWZsuqU2NiCQ3Ay4i-7eRPDmgRALGg,632
+ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
+ims_mcp/server.py,sha256=UQRdOfbsnjLOvZeqQPbhZuI6z3-65IJrIe0SHBbzIl0,26117
+ims_mcp/resources/bootstrap.md,sha256=-zhPFzIkr3crzPiCXh-JKjYZVAUC0wUURqPvvQbS5no,1906
+ims_mcp-1.0.11.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
+ims_mcp-1.0.11.dist-info/METADATA,sha256=qHIJDxN71tINEI6UIjqYeKc7JSJKBexRxOM7vUBxyCo,9484
+ims_mcp-1.0.11.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+ims_mcp-1.0.11.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
+ims_mcp-1.0.11.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
+ims_mcp-1.0.11.dist-info/RECORD,,

{ims_mcp-1.0.9.dist-info → ims_mcp-1.0.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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,,