signalwire-agents 1.0.10__py3-none-any.whl → 1.0.11__py3-none-any.whl

signalwire_agents/mcp_gateway/session_manager.py

@@ -0,0 +1,218 @@
+#!/usr/bin/env python3
+"""
+Copyright (c) 2025 SignalWire
+
+This file is part of the SignalWire AI Agents SDK.
+
+Licensed under the MIT License.
+See LICENSE file in the project root for full license information.
+"""
+
+"""
+Session Manager for MCP Gateway
+
+Manages lifecycle of MCP server sessions tied to SignalWire call IDs.
+Handles timeouts, cleanup, and resource limits.
+"""
+
+import threading
+import time
+import logging
+from typing import Dict, Optional, Any
+from datetime import datetime, timedelta
+from dataclasses import dataclass, field
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass
+class Session:
+    """Represents an active MCP session"""
+    session_id: str
+    service_name: str
+    process: Any  # MCPClient instance
+    created_at: datetime = field(default_factory=datetime.now)
+    last_accessed: datetime = field(default_factory=datetime.now)
+    timeout: int = 300  # seconds
+    metadata: Dict[str, Any] = field(default_factory=dict)
+    
+    @property
+    def is_expired(self) -> bool:
+        """Check if session has expired based on timeout"""
+        return datetime.now() > self.last_accessed + timedelta(seconds=self.timeout)
+    
+    @property
+    def is_alive(self) -> bool:
+        """Check if the underlying MCP client is still running"""
+        return self.process and self.process.process and self.process.process.poll() is None
+    
+    def touch(self):
+        """Update last accessed time"""
+        self.last_accessed = datetime.now()
+
+
+class SessionManager:
+    """Manages MCP server sessions with automatic cleanup"""
+    
+    def __init__(self, config: Dict[str, Any]):
+        self.config = config
+        self.sessions: Dict[str, Session] = {}
+        self.lock = threading.RLock()
+        self.cleanup_interval = config.get('session', {}).get('cleanup_interval', 60)
+        self.max_sessions_per_service = config.get('session', {}).get('max_sessions_per_service', 100)
+        self.default_timeout = config.get('session', {}).get('default_timeout', 300)
+        self._shutdown = threading.Event()
+        
+        # Start cleanup thread
+        self.cleanup_thread = threading.Thread(target=self._cleanup_loop, daemon=True)
+        self.cleanup_thread.start()
+        
+        logger.info(f"SessionManager initialized with cleanup_interval={self.cleanup_interval}s")
+    
+    def create_session(self, session_id: str, service_name: str, process: Any,
+                      timeout: Optional[int] = None, metadata: Optional[Dict[str, Any]] = None) -> Session:
+        """Create and register a new session"""
+        with self.lock:
+            # Check if session already exists
+            if session_id in self.sessions:
+                logger.warning(f"Session {session_id} already exists, closing old session")
+                self.close_session(session_id)
+            
+            # Check service limits
+            service_count = sum(1 for s in self.sessions.values() if s.service_name == service_name)
+            if service_count >= self.max_sessions_per_service:
+                raise RuntimeError(f"Max sessions limit reached for service {service_name}")
+            
+            # Create new session
+            session = Session(
+                session_id=session_id,
+                service_name=service_name,
+                process=process,
+                timeout=timeout or self.default_timeout,
+                metadata=metadata or {}
+            )
+            
+            self.sessions[session_id] = session
+            logger.info(f"Created session {session_id} for service {service_name}")
+            
+            return session
+    
+    def get_session(self, session_id: str) -> Optional[Session]:
+        """Get an active session by ID"""
+        with self.lock:
+            session = self.sessions.get(session_id)
+            
+            if session:
+                if not session.is_alive:
+                    logger.warning(f"Session {session_id} process is dead, removing")
+                    self.close_session(session_id)
+                    return None
+                
+                if session.is_expired:
+                    logger.info(f"Session {session_id} has expired, removing")
+                    self.close_session(session_id)
+                    return None
+                
+                # Update last accessed time
+                session.touch()
+                
+            return session
+    
+    def close_session(self, session_id: str) -> bool:
+        """Close and remove a session"""
+        with self.lock:
+            session = self.sessions.pop(session_id, None)
+            
+            if not session:
+                logger.warning(f"Attempted to close non-existent session {session_id}")
+                return False
+            
+            # Terminate the MCP client
+            if session.process:
+                try:
+                    # Stop the MCP client
+                    session.process.stop()
+                except Exception as e:
+                    logger.error(f"Error stopping MCP client for session {session_id}: {e}")
+            
+            logger.info(f"Closed session {session_id}")
+            return True
+    
+    def list_sessions(self) -> Dict[str, Dict[str, Any]]:
+        """List all active sessions with their info"""
+        with self.lock:
+            result = {}
+            
+            for session_id, session in list(self.sessions.items()):
+                # Check if still valid
+                if not session.is_alive or session.is_expired:
+                    self.close_session(session_id)
+                    continue
+                
+                result[session_id] = {
+                    'service_name': session.service_name,
+                    'created_at': session.created_at.isoformat(),
+                    'last_accessed': session.last_accessed.isoformat(),
+                    'timeout': session.timeout,
+                    'metadata': session.metadata,
+                    'time_remaining': max(0, (session.last_accessed + timedelta(seconds=session.timeout) - datetime.now()).total_seconds())
+                }
+            
+            return result
+    
+    def get_service_session_count(self, service_name: str) -> int:
+        """Get number of active sessions for a service"""
+        with self.lock:
+            return sum(1 for s in self.sessions.values() 
+                      if s.service_name == service_name and s.is_alive and not s.is_expired)
+    
+    def _cleanup_loop(self):
+        """Background thread that cleans up expired sessions"""
+        logger.info("Session cleanup thread started")
+        
+        while not self._shutdown.is_set():
+            try:
+                # Wait with timeout so we can check shutdown flag
+                if self._shutdown.wait(timeout=self.cleanup_interval):
+                    break
+                
+                with self.lock:
+                    expired_sessions = []
+                    
+                    for session_id, session in self.sessions.items():
+                        if session.is_expired or not session.is_alive:
+                            expired_sessions.append(session_id)
+                    
+                    for session_id in expired_sessions:
+                        logger.info(f"Cleaning up expired session {session_id}")
+                        self.close_session(session_id)
+                    
+                    if expired_sessions:
+                        logger.info(f"Cleaned up {len(expired_sessions)} expired sessions")
+                        
+            except Exception as e:
+                logger.error(f"Error in cleanup thread: {e}")
+        
+        logger.info("Session cleanup thread stopped")
+    
+    def shutdown(self):
+        """Shutdown all sessions and cleanup"""
+        logger.info("Shutting down SessionManager")
+        
+        # Signal cleanup thread to stop
+        self._shutdown.set()
+        
+        with self.lock:
+            # Close all sessions
+            session_ids = list(self.sessions.keys())
+            logger.info(f"Closing {len(session_ids)} active sessions")
+            for session_id in session_ids:
+                self.close_session(session_id)
+        
+        # Wait for cleanup thread to finish (with timeout)
+        if self.cleanup_thread.is_alive():
+            self.cleanup_thread.join(timeout=2.0)
+            if self.cleanup_thread.is_alive():
+                logger.warning("Cleanup thread did not stop gracefully")
+        
+        logger.info("SessionManager shutdown complete")

