Refactor agent launch functionality in multi-agent-api.py to remove blocking parameters, update version to 0.0.79 in pyproject.toml and uv.lock, and enhance server startup logic in agent.py with healthcheck endpoint and improved thread management.

Author: MervinPraison

src/praisonai-agents/multi-agent-api.py

@@ -1,29 +1,20 @@
 from praisonaiagents import Agent
 
-# Create a weather agent
 weather_agent = Agent(
     instructions="""You are a weather agent that can provide weather information for a given city.""",
     llm="gpt-4o-mini"
 )
 
-# Create a stock market agent 
 stock_agent = Agent(
     instructions="""You are a stock market agent that can provide information about stock prices and market trends.""",
     llm="gpt-4o-mini"
 )
 
-# Create a travel agent
 travel_agent = Agent(
     instructions="""You are a travel agent that can provide recommendations for destinations, hotels, and activities.""",
     llm="gpt-4o-mini"
 )
 
-# Register the first two agents with blocking=False so they don't block execution
-weather_agent.launch(path="/weather", port=3030, blocking=False)
-stock_agent.launch(path="/stock", port=3030, blocking=False)
-
-# Register the last agent with blocking=True to keep the server running
-# This must be the last launch() call
-travel_agent.launch(path="/travel", port=3030, blocking=True)
-
-# The script will block at the last launch() call until the user presses Ctrl+C
+weather_agent.launch(path="/weather", port=3030)
+stock_agent.launch(path="/stock", port=3030)
+travel_agent.launch(path="/travel", port=3030) 

src/praisonai-agents/praisonaiagents/agent/agent.py

@@ -22,16 +22,16 @@
 import uuid
 from dataclasses import dataclass
 
+# Global variables for API server
+_server_started = False
+_registered_agents = {}
+_shared_app = None
+
 # Don't import FastAPI dependencies here - use lazy loading instead
 
 if TYPE_CHECKING:
     from ..task.task import Task
 
-# Shared variables for API server
-_shared_app = None
-_server_started = False
-_registered_agents = {}
-
 @dataclass
 class ChatCompletionMessage:
     content: str
