ims-mcp 1.0.3__py3-none-any.whl → 1.0.5__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.3"
+__version__ = "1.0.5"
 __author__ = "Igor Solomatov"
 
 from ims_mcp.server import mcp

ims_mcp/server.py

@@ -14,10 +14,11 @@ The R2RClient automatically reads these environment variables, so no manual
 configuration is needed when running via uvx or other launchers.
 """
 
+import functools
 import os
 import sys
 import uuid
-from r2r import R2RClient
+from r2r import R2RClient, R2RException
 
 # Global client instance with authentication
 _authenticated_client = None
@@ -76,6 +77,34 @@ def get_authenticated_client() -> R2RClient:
     return client
 
 
+def retry_on_auth_error(func):
+    """Decorator to handle token expiry and automatically re-authenticate.
+    
+    When R2R server restarts, all authentication tokens are invalidated.
+    This decorator catches 401/403 errors, invalidates the cached client,
+    re-authenticates, and retries the request once.
+    """
+    def invalidate_client():
+        """Invalidate cached client to force re-authentication."""
+        global _authenticated_client
+        _authenticated_client = None
+    
+    @functools.wraps(func)
+    async def wrapper(*args, **kwargs):
+        try:
+            return await func(*args, **kwargs)
+        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)
+                invalidate_client()
+                # Retry once with fresh authentication
+                return await func(*args, **kwargs)
+            # Re-raise non-auth errors
+            raise
+    return wrapper
+
+
 def id_to_shorthand(id: str) -> str:
     """Convert a full ID to shortened version for display."""
     return str(id)[:7]
@@ -167,6 +196,7 @@ except Exception as e:
 
 # Search tool with filtering support
 @mcp.tool()
+@retry_on_auth_error
 async def search(
     query: str,
     filters: dict | None = None,
@@ -216,6 +246,7 @@ async def search(
 
 # RAG query tool with filtering and generation config
 @mcp.tool()
+@retry_on_auth_error
 async def rag(
     query: str,
     filters: dict | None = None,
@@ -270,6 +301,7 @@ async def rag(
 
 # Document upload tool with upsert semantics
 @mcp.tool()
+@retry_on_auth_error
 async def put_document(
     content: str,
     title: str,
@@ -337,11 +369,12 @@ async def put_document(
 
 # List documents tool
 @mcp.tool()
+@retry_on_auth_error
 async def list_documents(
     offset: float = 0,  # Use float to accept JSON "number" type, convert to int internally
     limit: float = 100,  # Use float to accept JSON "number" type, convert to int internally
     document_ids: list[str] | None = None,
-    compact_view: bool = False,
+    compact_view: bool = True,
     tags: list[str] | None = None,
     match_all_tags: bool = False,
 ) -> str:
@@ -352,7 +385,7 @@ async def list_documents(
         offset: Number of documents to skip (default: 0)
         limit: Maximum number of documents to return (default: 100, max: 100)
         document_ids: Optional list of specific document IDs to retrieve
-        compact_view: Show only ID and title (default: False - shows all details)
+        compact_view: Show only ID and title (default: True - compact view)
         tags: Optional list of tags to filter by (e.g., ["agents", "r1"])
         match_all_tags: If True, document must have ALL tags; if False (default), document must have ANY tag
 
@@ -440,6 +473,7 @@ async def list_documents(
 
 # Get document tool
 @mcp.tool()
+@retry_on_auth_error
 async def get_document(
     document_id: str | None = None,
     title: str | None = None,
@@ -530,6 +564,7 @@ async def get_document(
 
 # Delete document tool
 @mcp.tool()
+@retry_on_auth_error
 async def delete_document(document_id: str) -> str:
     """
     Delete a document by ID

{ims_mcp-1.0.3.dist-info → ims_mcp-1.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ims-mcp
-Version: 1.0.3
+Version: 1.0.5
 Summary: Model Context Protocol server for IMS (Instruction Management Systems)
 Author: Igor Solomatov
 License-Expression: MIT
@@ -229,13 +229,16 @@ List documents with pagination and optional tag filtering.
 - `offset` (int, optional): Documents to skip (default: 0)
 - `limit` (int, optional): Max documents (default: 100)
 - `document_ids` (list[str], optional): Specific IDs to retrieve
-- `compact_view` (bool, optional): Show only ID and title (default: False)
+- `compact_view` (bool, optional): Show only ID and title (default: True)
 - `tags` (list[str], optional): Filter by tags (e.g., `["agents", "r1"]`)
 - `match_all_tags` (bool, optional): If True, document must have ALL tags; if False (default), document must have ANY tag
 
 **Examples:**
 ```python
-# List all documents
+# List all documents (compact view - ID and title only)
+list_documents(offset=0, limit=10)
+
+# List with full details
 list_documents(offset=0, limit=10, compact_view=False)
 
 # Filter by tags (ANY mode - documents with "research" OR "ml")

ims_mcp-1.0.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+ims_mcp/__init__.py,sha256=jqYWnp0t_xvHX7m0FMHJNOO_3KongXmW-1zwKArilc8,631
+ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
+ims_mcp/server.py,sha256=lHdhhVeohXMX9KhqX4mYlHmdOkEH-Ee_k3e7Rg2uNBY,22999
+ims_mcp-1.0.5.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
+ims_mcp-1.0.5.dist-info/METADATA,sha256=GYmWYRXsTNf6ZPq0a6_l7epsERXVLfxuQB3uESR9GL4,9295
+ims_mcp-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ims_mcp-1.0.5.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
+ims_mcp-1.0.5.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
+ims_mcp-1.0.5.dist-info/RECORD,,

ims_mcp-1.0.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-ims_mcp/__init__.py,sha256=IXSdAG7JlNgU0FZHnI56CEvjhIHzAc3cxuuEINXfbRQ,631
-ims_mcp/__main__.py,sha256=z4P1aCVfOgS3cTM2wgJd2pxjMmKCkGkiqYDRGgrspxw,191
-ims_mcp/server.py,sha256=g_MwgGhVN9jv03qKYTLWNRz_Q-7_cMl9zmbCaVh6LNI,21741
-ims_mcp-1.0.3.dist-info/licenses/LICENSE,sha256=4d1dlH04mbnN3ya4lybcVOUwljRHGy-aSc9MYqGYW44,2534
-ims_mcp-1.0.3.dist-info/METADATA,sha256=TuJNag9JatA25mo5PWAm4hAfsn6FtSJ0MH1UcZyOCv4,9200
-ims_mcp-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ims_mcp-1.0.3.dist-info/entry_points.txt,sha256=xCH9I8g1pTTEqrfjnE-ANHaZo4W6EBJVy0Lg5z8SaIQ,48
-ims_mcp-1.0.3.dist-info/top_level.txt,sha256=wEXA33qFr_eov3S1PY2OF6EQBA2rtAWB_ZNJOzNNQuM,8
-ims_mcp-1.0.3.dist-info/RECORD,,