{signalwire_agents-1.0.10.data → signalwire_agents-1.0.11.data}/data/share/man/man1/sw-agent-init.1

@@ -1,7 +1,7 @@
 .\" Man page for sw-agent-init
 .\" Copyright (c) 2025 SignalWire
 .\" Licensed under the MIT License
-.TH SW-AGENT-INIT 1 "November 2025" "SignalWire Agents SDK 1.0.6" "SignalWire Commands"
+.TH SW-AGENT-INIT 1 "November 2025" "SignalWire Agents SDK 1.0.8" "SignalWire Commands"
 .SH NAME
 sw-agent-init \- create a new SignalWire AI agent project
 .SH SYNOPSIS
@@ -14,14 +14,14 @@ sw-agent-init \- create a new SignalWire AI agent project
 .SH DESCRIPTION
 .B sw-agent-init
 is an interactive command-line tool for creating new SignalWire AI agent
-projects. It generates a complete project structure with all the necessary
-files to get started building voice AI agents.
+projects. It supports both local agent projects (FastAPI/uvicorn) and
+cloud function deployments (AWS Lambda, Google Cloud Functions, Azure Functions).
 .PP
 When run without arguments, the tool enters interactive mode and prompts
-for project configuration. When given a project name, it runs in quick
-mode with sensible defaults.
+for project configuration including platform selection. When given a project
+name, it runs in quick mode with sensible defaults (local platform by default).
 .PP