@@ -1408,17 +1408,15 @@ async def execute_tool_async(self, function_name: str, arguments: Dict[str, Any]
             logging.error(f"Error in execute_tool_async: {str(e)}", exc_info=True)
             return {"error": f"Error in execute_tool_async: {str(e)}"}
 
-    def launch(self, path: str = '/', port: int = 8000, host: str = '0.0.0.0', autostart: bool = True, debug: bool = False, blocking: bool = True):
+    def launch(self, path: str = '/', port: int = 8000, host: str = '0.0.0.0', debug: bool = False):
         """
         Launch the agent as an HTTP API endpoint.
         
         Args:
             path: API endpoint path (default: '/')
             port: Server port (default: 8000)
             host: Server host (default: '0.0.0.0')
-            autostart: Whether to start the server automatically (default: True)
             debug: Enable debug mode for uvicorn (default: False)
-            blocking: If True, blocks the main thread to keep the server running (default: True)
             
         Returns:
             None
@@ -1431,6 +1429,8 @@ def launch(self, path: str = '/', port: int = 8000, host: str = '0.0.0.0', autos
             from fastapi import FastAPI, HTTPException, Request
             from fastapi.responses import JSONResponse
             from pydantic import BaseModel
+            import threading
+            import time
 
             # Define the request model here since we need pydantic
             class AgentQuery(BaseModel):
@@ -1458,6 +1458,11 @@ class AgentQuery(BaseModel):
             @_shared_app.get("/")
             async def root():
                 return {"message": "Welcome to PraisonAI Agents API. See /docs for usage."}
+            
+            # Add healthcheck endpoint
+            @_shared_app.get("/health")
+            async def healthcheck():
+                return {"status": "ok", "agents": list(_registered_agents.keys())}
 
         # Normalize path to ensure it starts with /
         if not path.startswith('/'):
@@ -1516,47 +1521,68 @@ async def handle_agent_query(request: Request, query_data: Optional[AgentQuery]
 
         print(f"🚀 Agent '{self.name}' available at http://{host}:{port}{path}")
 
-        # Start the server if this is the first launch call and autostart is True
-        if autostart and not _server_started:
+        # Start the server if it's not already running
+        if not _server_started:
             _server_started = True
 
-            # Add healthcheck endpoint
-            @_shared_app.get("/health")
-            async def healthcheck():
-                return {"status": "ok", "agents": list(_registered_agents.keys())}
-            
-            # Start the server in a separate thread to not block execution
-            import threading
+            # Start the server in a separate thread
             def run_server():
                 try:
+                    print(f"✅ FastAPI server started at http://{host}:{port}")
+                    print(f"📚 API documentation available at http://{host}:{port}/docs")
+                    print(f"🔌 Available endpoints: {', '.join(list(_registered_agents.keys()))}")
                     uvicorn.run(_shared_app, host=host, port=port, log_level="debug" if debug else "info")
                 except Exception as e:
                     logging.error(f"Error starting server: {str(e)}", exc_info=True)
                     print(f"❌ Error starting server: {str(e)}")
 
+            # Run server in a background thread
             server_thread = threading.Thread(target=run_server, daemon=True)
             server_thread.start()
 
-            # Give the server a moment to start up
-            import time
+            # Wait for a moment to allow the server to start and register endpoints
             time.sleep(0.5)
+        else:
+            # If server is already running, wait a moment to make sure the endpoint is registered
+            time.sleep(0.1)
+            print(f"🔌 Available endpoints: {', '.join(list(_registered_agents.keys()))}")
+        
+        # Get the stack frame to check if this is the last launch() call in the script
+        import inspect
+        stack = inspect.stack()
+        
+        # If this is called from a Python script (not interactive), try to detect if it's the last launch call
+        if len(stack) > 1 and stack[1].filename.endswith('.py'):
+            caller_frame = stack[1]
+            caller_line = caller_frame.lineno
 
-            print(f"✅ FastAPI server started at http://{host}:{port}")
-            print(f"📚 API documentation available at http://{host}:{port}/docs")
-            
-            # If blocking is True, keep the main thread alive
-            if blocking:
-                print("\nServer is running in blocking mode. Press Ctrl+C to stop...")
+            try:
+                # Read the file to check if there are more launch calls after this one
+                with open(caller_frame.filename, 'r') as f:
+                    lines = f.readlines()
+                
+                # Check if there are more launch() calls after the current line
+                has_more_launches = False
+                for line in lines[caller_line:]:
+                    if '.launch(' in line and not line.strip().startswith('#'):
+                        has_more_launches = True
+                        break
+                
+                # If this is the last launch call, block the main thread
+                if not has_more_launches:
+                    try:
+                        print("\nAll agents registered. Press Ctrl+C to stop the server.")
+                        while True:
+                            time.sleep(1)
+                    except KeyboardInterrupt:
+                        print("\nServer stopped")
+            except Exception as e:
+                # If something goes wrong with detection, block anyway to be safe
+                logging.error(f"Error in launch detection: {e}")
                 try:
                     while True:
                         time.sleep(1)
                 except KeyboardInterrupt:
                     print("\nServer stopped")
-            else:
-                # Note for non-blocking mode
-                print("\nNote: Server is running in a background thread. To keep it alive, either:")
-                print("1. Set blocking=True when calling launch()")
-                print("2. Keep your main application running")
-                print("3. Use a loop in your code to prevent the program from exiting")
-            
+        
         return None 

src/praisonai-agents/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "praisonaiagents"
-version = "0.0.78"
+version = "0.0.79"
 description = "Praison AI agents for completing complex tasks with Self Reflection Agents"
 authors = [
     { name="Mervin Praison" }