mcpower-proxy 0.0.63__tar.gz → 0.0.65__tar.gz

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcpower-proxy
-Version: 0.0.63
+Version: 0.0.65
 Summary: MCPower Security proxy
 Author-email: MCPower Security <support@mcpower.tech>
 License:                                  Apache License

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcpower-proxy"
-version = "0.0.63"
+version = "0.0.65"
 description = "MCPower Security proxy"
 readme = "README.md"
 requires-python = "~=3.11.0"

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/src/mcpower_proxy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcpower-proxy
-Version: 0.0.63
+Version: 0.0.65
 Summary: MCPower Security proxy
 Author-email: MCPower Security <support@mcpower.tech>
 License:                                  Apache License

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/src/modules/logs/audit_trail.py

@@ -130,16 +130,20 @@ class AuditTrailLogger:
         
         This is called by the middleware after workspace roots are available.
         All queued logs will be written with app_uid as the first key.
+        Supports updating app_uid when workspace context changes.
         
         Args:
             app_uid: The application UID from workspace root
         """
-        if self.app_uid is not None:
-            self.logger.warning(f"app_uid already set to {self.app_uid}, ignoring new value {app_uid}")
+        if self.app_uid == app_uid:
             return
         
+        if self.app_uid is not None:
+            self.logger.info(f"app_uid changed from {self.app_uid} to {app_uid}")
+        else:
+            self.logger.debug(f"app_uid set to: {app_uid}")
+        
         self.app_uid = app_uid
-        self.logger.debug(f"✅ app_uid set to: {app_uid}")
         
         # Flush all pending logs
         if self._pending_logs:

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/src/wrapper/__version__.py

@@ -3,4 +3,4 @@
 Wrapper MCP Server Version
 """
 
-__version__ = "0.0.63"
+__version__ = "0.0.65"

{mcpower_proxy-0.0.63 → mcpower_proxy-0.0.65}/src/wrapper/middleware.py

@@ -55,6 +55,7 @@ class SecurityMiddleware(Middleware):
     app_id: str = ""
     _TOOLS_INIT_DEBOUNCE_SECONDS = 60
     _last_tools_init_time: Optional[float] = None
+    _last_workspace_root: Optional[str] = None
 
     def __init__(self,
                  wrapped_server_configs: dict,
@@ -69,6 +70,7 @@ class SecurityMiddleware(Middleware):
         self.logger = logger
         self.audit_logger = audit_logger
         self.app_id = ""
+        self._last_workspace_root = None
 
         self.wrapped_server_name, self.wrapped_server_transport = (
             extract_wrapped_server_info(self.wrapper_server_name, self.logger, self.wrapped_server_configs)
@@ -85,14 +87,13 @@ class SecurityMiddleware(Middleware):
     async def on_message(self, context: MiddlewareContext, call_next: CallNext) -> Any:
         self.logger.info(f"on_message: {redact(safe_json_dumps(context))}")
 
-        # Ensure app_id is set before making API calls
-        if not self.app_id:
-            workspace_roots = await self._extract_workspace_roots(context)
-            if workspace_roots:
-                self.app_id = read_app_uid(logger=self.logger, project_folder_path=workspace_roots[0])
-            else:
-                # Fallback: read app_uid from ~/.mcpower when no workspace roots
-                self.app_id = read_app_uid(logger=self.logger, project_folder_path=str(Path.home() / ".mcpower"))
+        # Check workspace roots and re-initialize app_uid if workspace changed
+        workspace_roots = await self._extract_workspace_roots(context)
+        current_workspace_root = workspace_roots[0] if workspace_roots else str(Path.home() / ".mcpower")
+        if current_workspace_root != self._last_workspace_root:
+            self.logger.debug(f"Workspace root changed from {self._last_workspace_root} to {current_workspace_root}")
+            self._last_workspace_root = current_workspace_root
+            self.app_id = read_app_uid(logger=self.logger, project_folder_path=current_workspace_root)
             self.audit_logger.set_app_uid(self.app_id)
 
         operation_type = "message"