-The generated project includes:
+For local projects, the generated project includes:
 .IP \(bu 2
 Agent module with sample SWAIG function
 .IP \(bu 2
@@ -46,8 +46,36 @@ interactive mode and prompts for a name.
 .BR \-h ", " \-\-help
 Display help message and exit.
 .TP
+.BI \-p ", " \-\-platform " PLATFORM"
+Target deployment platform. Valid values:
+.RS
+.IP \(bu 2
+.B local
+\- Local agent with FastAPI/uvicorn server (default)
+.IP \(bu 2
+.B aws
+\- AWS Lambda function with API Gateway
+.IP \(bu 2
+.B gcp
+\- Google Cloud Function (Gen 2)
+.IP \(bu 2
+.B azure
+\- Azure Function
+.RE
+.TP
+.BI \-r ", " \-\-region " REGION"
+Cloud region for deployment (only for cloud platforms). Default regions:
+.RS
+.IP \(bu 2
+AWS: us-east-1
+.IP \(bu 2
+GCP: us-central1
+.IP \(bu 2
+Azure: eastus
+.RE
+.TP
 .BI \-\-type " TYPE"
-Agent type to create. Valid values:
+Agent type to create (only for local platform). Valid values:
 .RS
 .IP \(bu 2
 .B basic
@@ -58,8 +86,8 @@ Agent type to create. Valid values:
 .RE
 .TP
 .B \-\-no\-venv
-Skip virtual environment creation. By default, a Python virtual
-environment is created in the project directory.
+Skip virtual environment creation (only for local platform). By default,
+a Python virtual environment is created in the project directory.
 .TP
 .BI \-\-dir " DIRECTORY"
 Parent directory for the project. The project will be created as
@@ -68,14 +96,20 @@ a subdirectory. Default is the current directory.
 When run without a project name, the tool prompts for:
 .SS Project Configuration
 .IP \(bu 2
+.B Target platform
+\- Local, AWS Lambda, GCP, or Azure
+.IP \(bu 2
 .B Project name
 \- Name of the project directory
 .IP \(bu 2
 .B Project directory
 \- Where to create the project
 .IP \(bu 2
+.B Region
+\- Cloud region (for cloud platforms only)
+.IP \(bu 2
 .B Agent type
-\- Basic or full setup
+\- Basic or full setup (for local platform only)
 .SS Feature Selection
 The following features can be enabled or disabled:
 .IP \(bu 2
@@ -189,8 +223,8 @@ Run without arguments for guided setup:
 sw-agent-init
 .fi
 .RE
-.SS Quick Mode
-Create a basic agent project:
+.SS Quick Mode - Local
+Create a basic local agent project:
 .PP
 .RS
 .nf
@@ -198,7 +232,7 @@ sw-agent-init myagent
 .fi
 .RE
 .PP
-Create a full-featured project:
+Create a full-featured local project:
 .PP
 .RS
 .nf
@@ -221,7 +255,39 @@ Create in a specific directory:
 sw-agent-init myagent --dir /opt/projects
 .fi
 .RE
-.SS After Creation
+.SS Quick Mode - Cloud Functions
+Create an AWS Lambda project:
+.PP
+.RS
+.nf
+sw-agent-init myagent -p aws
+.fi
+.RE
+.PP
+Create an AWS Lambda project in a specific region:
+.PP
+.RS
+.nf
+sw-agent-init myagent -p aws -r us-west-2
+.fi
+.RE
+.PP
+Create a Google Cloud Function project:
+.PP
+.RS
+.nf
+sw-agent-init myagent -p gcp
+.fi
+.RE
+.PP
+Create an Azure Function project:
+.PP
+.RS
+.nf
+sw-agent-init myagent -p azure
+.fi
+.RE
+.SS After Creation - Local
 Start the agent:
 .PP
 .RS
@@ -251,6 +317,33 @@ swaig-test agents/main_agent.py --exec get_info --topic "test"
 swaig-test agents/main_agent.py --dump-swml
 .fi
 .RE
+.SS After Creation - Cloud Functions
+Deploy to AWS Lambda:
+.PP
+.RS
+.nf
+cd myagent
+./deploy.sh
+.fi
+.RE
+.PP
+Deploy to Google Cloud:
+.PP
+.RS
+.nf
+cd myagent
+./deploy.sh
+.fi
+.RE
+.PP
+Deploy to Azure:
+.PP
+.RS
+.nf
+cd myagent
+./deploy.sh
+.fi
+.RE
 .SH AGENT TYPES
 .SS Basic Agent
 The basic agent type includes:

{signalwire_agents-1.0.10.dist-info → signalwire_agents-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: signalwire_agents
-Version: 1.0.10
+Version: 1.0.11
 Summary: SignalWire AI Agents SDK
 Author-email: SignalWire Team <info@signalwire.com>
 License: MIT
@@ -75,6 +75,9 @@ Requires-Dist: pgvector>=0.2.0; extra == "search-all"
 Provides-Extra: pgvector
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "pgvector"
 Requires-Dist: pgvector>=0.2.0; extra == "pgvector"
+Provides-Extra: mcp-gateway
+Requires-Dist: flask>=2.0.0; extra == "mcp-gateway"
+Requires-Dist: flask-limiter>=3.5.0; extra == "mcp-gateway"
 Provides-Extra: all
 Requires-Dist: sentence-transformers>=2.2.0; extra == "all"
 Requires-Dist: scikit-learn>=1.3.0; extra == "all"

{signalwire_agents-1.0.10.dist-info → signalwire_agents-1.0.11.dist-info}/RECORD

@@ -1,11 +1,11 @@
-signalwire_agents/__init__.py,sha256=ayt9AO0brsom1nxCluBItWnJC9Qv_kwaTknpL6YPsQE,5031
+signalwire_agents/__init__.py,sha256=gW38F5mpUdBDy9opRa56Adm8kHMib6gvWd_4zfmNW7s,5031
 signalwire_agents/agent_server.py,sha256=5GPt6XekVxep9esZY9s4MG6sOR-VF_m53U25r3WoML4,30964
 signalwire_agents/schema.json,sha256=YQv4-KiegE00XvxoLMKAml6aCGitnt3kBq31ECxTHK8,385886
 signalwire_agents/agents/bedrock.py,sha256=J582gooNtxtep4xdVOfyDzRtHp_XrurPMS93xf2Xod0,10836
 signalwire_agents/cli/__init__.py,sha256=XbxAQFaCIdGXIXJiriVBWoFPOJsC401u21588nO4TG8,388
 signalwire_agents/cli/build_search.py,sha256=wDuonVY9fcqspThRHmbArTuZLKsiStu-ozVExtqawA8,54937
 signalwire_agents/cli/config.py,sha256=2i4e0BArdKsaXxjeueYYRNke7GWicHPYC2wuitVrP7A,2541
-signalwire_agents/cli/init_project.py,sha256=A7Qx7UoMOG1fdWWyn4ceH6oc5BV6jVAxudPrtOVBBjM,36396
+signalwire_agents/cli/init_project.py,sha256=nNBPni3x1xzdJ5pAZsQRUYK1t1e64NzwUUkq-Uc3mzo,78835
 signalwire_agents/cli/swaig_test_wrapper.py,sha256=t63HQpEc1Up5AcysEHP1OsEQcgSMKH-9H1L2IhFso18,1533
 signalwire_agents/cli/test_swaig.py,sha256=-v-XjTUWZNxmMJuOF5_cB1Jz8x8emJoqgqS_8jLeT4Y,31487
 signalwire_agents/cli/types.py,sha256=U8Abc4Atb5-wMbhM3MjcuIXsbONLOu1ucePWOCgqdco,1810
@@ -64,6 +64,10 @@ signalwire_agents/core/mixins/tool_mixin.py,sha256=6CaNdaspHcfte0qSB_bSN8PTsqxRZ
 signalwire_agents/core/mixins/web_mixin.py,sha256=FvllyWjdms5M8i37hCx9WLbeqJdd84Uymqb0aI0c6BU,50639
 signalwire_agents/core/security/__init__.py,sha256=4Mr7baQ_xR_hfJ72YxQRAT_GFa663YjFX_PumJ35Xds,191
 signalwire_agents/core/security/session_manager.py,sha256=s5hXYcFnrsYFoyo-zcN7EJy-wInZQI_cWTBHX9MxHR4,9164
+signalwire_agents/mcp_gateway/__init__.py,sha256=uZhyt0cBa9Wk_OPr_Ewphh6apsaf_5O9G48NVkm_00I,619
+signalwire_agents/mcp_gateway/gateway_service.py,sha256=hytoEhiXSyZxk_LxV-zo0mfCCAJf-y2Gs5gJ__iIRB0,23081
+signalwire_agents/mcp_gateway/mcp_manager.py,sha256=DaR49BX5b6hn2mXJygS3idNEr3IDImWGNdsrQSYxiEE,19669
+signalwire_agents/mcp_gateway/session_manager.py,sha256=sGMPT8aSzS2RRR_ewkd0nxwYI-LK5CaWrendGhXAW9c,8505
 signalwire_agents/prefabs/__init__.py,sha256=MW11J63XH7KxF2MWguRsMFM9iqMWexaEO9ynDPL_PDM,715
 signalwire_agents/prefabs/concierge.py,sha256=XB-Ziy0tteYI5NHu7Sinx-N960nzkhq2m4v8Ei8UHas,9921
 signalwire_agents/prefabs/faq_bot.py,sha256=cUuHhnDB8S4aVg-DiQe4jBmCAPrYQrND_Mff9iaeEa0,10572
@@ -131,12 +135,12 @@ signalwire_agents/utils/token_generators.py,sha256=4Mr7baQ_xR_hfJ72YxQRAT_GFa663
 signalwire_agents/utils/validators.py,sha256=4Mr7baQ_xR_hfJ72YxQRAT_GFa663YjFX_PumJ35Xds,191
 signalwire_agents/web/__init__.py,sha256=XE_pSTY9Aalzr7J7wqFth1Zr3cccQHPPcF5HWNrOpz8,383
 signalwire_agents/web/web_service.py,sha256=a2PSHJgX1tlZr0Iz1A1UouZjXEePJAZL632evvLVM38,21071
-signalwire_agents-1.0.10.data/data/share/man/man1/sw-agent-init.1,sha256=EFE8iOj2aGrZRwSXRoZdrZTZXOROuQmuIrZuzp9yI2s,6880
-signalwire_agents-1.0.10.data/data/share/man/man1/sw-search.1,sha256=9jJ6V6t6DgmXByz8Lw9exjf683Cw3sJGro8-eB0M9EY,10413
-signalwire_agents-1.0.10.data/data/share/man/man1/swaig-test.1,sha256=Ri0EITo8YMFowkcYltwPSwU4VJdRzo7XTWloi5WddCg,7815
-signalwire_agents-1.0.10.dist-info/licenses/LICENSE,sha256=NYvAsB-rTcSvG9cqHt9EUHAWLiA9YzM4Qfz-mPdvDR0,1067
-signalwire_agents-1.0.10.dist-info/METADATA,sha256=WetB2a73mHJZC_2t_qiZIqgVxyp8v8CC2adEl25TRzw,41600
-signalwire_agents-1.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-signalwire_agents-1.0.10.dist-info/entry_points.txt,sha256=GDaOxm8I9Bf3Sd2jDSGRXDG2ezDMgmX6ockjpDfK_IY,200
-signalwire_agents-1.0.10.dist-info/top_level.txt,sha256=kDGS6ZYv84K9P5Kyg9_S8P_pbUXoHkso0On_DB5bbWc,18
-signalwire_agents-1.0.10.dist-info/RECORD,,
+signalwire_agents-1.0.11.data/data/share/man/man1/sw-agent-init.1,sha256=J4k5Oi74BnWCPCvsaw00vuyyqDPuIECjJIPu5OynvJc,8381
+signalwire_agents-1.0.11.data/data/share/man/man1/sw-search.1,sha256=9jJ6V6t6DgmXByz8Lw9exjf683Cw3sJGro8-eB0M9EY,10413
+signalwire_agents-1.0.11.data/data/share/man/man1/swaig-test.1,sha256=Ri0EITo8YMFowkcYltwPSwU4VJdRzo7XTWloi5WddCg,7815
+signalwire_agents-1.0.11.dist-info/licenses/LICENSE,sha256=NYvAsB-rTcSvG9cqHt9EUHAWLiA9YzM4Qfz-mPdvDR0,1067
+signalwire_agents-1.0.11.dist-info/METADATA,sha256=s_HNeVZzGACli-KI2PbUvToviz9aFZvigOcCtd__lIw,41740
+signalwire_agents-1.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+signalwire_agents-1.0.11.dist-info/entry_points.txt,sha256=TDmLd7TVuj2AWfhxhaw6C26Do6oTdSQltsB4V7jpOjw,265
+signalwire_agents-1.0.11.dist-info/top_level.txt,sha256=kDGS6ZYv84K9P5Kyg9_S8P_pbUXoHkso0On_DB5bbWc,18
+signalwire_agents-1.0.11.dist-info/RECORD,,

{signalwire_agents-1.0.10.dist-info → signalwire_agents-1.0.11.dist-info}/entry_points.txt

@@ -1,4 +1,5 @@
 [console_scripts]
+mcp-gateway = signalwire_agents.mcp_gateway.gateway_service:main
 sw-agent-init = signalwire_agents.cli.init_project:main
 sw-search = signalwire_agents.cli.build_search:console_entry_point
 swaig-test = signalwire_agents.cli.swaig_test_wrapper:main