Compare commits

...
1 Commits
Author SHA1 Message Date
Mervin Praison bf483f3bd6 Release v2.7.4 2025-12-29 13:01:04 +00:00
38 changed files with 6975 additions and 118 deletions

No files matched your search

+1 -1
View File
@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.7.3" \
"praisonai>=2.7.4" \
"praisonai[chat]" \
"embedchain[github,youtube]"
+1 -1
View File
@@ -20,7 +20,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.7.3" \
"praisonai>=2.7.4" \
"praisonai[ui]" \
"praisonai[chat]" \
"praisonai[realtime]" \
+1 -1
View File
@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.7.3" \
"praisonai>=2.7.4" \
"praisonai[ui]" \
"praisonai[crewai]"
+17
View File
@@ -14,6 +14,7 @@ examples/
│ ├── memory/ # Memory and sessions
│ ├── code/ # Code editing and external CLI tools
│ └── ...
├── serve/ # Server and endpoints examples
├── yaml/ # YAML workflow examples
└── cookbooks/ # Complete use-case examples
```
@@ -31,6 +32,22 @@ examples/
| **Code** | [python/code/](python/code/) | [📖](https://docs.praison.ai/code) |
| **YAML** | [yaml/](yaml/) | [📖](https://docs.praison.ai/features/yaml-workflows) |
| **Doctor** | [doctor/](doctor/) | [📖](https://docs.praison.ai/cli/doctor) |
| **Serve** | [serve/](serve/) | [📖](https://docs.praison.ai/cli/serve) |
| **Endpoints** | [serve/](serve/) | [📖](https://docs.praison.ai/cli/endpoints) |
## Serve Examples
| Example | Description | CLI Command |
|---------|-------------|-------------|
| [unified_server.py](serve/unified_server.py) | All providers in one server | `praisonai serve unified` |
| [agent_as_api_single.py](serve/agent_as_api_single.py) | Single agent HTTP API | `praisonai serve agents` |
| [agents_as_api_router.py](serve/agents_as_api_router.py) | Multi-agent router API | `praisonai serve agents` |
| [a2a_server_client.py](serve/a2a_server_client.py) | A2A protocol server | `praisonai serve a2a` |
| [a2u_events_stream.py](serve/a2u_events_stream.py) | A2U event stream | `praisonai serve a2u` |
| [mcp_http_server.py](serve/mcp_http_server.py) | MCP HTTP server | `praisonai serve mcp` |
| [tools_as_mcp_server.py](serve/tools_as_mcp_server.py) | Tools as MCP server | `praisonai serve tools` |
| [agent_launch_modes.py](serve/agent_launch_modes.py) | Agent.launch() API | Python only |
| [endpoints_unified_client.py](serve/endpoints_unified_client.py) | Unified client | `praisonai endpoints` |
## Running Examples
+143
View File
@@ -0,0 +1,143 @@
#!/usr/bin/env python3
"""
Example: LLM-based Recipes
Demonstrates usage of LLM-powered recipes:
- ai-blog-generator
- ai-social-media-generator
- ai-faq-generator
- ai-sentiment-analyzer
"""
import os
import sys
# Ensure API key is set
if not os.environ.get("OPENAI_API_KEY"):
print("Please set OPENAI_API_KEY environment variable")
sys.exit(1)
# Add recipe tools to path
sys.path.insert(0, "/Users/praison/PraisonAI-tools/praisonai_tools/recipe_tools")
from llm_tool import LLMTool, llm_complete
def example_blog_generator():
"""Generate a blog post."""
print("\n=== AI Blog Generator ===")
llm = LLMTool(provider="openai", model="gpt-4o-mini")
topic = "The Future of AI in Healthcare"
response = llm.complete(
f"""Write a short blog post (3 paragraphs) about: {topic}
Include:
- Engaging introduction
- Key points
- Call to action
Format as markdown.""",
system="You are an expert blog writer specializing in technology topics.",
max_tokens=500,
)
print(f"Topic: {topic}")
print(f"\nGenerated Blog Post:\n{response.content}")
print(f"\nTokens used: {response.usage.get('total_tokens', 'N/A')}")
def example_social_media_generator():
"""Generate social media posts."""
print("\n=== AI Social Media Generator ===")
llm = LLMTool(provider="openai", model="gpt-4o-mini")
content = "We just launched our new AI-powered productivity app!"
response = llm.complete(
f"""Create social media posts for the following announcement:
{content}
Generate posts for:
1. Twitter (280 chars max)
2. LinkedIn (professional tone)
3. Instagram (with emoji suggestions)
Format each clearly.""",
system="You are a social media marketing expert.",
max_tokens=400,
)
print(f"Content: {content}")
print(f"\nGenerated Posts:\n{response.content}")
def example_faq_generator():
"""Generate FAQ from content."""
print("\n=== AI FAQ Generator ===")
llm = LLMTool(provider="openai", model="gpt-4o-mini")
documentation = """
Our product is a cloud-based project management tool.
It supports team collaboration, task tracking, and reporting.
Pricing starts at $10/month per user.
We offer a 14-day free trial.
Integration with Slack, GitHub, and Jira is available.
"""
response = llm.complete(
f"""Based on this documentation, generate 5 FAQ questions and answers:
{documentation}
Format as:
Q: [question]
A: [answer]
""",
system="You are a technical writer creating FAQ documentation.",
max_tokens=400,
)
print(f"Generated FAQ:\n{response.content}")
def example_sentiment_analyzer():
"""Analyze sentiment in text."""
print("\n=== AI Sentiment Analyzer ===")
llm = LLMTool(provider="openai", model="gpt-4o-mini")
reviews = [
"This product is amazing! Best purchase ever.",
"Terrible experience. Would not recommend.",
"It's okay, nothing special but does the job.",
]
for review in reviews:
response = llm.complete(
f"""Analyze the sentiment of this text and respond with ONLY one word: POSITIVE, NEGATIVE, or NEUTRAL
Text: {review}""",
max_tokens=10,
)
print(f"Review: {review[:50]}...")
print(f"Sentiment: {response.content.strip()}\n")
if __name__ == "__main__":
print("=" * 60)
print("LLM-Based Recipes Examples")
print("=" * 60)
example_blog_generator()
example_social_media_generator()
example_faq_generator()
example_sentiment_analyzer()
print("\n" + "=" * 60)
print("All examples completed!")
print("=" * 60)
+56 -31
View File
@@ -1,47 +1,72 @@
# Recipe Server Example
# PraisonAI Serve Examples
This example demonstrates how to configure and run the PraisonAI recipe server with authentication.
Examples for all PraisonAI server types and the unified endpoints CLI.
## Files
## Examples
- `serve.yaml` - Server configuration file
- `serve_example.py` - Python client example with authentication
| Example | Description | CLI Command |
|---------|-------------|-------------|
| `unified_server.py` | All providers in one server | `praisonai serve unified` |
| `agent_as_api_single.py` | Single agent HTTP API | `praisonai serve agents` |
| `agents_as_api_router.py` | Multi-agent router API | `praisonai serve agents` |
| `a2a_server_client.py` | A2A protocol server | `praisonai serve a2a` |
| `a2u_events_stream.py` | A2U event stream | `praisonai serve a2u` |
| `mcp_http_server.py` | MCP HTTP server | `praisonai serve mcp` |
| `tools_as_mcp_server.py` | Tools as MCP server | `praisonai serve tools` |
| `agent_launch_modes.py` | Agent.launch() API | Python only |
| `endpoints_unified_client.py` | Unified client | `praisonai endpoints` |
| `serve_example.py` | Recipe server client | `praisonai recipe serve` |
## Server Commands
| Command | Description |
|---------|-------------|
| `praisonai serve unified --port 8765` | Start unified server |
| `praisonai serve agents --file agents.yaml --port 8000` | Start agents API |
| `praisonai serve recipe --port 8765` | Start recipe server |
| `praisonai serve mcp --transport http --port 8080` | Start MCP server |
| `praisonai serve tools --port 8081` | Start tools MCP server |
| `praisonai serve a2a --port 8082` | Start A2A server |
| `praisonai serve a2u --port 8083` | Start A2U server |
## Client Commands
| Command | Description |
|---------|-------------|
| `praisonai endpoints list` | List all endpoints |
| `praisonai endpoints list --type agents-api` | Filter by type |
| `praisonai endpoints describe <name>` | Get endpoint details |
| `praisonai endpoints invoke <name> --input-json '{}'` | Invoke endpoint |
| `praisonai endpoints health` | Check server health |
| `praisonai endpoints types` | List provider types |
| `praisonai endpoints discovery` | Show discovery document |
## Quick Start
### 1. Install Dependencies
```bash
pip install praisonai[serve]
# Terminal 1: Start unified server
export OPENAI_API_KEY=your-key
praisonai serve unified --port 8765
# Terminal 2: Use endpoints CLI
praisonai endpoints health
praisonai endpoints list
praisonai endpoints types
```
### 2. Set Environment Variables
## Provider Types
```bash
export OPENAI_API_KEY=your-openai-key
export PRAISONAI_API_KEY=your-secret-api-key
```
### 3. Start the Server
```bash
# Using config file
praisonai recipe serve --config serve.yaml
# Or with CLI flags
praisonai recipe serve --port 8765 --auth api-key
```
### 4. Test with the Example
```bash
python serve_example.py
```
| Type | Description |
|------|-------------|
| `recipe` | Recipe runner endpoints |
| `agents-api` | Single/multi-agent HTTP API |
| `mcp` | MCP server (stdio, http, sse) |
| `tools-mcp` | Tools exposed as MCP server |
| `a2a` | Agent-to-agent protocol |
| `a2u` | Agent-to-user event stream |
## Configuration Options
See `serve.yaml` for all available options:
| Option | Description | Default |
|--------|-------------|---------|
| `host` | Server bind address | 127.0.0.1 |
+190
View File
@@ -0,0 +1,190 @@
#!/usr/bin/env python3
"""
A2A (Agent-to-Agent) Server and Client Example
Demonstrates:
- Launching an agent as A2A server
- Agent card discovery
- Sending A2A messages
- JSON-RPC protocol
Usage:
# Run this example (starts server)
python a2a_server_client.py
# Or use CLI:
praisonai serve a2a --port 8082
# Test with curl:
curl http://localhost:8082/.well-known/agent.json
curl -X POST http://localhost:8082/a2a \
-H "Content-Type: application/json" \
-d '{"jsonrpc":"2.0","method":"message/send","id":"1","params":{"message":{"role":"user","parts":[{"type":"text","text":"Hello!"}]}}}'
"""
import os
import sys
import json
import threading
import time
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def run_server():
"""Run A2A server."""
try:
from praisonaiagents import Agent
from praisonaiagents.ui.a2a import A2A
from fastapi import FastAPI
import uvicorn
except ImportError as e:
print(f"Error: Missing dependency - {e}")
print("Install with: pip install 'praisonaiagents[api]'")
sys.exit(1)
# Create agent
agent = Agent(
name="A2A-Assistant",
role="Helpful AI Assistant",
goal="Help other agents and users with their questions",
backstory="You are a knowledgeable assistant accessible via A2A protocol.",
llm="gpt-4o-mini"
)
# Create A2A interface
a2a = A2A(
agent=agent,
url="http://localhost:8082/a2a",
name="A2A-Assistant",
description="A helpful AI assistant accessible via A2A protocol"
)
# Create FastAPI app
app = FastAPI(title="PraisonAI A2A Server")
app.include_router(a2a.get_router())
# Add discovery endpoint
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-a2a",
"providers": [{"type": "a2a", "name": "A2A Protocol", "capabilities": ["agent-card", "message-send"]}],
"endpoints": [{"name": "a2a", "provider_type": "a2a", "streaming": ["sse"]}]
}
@app.get("/")
async def root():
return {
"message": "PraisonAI A2A Server",
"agent_card": "/.well-known/agent.json",
"a2a_endpoint": "/a2a",
"discovery": "/__praisonai__/discovery"
}
print("Starting A2A server on http://localhost:8082")
print("Agent Card: http://localhost:8082/.well-known/agent.json")
print("A2A Endpoint: http://localhost:8082/a2a")
uvicorn.run(app, host="127.0.0.1", port=8082, log_level="warning")
def run_client():
"""Run A2A client test."""
import urllib.request
import urllib.error
base_url = "http://localhost:8082"
print("\n--- A2A Client Test ---")
# 1. Get agent card
print("\n1. Getting Agent Card...")
try:
req = urllib.request.Request(f"{base_url}/.well-known/agent.json")
with urllib.request.urlopen(req, timeout=5) as resp:
agent_card = json.loads(resp.read().decode())
print(f" Agent: {agent_card.get('name')}")
print(f" URL: {agent_card.get('url')}")
print(f" Capabilities: {agent_card.get('capabilities', {})}")
except Exception as e:
print(f" Error: {e}")
return
# 2. Send A2A message
print("\n2. Sending A2A Message...")
message = {
"jsonrpc": "2.0",
"method": "message/send",
"id": "test-1",
"params": {
"message": {
"role": "user",
"parts": [{"type": "text", "text": "Hello! What can you help me with?"}]
}
}
}
try:
req = urllib.request.Request(
f"{base_url}/a2a",
data=json.dumps(message).encode(),
headers={"Content-Type": "application/json"},
method="POST"
)
with urllib.request.urlopen(req, timeout=30) as resp:
response = json.loads(resp.read().decode())
if "result" in response:
result_msg = response["result"].get("message", {})
parts = result_msg.get("parts", [])
for part in parts:
if part.get("type") == "text":
print(f" Response: {part.get('text', '')[:200]}...")
elif "error" in response:
print(f" Error: {response['error']}")
except Exception as e:
print(f" Error: {e}")
# 3. Check discovery
print("\n3. Checking Discovery...")
try:
req = urllib.request.Request(f"{base_url}/__praisonai__/discovery")
with urllib.request.urlopen(req, timeout=5) as resp:
discovery = json.loads(resp.read().decode())
print(f" Server: {discovery.get('server_name')}")
print(f" Providers: {[p.get('type') for p in discovery.get('providers', [])]}")
except Exception as e:
print(f" Error: {e}")
print("\n✓ A2A client test completed!")
def main():
print("=" * 60)
print("A2A (Agent-to-Agent) Server and Client Example")
print("=" * 60)
mode = sys.argv[1] if len(sys.argv) > 1 else "server"
if mode == "client":
run_client()
elif mode == "test":
# Start server in background, run client, then stop
print("Starting server in background...")
server_thread = threading.Thread(target=run_server, daemon=True)
server_thread.start()
time.sleep(3) # Wait for server to start
run_client()
else:
print("\nStarting A2A server...")
print("Press Ctrl+C to stop")
print("\nTo test, run in another terminal:")
print(" python a2a_server_client.py client")
run_server()
if __name__ == "__main__":
main()
+189
View File
@@ -0,0 +1,189 @@
#!/usr/bin/env python3
"""
A2U (Agent-to-User) Event Stream Example
Demonstrates:
- Launching an A2U event stream server
- Subscribing to agent events
- SSE (Server-Sent Events) streaming
- Event filtering
Usage:
# Run this example (starts server)
python a2u_events_stream.py
# Or use CLI:
praisonai serve a2u --port 8083
# Test with curl:
curl http://localhost:8083/a2u/info
curl -N http://localhost:8083/a2u/events/events
"""
import os
import sys
import json
import threading
import time
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def run_server():
"""Run A2U server."""
try:
from fastapi import FastAPI
import uvicorn
from praisonai.endpoints.a2u_server import (
create_a2u_routes,
emit_agent_started,
emit_agent_thinking,
emit_agent_response,
emit_agent_completed,
)
except ImportError as e:
print(f"Error: Missing dependency - {e}")
print("Install with: pip install praisonai[serve]")
sys.exit(1)
app = FastAPI(title="PraisonAI A2U Server")
# Add A2U routes
create_a2u_routes(app)
# Add discovery endpoint
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-a2u",
"providers": [{"type": "a2u", "name": "A2U Event Stream", "capabilities": ["subscribe", "stream"]}],
"endpoints": [{"name": "events", "provider_type": "a2u", "streaming": ["sse"]}]
}
@app.get("/")
async def root():
return {
"message": "PraisonAI A2U Server",
"info": "/a2u/info",
"events": "/a2u/events/<stream>",
"discovery": "/__praisonai__/discovery"
}
# Demo endpoint to emit events
@app.post("/demo/emit")
async def demo_emit():
"""Emit demo events for testing."""
agent_id = "demo-agent-1"
emit_agent_started(agent_id, "DemoAgent")
emit_agent_thinking(agent_id, "Processing your request...")
emit_agent_response(agent_id, "Here is my response to your query.")
emit_agent_completed(agent_id, {"status": "success"})
return {"message": "Demo events emitted", "agent_id": agent_id}
print("Starting A2U server on http://localhost:8083")
print("Info: http://localhost:8083/a2u/info")
print("Events: http://localhost:8083/a2u/events/events")
print("Demo: POST http://localhost:8083/demo/emit")
uvicorn.run(app, host="127.0.0.1", port=8083, log_level="warning")
def run_client():
"""Run A2U client test."""
import urllib.request
import urllib.error
base_url = "http://localhost:8083"
print("\n--- A2U Client Test ---")
# 1. Get A2U info
print("\n1. Getting A2U Info...")
try:
req = urllib.request.Request(f"{base_url}/a2u/info")
with urllib.request.urlopen(req, timeout=5) as resp:
info = json.loads(resp.read().decode())
print(f" Name: {info.get('name')}")
print(f" Version: {info.get('version')}")
print(f" Streams: {info.get('streams')}")
print(f" Event Types: {info.get('event_types')}")
except Exception as e:
print(f" Error: {e}")
return
# 2. Subscribe to events
print("\n2. Subscribing to Events...")
try:
req = urllib.request.Request(
f"{base_url}/a2u/subscribe",
data=json.dumps({"stream": "events", "filters": []}).encode(),
headers={"Content-Type": "application/json"},
method="POST"
)
with urllib.request.urlopen(req, timeout=5) as resp:
sub = json.loads(resp.read().decode())
print(f" Subscription ID: {sub.get('subscription_id')}")
print(f" Stream URL: {sub.get('stream_url')}")
except Exception as e:
print(f" Error: {e}")
# 3. Emit demo events
print("\n3. Emitting Demo Events...")
try:
req = urllib.request.Request(
f"{base_url}/demo/emit",
data=b"{}",
headers={"Content-Type": "application/json"},
method="POST"
)
with urllib.request.urlopen(req, timeout=5) as resp:
result = json.loads(resp.read().decode())
print(f" {result.get('message')}")
print(f" Agent ID: {result.get('agent_id')}")
except Exception as e:
print(f" Error: {e}")
# 4. Check discovery
print("\n4. Checking Discovery...")
try:
req = urllib.request.Request(f"{base_url}/__praisonai__/discovery")
with urllib.request.urlopen(req, timeout=5) as resp:
discovery = json.loads(resp.read().decode())
print(f" Server: {discovery.get('server_name')}")
print(f" Providers: {[p.get('type') for p in discovery.get('providers', [])]}")
except Exception as e:
print(f" Error: {e}")
print("\n✓ A2U client test completed!")
print("\nTo stream events, run:")
print(f" curl -N {base_url}/a2u/events/events")
def main():
print("=" * 60)
print("A2U (Agent-to-User) Event Stream Example")
print("=" * 60)
mode = sys.argv[1] if len(sys.argv) > 1 else "server"
if mode == "client":
run_client()
elif mode == "test":
print("Starting server in background...")
server_thread = threading.Thread(target=run_server, daemon=True)
server_thread.start()
time.sleep(2)
run_client()
else:
print("\nStarting A2U server...")
print("Press Ctrl+C to stop")
print("\nTo test, run in another terminal:")
print(" python a2u_events_stream.py client")
run_server()
if __name__ == "__main__":
main()
+132
View File
@@ -0,0 +1,132 @@
#!/usr/bin/env python3
"""
Single Agent as HTTP API Example
Demonstrates launching a single agent as an HTTP API server with:
- REST endpoint for queries
- Streaming support (SSE)
- Discovery endpoint
- Health check
Usage:
# Run this example
python agent_as_api_single.py
# Or use CLI:
praisonai serve agents --file agents.yaml --port 8000
# Test with curl:
curl -X POST http://localhost:8000/agent \
-H "Content-Type: application/json" \
-d '{"query": "What is AI?"}'
"""
import os
import sys
# Ensure API key is set
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def main():
print("=" * 60)
print("Single Agent as HTTP API Example")
print("=" * 60)
try:
from praisonaiagents import Agent
except ImportError:
print("Error: praisonaiagents not installed")
print("Install with: pip install praisonaiagents")
sys.exit(1)
# Create a single agent
agent = Agent(
name="Assistant",
role="Helpful AI Assistant",
goal="Help users with their questions accurately and concisely",
backstory="You are a knowledgeable assistant who provides clear, helpful answers.",
llm="gpt-4o-mini",
verbose=True
)
print(f"Agent: {agent.name}")
print(f"Role: {agent.role}")
# Launch as HTTP API
print("\nLaunching agent as HTTP API...")
print("Endpoints:")
print(" POST /agent - Query the agent")
print(" GET /health - Health check")
print(" GET /__praisonai__/discovery - Discovery document")
print("\nPress Ctrl+C to stop")
try:
# Use Agent.launch() to start HTTP server
agent.launch(
port=8000,
host="127.0.0.1",
path="/agent"
)
except KeyboardInterrupt:
print("\nServer stopped")
except Exception as e:
print(f"Error launching server: {e}")
# Fallback: manual FastAPI setup
print("\nFallback: Using manual FastAPI setup...")
launch_manual(agent)
def launch_manual(agent):
"""Manual FastAPI server setup as fallback."""
try:
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import uvicorn
except ImportError:
print("Error: FastAPI/uvicorn not installed")
print("Install with: pip install fastapi uvicorn")
return
app = FastAPI(title="PraisonAI Agent API")
@app.post("/agent")
async def query_agent(request: Request):
try:
body = await request.json()
query = body.get("query", "")
if not query:
return JSONResponse({"error": "query required"}, status_code=400)
response = agent.chat(query)
return {"response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.get("/health")
async def health():
return {"status": "healthy", "agent": agent.name}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-agent",
"providers": [{"type": "agents-api", "name": "Single Agent API"}],
"endpoints": [{"name": "agent", "provider_type": "agents-api"}]
}
@app.get("/")
async def root():
return {
"message": f"PraisonAI Agent: {agent.name}",
"endpoints": ["/agent", "/health", "/__praisonai__/discovery"]
}
uvicorn.run(app, host="127.0.0.1", port=8000)
if __name__ == "__main__":
main()
+220
View File
@@ -0,0 +1,220 @@
#!/usr/bin/env python3
"""
Agent.launch() and Agents.launch() Example
Demonstrates the high-level Python API for launching servers:
- Agent.launch() for single agent
- Agents.launch() for multi-agent
- Different server modes (HTTP, MCP, A2A)
Usage:
# Run this example
python agent_launch_modes.py
# Test endpoints:
curl http://localhost:8000/agent
curl http://localhost:8000/health
"""
import os
import sys
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def demo_single_agent_launch():
"""Demo Agent.launch() for single agent."""
print("\n--- Single Agent Launch ---")
try:
from praisonaiagents import Agent
except ImportError:
print("Error: praisonaiagents not installed")
return False
agent = Agent(
name="Assistant",
role="Helpful AI Assistant",
goal="Help users with their questions",
llm="gpt-4o-mini"
)
print(f"Agent: {agent.name}")
print("Launching as HTTP API on port 8000...")
print("\nEndpoints:")
print(" POST /agent - Query the agent")
print(" GET /health - Health check")
print("\nPress Ctrl+C to stop")
try:
agent.launch(port=8000, host="127.0.0.1", path="/agent")
except AttributeError:
print("Agent.launch() not available, using fallback...")
launch_single_fallback(agent)
except KeyboardInterrupt:
print("\nStopped")
return True
def demo_multi_agent_launch():
"""Demo Agents.launch() for multi-agent."""
print("\n--- Multi-Agent Launch ---")
try:
from praisonaiagents import Agent, PraisonAIAgents
except ImportError:
print("Error: praisonaiagents not installed")
return False
researcher = Agent(
name="Researcher",
role="Research Specialist",
goal="Find accurate information",
llm="gpt-4o-mini"
)
writer = Agent(
name="Writer",
role="Content Writer",
goal="Create engaging content",
llm="gpt-4o-mini"
)
agents = PraisonAIAgents(agents=[researcher, writer])
print(f"Agents: {[researcher.name, writer.name]}")
print("Launching as HTTP API on port 8000...")
print("\nEndpoints:")
print(" POST /agents - Route to best agent")
print(" POST /agents/researcher - Query researcher")
print(" POST /agents/writer - Query writer")
print(" GET /agents/list - List agents")
print("\nPress Ctrl+C to stop")
try:
agents.launch(port=8000, host="127.0.0.1")
except AttributeError:
print("Agents.launch() not available, using fallback...")
launch_multi_fallback([researcher, writer])
except KeyboardInterrupt:
print("\nStopped")
return True
def launch_single_fallback(agent):
"""Fallback implementation for Agent.launch()."""
try:
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import uvicorn
except ImportError:
print("Error: FastAPI/uvicorn not installed")
return
app = FastAPI(title=f"PraisonAI Agent: {agent.name}")
@app.post("/agent")
async def query(request: Request):
try:
body = await request.json()
query_text = body.get("query", "")
response = agent.chat(query_text)
return {"response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.get("/health")
async def health():
return {"status": "healthy", "agent": agent.name}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": f"praisonai-{agent.name.lower()}",
"providers": [{"type": "agents-api", "name": "Single Agent API"}],
"endpoints": [{"name": "agent", "provider_type": "agents-api"}]
}
uvicorn.run(app, host="127.0.0.1", port=8000)
def launch_multi_fallback(agents):
"""Fallback implementation for Agents.launch()."""
try:
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import uvicorn
except ImportError:
print("Error: FastAPI/uvicorn not installed")
return
app = FastAPI(title="PraisonAI Multi-Agent API")
agent_map = {a.name.lower(): a for a in agents}
@app.post("/agents")
async def route(request: Request):
try:
body = await request.json()
query_text = body.get("query", "")
# Simple routing
agent = agents[0]
response = agent.chat(query_text)
return {"agent": agent.name, "response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.post("/agents/{name}")
async def query_agent(name: str, request: Request):
agent = agent_map.get(name.lower())
if not agent:
return JSONResponse({"error": f"Agent not found: {name}"}, status_code=404)
try:
body = await request.json()
response = agent.chat(body.get("query", ""))
return {"agent": agent.name, "response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.get("/agents/list")
async def list_agents():
return {"agents": [{"name": a.name, "role": a.role} for a in agents]}
@app.get("/health")
async def health():
return {"status": "healthy", "agents": len(agents)}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-agents",
"providers": [{"type": "agents-api", "name": "Multi-Agent API"}],
"endpoints": [
{"name": "agents", "provider_type": "agents-api"},
*[{"name": f"agents/{a.name.lower()}", "provider_type": "agents-api"} for a in agents]
]
}
uvicorn.run(app, host="127.0.0.1", port=8000)
def main():
print("=" * 60)
print("Agent.launch() and Agents.launch() Example")
print("=" * 60)
mode = sys.argv[1] if len(sys.argv) > 1 else "single"
if mode == "multi":
demo_multi_agent_launch()
else:
demo_single_agent_launch()
if __name__ == "__main__":
main()
+189
View File
@@ -0,0 +1,189 @@
#!/usr/bin/env python3
"""
Multi-Agent Router as HTTP API Example
Demonstrates launching multiple agents as an HTTP API with:
- Router endpoint for automatic agent selection
- Per-agent endpoints
- Streaming support (SSE)
- Discovery endpoint
Usage:
# Run this example
python agents_as_api_router.py
# Or use CLI:
praisonai serve agents --file agents.yaml --port 8000
# Test with curl:
curl -X POST http://localhost:8000/agents \
-H "Content-Type: application/json" \
-d '{"query": "Research AI trends"}'
"""
import os
import sys
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def main():
print("=" * 60)
print("Multi-Agent Router as HTTP API Example")
print("=" * 60)
try:
from praisonaiagents import Agent, PraisonAIAgents
except ImportError:
print("Error: praisonaiagents not installed")
print("Install with: pip install praisonaiagents")
sys.exit(1)
# Create multiple agents
researcher = Agent(
name="Researcher",
role="Research Specialist",
goal="Find accurate and relevant information",
backstory="Expert at finding and synthesizing information from various sources.",
llm="gpt-4o-mini"
)
writer = Agent(
name="Writer",
role="Content Writer",
goal="Create engaging content from research",
backstory="Skilled writer who transforms research into readable content.",
llm="gpt-4o-mini"
)
analyst = Agent(
name="Analyst",
role="Data Analyst",
goal="Analyze data and provide insights",
backstory="Expert at analyzing data and extracting meaningful insights.",
llm="gpt-4o-mini"
)
agents = [researcher, writer, analyst]
print(f"Agents: {[a.name for a in agents]}")
# Create multi-agent system
praison = PraisonAIAgents(agents=agents)
print("\nLaunching multi-agent router as HTTP API...")
print("Endpoints:")
print(" POST /agents - Route to best agent")
print(" POST /agents/researcher - Query researcher")
print(" POST /agents/writer - Query writer")
print(" POST /agents/analyst - Query analyst")
print(" GET /agents/list - List all agents")
print(" GET /health - Health check")
print(" GET /__praisonai__/discovery - Discovery document")
print("\nPress Ctrl+C to stop")
try:
# Use Agents.launch() to start HTTP server
praison.launch(
port=8000,
host="127.0.0.1"
)
except KeyboardInterrupt:
print("\nServer stopped")
except Exception as e:
print(f"Error launching server: {e}")
print("\nFallback: Using manual FastAPI setup...")
launch_manual(agents)
def launch_manual(agents):
"""Manual FastAPI server setup as fallback."""
try:
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import uvicorn
except ImportError:
print("Error: FastAPI/uvicorn not installed")
print("Install with: pip install fastapi uvicorn")
return
app = FastAPI(title="PraisonAI Multi-Agent API")
# Create agent lookup
agent_map = {a.name.lower(): a for a in agents}
@app.post("/agents")
async def route_query(request: Request):
"""Route query to best agent."""
try:
body = await request.json()
query = body.get("query", "")
if not query:
return JSONResponse({"error": "query required"}, status_code=400)
# Simple routing based on keywords
query_lower = query.lower()
if any(w in query_lower for w in ["research", "find", "search"]):
agent = agent_map.get("researcher", agents[0])
elif any(w in query_lower for w in ["write", "create", "draft"]):
agent = agent_map.get("writer", agents[0])
elif any(w in query_lower for w in ["analyze", "data", "insight"]):
agent = agent_map.get("analyst", agents[0])
else:
agent = agents[0]
response = agent.chat(query)
return {"agent": agent.name, "response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.post("/agents/{agent_name}")
async def query_agent(agent_name: str, request: Request):
"""Query specific agent."""
agent = agent_map.get(agent_name.lower())
if not agent:
return JSONResponse({"error": f"Agent not found: {agent_name}"}, status_code=404)
try:
body = await request.json()
query = body.get("query", "")
response = agent.chat(query)
return {"agent": agent.name, "response": response}
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
@app.get("/agents/list")
async def list_agents():
return {"agents": [{"name": a.name, "role": a.role} for a in agents]}
@app.get("/health")
async def health():
return {"status": "healthy", "agents": len(agents)}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-agents-router",
"providers": [{"type": "agents-api", "name": "Multi-Agent Router"}],
"endpoints": [
{"name": "agents", "provider_type": "agents-api"},
*[{"name": f"agents/{a.name.lower()}", "provider_type": "agents-api"} for a in agents]
]
}
@app.get("/")
async def root():
return {
"message": "PraisonAI Multi-Agent Router",
"agents": [a.name for a in agents],
"endpoints": ["/agents", "/agents/list", "/health"]
}
uvicorn.run(app, host="127.0.0.1", port=8000)
if __name__ == "__main__":
main()
+116
View File
@@ -0,0 +1,116 @@
#!/usr/bin/env python3
"""
Unified Endpoints Client Example
Demonstrates using the praisonai endpoints CLI to interact with any PraisonAI server.
Works with recipe, agents-api, mcp, a2a, a2u, and tools-mcp servers.
Usage:
# Start any server first (e.g., unified server)
praisonai serve unified --port 8765
# Then run this example
python endpoints_unified_client.py
# Or use CLI directly:
praisonai endpoints list
praisonai endpoints types
praisonai endpoints health
praisonai endpoints discovery
"""
import json
import os
import subprocess
import sys
def run_cli(args: list) -> tuple:
"""Run praisonai endpoints CLI command."""
cmd = ["python3", "-m", "praisonai.cli.main", "endpoints"] + args
result = subprocess.run(cmd, capture_output=True, text=True)
return result.returncode, result.stdout, result.stderr
def main():
print("=" * 60)
print("PraisonAI Unified Endpoints Client Example")
print("=" * 60)
server_url = os.environ.get("PRAISONAI_ENDPOINTS_URL", "http://localhost:8765")
print(f"Server URL: {server_url}")
# 1. List provider types
print("\n--- Provider Types ---")
code, out, err = run_cli(["types", "--format", "json"])
if code == 0:
types = json.loads(out)
print(f"Supported types: {[t['type'] for t in types]}")
else:
print(f"Error: {err}")
# 2. Health check
print("\n--- Health Check ---")
code, out, err = run_cli(["health", "--url", server_url])
if code == 0:
print("✓ Server is healthy")
print(out.strip())
elif code == 8:
print("✗ Connection error - server not running")
print("\nStart a server first:")
print(" praisonai serve unified --port 8765")
print(" praisonai serve agents --file agents.yaml --port 8765")
print(" praisonai serve a2a --port 8765")
return 1
else:
print(f"✗ Health check failed: {err}")
return 1
# 3. Discovery document
print("\n--- Discovery Document ---")
code, out, err = run_cli(["discovery", "--url", server_url])
if code == 0:
try:
discovery = json.loads(out)
print(f"Server: {discovery.get('server_name', 'unknown')}")
print(f"Schema version: {discovery.get('schema_version', 'unknown')}")
providers = discovery.get('providers', [])
print(f"Providers: {[p.get('type') for p in providers]}")
endpoints = discovery.get('endpoints', [])
print(f"Endpoints: {[e.get('name') for e in endpoints]}")
except json.JSONDecodeError:
print(out.strip())
else:
print(f"Discovery not available: {err}")
# 4. List endpoints
print("\n--- List Endpoints ---")
code, out, err = run_cli(["list", "--url", server_url, "--format", "json"])
if code == 0:
try:
endpoints = json.loads(out)
if isinstance(endpoints, list):
for ep in endpoints[:5]: # Show first 5
print(f" - {ep.get('name')}: {ep.get('description', '')[:50]}")
else:
print(out.strip()[:200])
except json.JSONDecodeError:
print(out.strip()[:200])
else:
print(f"List failed: {err}")
# 5. Filter by type
print("\n--- Filter by Provider Type ---")
for ptype in ["agents-api", "a2a", "recipe"]:
code, out, err = run_cli(["list", "--url", server_url, "--type", ptype])
if code == 0 and out.strip():
print(f" {ptype}: {len(out.strip().split(chr(10)))} endpoints")
print("\n" + "=" * 60)
print("Example completed successfully!")
print("=" * 60)
return 0
if __name__ == "__main__":
sys.exit(main())
+138
View File
@@ -0,0 +1,138 @@
#!/usr/bin/env python3
"""
MCP HTTP Server Example
Demonstrates launching an MCP server with HTTP transport:
- HTTP-based MCP protocol
- Tool registration
- Discovery endpoint
Usage:
# Run this example
python mcp_http_server.py
# Or use CLI:
praisonai serve mcp --transport http --port 8080
# Test with curl:
curl http://localhost:8080/mcp/tools
curl -X POST http://localhost:8080/mcp/tools/call \
-H "Content-Type: application/json" \
-d '{"tool": "search", "arguments": {"query": "AI news"}}'
"""
import sys
def main():
print("=" * 60)
print("MCP HTTP Server Example")
print("=" * 60)
try:
from fastapi import FastAPI
import uvicorn
except ImportError:
print("Error: FastAPI/uvicorn not installed")
print("Install with: pip install fastapi uvicorn")
sys.exit(1)
# Define tools
def search(query: str) -> str:
"""Search the web for information."""
return f"Search results for: {query}"
def calculate(expression: str) -> str:
"""Calculate a math expression safely."""
try:
# Safe eval for basic math
allowed = set("0123456789+-*/.() ")
if all(c in allowed for c in expression):
return str(eval(expression))
return "Error: Invalid expression"
except Exception as e:
return f"Error: {e}"
def get_weather(city: str) -> str:
"""Get weather for a city (mock)."""
return f"Weather in {city}: Sunny, 72°F"
tools = {
"search": {"func": search, "description": "Search the web for information"},
"calculate": {"func": calculate, "description": "Calculate a math expression"},
"get_weather": {"func": get_weather, "description": "Get weather for a city"},
}
app = FastAPI(title="PraisonAI MCP Server")
@app.get("/mcp/tools")
async def list_tools():
"""List available MCP tools."""
return {
"tools": [
{
"name": name,
"description": info["description"],
"inputSchema": {
"type": "object",
"properties": {
list(info["func"].__code__.co_varnames)[0]: {"type": "string"}
},
"required": [list(info["func"].__code__.co_varnames)[0]]
}
}
for name, info in tools.items()
]
}
@app.post("/mcp/tools/call")
async def call_tool(request: dict):
"""Call an MCP tool."""
tool_name = request.get("tool")
arguments = request.get("arguments", {})
if tool_name not in tools:
return {"error": f"Tool not found: {tool_name}"}
try:
func = tools[tool_name]["func"]
# Get first argument
arg_name = list(func.__code__.co_varnames)[0]
arg_value = arguments.get(arg_name, "")
result = func(arg_value)
return {"result": result, "tool": tool_name}
except Exception as e:
return {"error": str(e)}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-mcp",
"providers": [{"type": "mcp", "name": "MCP Server", "capabilities": ["list-tools", "call-tool"]}],
"endpoints": [{"name": "mcp", "provider_type": "mcp"}]
}
@app.get("/health")
async def health():
return {"status": "healthy", "tools": len(tools)}
@app.get("/")
async def root():
return {
"message": "PraisonAI MCP Server (HTTP)",
"tools": list(tools.keys()),
"endpoints": ["/mcp/tools", "/mcp/tools/call", "/health"]
}
print(f"Tools: {list(tools.keys())}")
print("\nStarting MCP HTTP server on http://localhost:8080")
print("List tools: GET http://localhost:8080/mcp/tools")
print("Call tool: POST http://localhost:8080/mcp/tools/call")
print("\nPress Ctrl+C to stop")
uvicorn.run(app, host="127.0.0.1", port=8080)
if __name__ == "__main__":
main()
+222
View File
@@ -0,0 +1,222 @@
#!/usr/bin/env python3
"""
Tools as MCP Server Example
Demonstrates exposing Python tools as an MCP server:
- Register custom functions as MCP tools
- SSE transport for Claude Desktop/Cursor
- Stdio transport for local integration
Usage:
# Run this example (SSE mode)
python tools_as_mcp_server.py
# Or use CLI:
praisonai serve tools --port 8081
# Test with curl:
curl http://localhost:8081/sse
"""
import sys
def main():
print("=" * 60)
print("Tools as MCP Server Example")
print("=" * 60)
try:
from praisonaiagents.mcp import ToolsMCPServer
except ImportError:
print("Error: praisonaiagents[mcp] not installed")
print("Install with: pip install 'praisonaiagents[mcp]'")
print("\nFallback: Using manual implementation...")
run_manual_server()
return
# Define custom tools
def search_web(query: str) -> str:
"""Search the web for information."""
return f"Search results for: {query}"
def analyze_text(text: str) -> str:
"""Analyze text for sentiment and key points."""
word_count = len(text.split())
return f"Analysis: {word_count} words, sentiment: neutral"
def translate(text: str, target_language: str = "Spanish") -> str:
"""Translate text to target language (mock)."""
return f"[Translated to {target_language}]: {text}"
def summarize(text: str) -> str:
"""Summarize long text."""
words = text.split()
if len(words) > 20:
return " ".join(words[:20]) + "..."
return text
# Create MCP server
server = ToolsMCPServer(name="praisonai-tools")
# Register tools
server.register_tool(search_web)
server.register_tool(analyze_text)
server.register_tool(translate)
server.register_tool(summarize)
print(f"Registered tools: {[search_web.__name__, analyze_text.__name__, translate.__name__, summarize.__name__]}")
print("\nStarting MCP server with SSE transport...")
print("SSE endpoint: http://localhost:8081/sse")
print("\nClaude Desktop config (claude_desktop_config.json):")
print(''' {
"mcpServers": {
"praisonai-tools": {
"url": "http://localhost:8081/sse"
}
}
}''')
print("\nPress Ctrl+C to stop")
server.run(transport="sse", host="0.0.0.0", port=8081)
def run_manual_server():
"""Manual MCP server implementation as fallback."""
try:
from fastapi import FastAPI
from fastapi.responses import StreamingResponse
import uvicorn
import json
import asyncio
except ImportError:
print("Error: FastAPI/uvicorn not installed")
print("Install with: pip install fastapi uvicorn")
sys.exit(1)
# Define tools
tools = {
"search_web": {
"func": lambda query: f"Search results for: {query}",
"description": "Search the web for information",
"params": {"query": "string"}
},
"analyze_text": {
"func": lambda text: f"Analysis: {len(text.split())} words",
"description": "Analyze text for sentiment and key points",
"params": {"text": "string"}
},
}
app = FastAPI(title="PraisonAI Tools MCP Server")
@app.get("/sse")
async def sse_endpoint():
"""SSE endpoint for MCP."""
async def event_stream():
# Send initial tools list
tools_list = {
"jsonrpc": "2.0",
"method": "tools/list",
"result": {
"tools": [
{
"name": name,
"description": info["description"],
"inputSchema": {
"type": "object",
"properties": {k: {"type": v} for k, v in info["params"].items()},
"required": list(info["params"].keys())
}
}
for name, info in tools.items()
]
}
}
yield f"data: {json.dumps(tools_list)}\n\n"
# Keep connection alive
while True:
await asyncio.sleep(30)
yield f"data: {json.dumps({'type': 'ping'})}\n\n"
return StreamingResponse(
event_stream(),
media_type="text/event-stream",
headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
)
@app.post("/messages/")
async def handle_message(request: dict):
"""Handle MCP messages."""
method = request.get("method")
params = request.get("params", {})
req_id = request.get("id")
if method == "tools/list":
return {
"jsonrpc": "2.0",
"id": req_id,
"result": {
"tools": [
{
"name": name,
"description": info["description"],
"inputSchema": {
"type": "object",
"properties": {k: {"type": v} for k, v in info["params"].items()}
}
}
for name, info in tools.items()
]
}
}
elif method == "tools/call":
tool_name = params.get("name")
arguments = params.get("arguments", {})
if tool_name not in tools:
return {"jsonrpc": "2.0", "id": req_id, "error": {"code": -32601, "message": f"Tool not found: {tool_name}"}}
try:
func = tools[tool_name]["func"]
arg = list(arguments.values())[0] if arguments else ""
result = func(arg)
return {
"jsonrpc": "2.0",
"id": req_id,
"result": {"content": [{"type": "text", "text": result}]}
}
except Exception as e:
return {"jsonrpc": "2.0", "id": req_id, "error": {"code": -32000, "message": str(e)}}
return {"jsonrpc": "2.0", "id": req_id, "error": {"code": -32601, "message": f"Unknown method: {method}"}}
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-tools-mcp",
"providers": [{"type": "tools-mcp", "name": "Tools MCP Server", "capabilities": ["list-tools", "call-tool"]}],
"endpoints": [{"name": "tools", "provider_type": "tools-mcp"}]
}
@app.get("/")
async def root():
return {
"message": "PraisonAI Tools MCP Server",
"tools": list(tools.keys()),
"sse": "/sse",
"messages": "/messages/"
}
print(f"Tools: {list(tools.keys())}")
print("\nStarting Tools MCP server on http://localhost:8081")
print("SSE endpoint: http://localhost:8081/sse")
print("\nPress Ctrl+C to stop")
uvicorn.run(app, host="127.0.0.1", port=8081)
if __name__ == "__main__":
main()
+241
View File
@@ -0,0 +1,241 @@
#!/usr/bin/env python3
"""
Unified Server Example
Demonstrates launching a unified server with all provider types:
- agents-api: Single/multi-agent HTTP API
- recipe: Recipe runner endpoints
- mcp: MCP server
- a2a: Agent-to-agent protocol
- a2u: Agent-to-user event stream
Usage:
# Run this example
python unified_server.py
# Or use CLI:
praisonai serve unified --port 8765
# Test with curl:
curl http://localhost:8765/__praisonai__/discovery
curl http://localhost:8765/health
"""
import os
import sys
if not os.environ.get("OPENAI_API_KEY"):
print("Error: OPENAI_API_KEY environment variable not set")
sys.exit(1)
def main():
print("=" * 60)
print("Unified Server Example")
print("=" * 60)
try:
from fastapi import FastAPI
import uvicorn
from praisonaiagents import Agent
except ImportError as e:
print(f"Error: Missing dependency - {e}")
print("Install with: pip install praisonaiagents fastapi uvicorn")
sys.exit(1)
# Create agents
assistant = Agent(
name="Assistant",
role="Helpful AI Assistant",
goal="Help users with their questions",
llm="gpt-4o-mini"
)
researcher = Agent(
name="Researcher",
role="Research Specialist",
goal="Find accurate information",
llm="gpt-4o-mini"
)
agents = [assistant, researcher]
agent_map = {a.name.lower(): a for a in agents}
# Create unified app
app = FastAPI(
title="PraisonAI Unified Server",
description="Unified server with all PraisonAI providers"
)
# Discovery endpoint
@app.get("/__praisonai__/discovery")
async def discovery():
return {
"schema_version": "1.0.0",
"server_name": "praisonai-unified",
"providers": [
{"type": "agents-api", "name": "Agents API", "capabilities": ["invoke", "health"]},
{"type": "recipe", "name": "Recipe Runner", "capabilities": ["list", "describe", "invoke"]},
{"type": "mcp", "name": "MCP Server", "capabilities": ["list-tools", "call-tool"]},
{"type": "a2a", "name": "A2A Protocol", "capabilities": ["agent-card", "message-send"]},
{"type": "a2u", "name": "A2U Event Stream", "capabilities": ["subscribe", "stream"]},
],
"endpoints": [
{"name": "agents", "provider_type": "agents-api", "description": "Multi-agent router"},
{"name": "agent", "provider_type": "agents-api", "description": "Single agent endpoint"},
{"name": "mcp/tools", "provider_type": "mcp", "description": "MCP tools"},
{"name": "a2a", "provider_type": "a2a", "description": "A2A protocol"},
{"name": "a2u/events", "provider_type": "a2u", "description": "Event stream"},
]
}
# Health endpoint
@app.get("/health")
async def health():
return {
"status": "healthy",
"providers": ["agents-api", "recipe", "mcp", "a2a", "a2u"],
"agents": len(agents)
}
# Agents API endpoints
@app.post("/agents")
async def route_agents(request: dict):
query = request.get("query", "")
agent = agents[0] # Simple routing
try:
response = agent.chat(query)
return {"agent": agent.name, "response": response}
except Exception as e:
return {"error": str(e)}
@app.post("/agents/{name}")
async def query_agent(name: str, request: dict):
agent = agent_map.get(name.lower())
if not agent:
return {"error": f"Agent not found: {name}"}
try:
response = agent.chat(request.get("query", ""))
return {"agent": agent.name, "response": response}
except Exception as e:
return {"error": str(e)}
@app.get("/agents/list")
async def list_agents():
return {"agents": [{"name": a.name, "role": a.role} for a in agents]}
# Single agent endpoint
@app.post("/agent")
async def single_agent(request: dict):
query = request.get("query", "")
try:
response = assistant.chat(query)
return {"response": response}
except Exception as e:
return {"error": str(e)}
# MCP endpoints
mcp_tools = {
"search": {"description": "Search the web", "func": lambda q: f"Results for: {q}"},
"calculate": {"description": "Calculate expression", "func": lambda e: str(eval(e)) if e.replace(" ", "").replace("+", "").replace("-", "").replace("*", "").replace("/", "").replace(".", "").isdigit() or True else "Error"},
}
@app.get("/mcp/tools")
async def list_mcp_tools():
return {
"tools": [
{"name": name, "description": info["description"]}
for name, info in mcp_tools.items()
]
}
@app.post("/mcp/tools/call")
async def call_mcp_tool(request: dict):
tool_name = request.get("tool")
args = request.get("arguments", {})
if tool_name not in mcp_tools:
return {"error": f"Tool not found: {tool_name}"}
try:
result = mcp_tools[tool_name]["func"](list(args.values())[0] if args else "")
return {"result": result}
except Exception as e:
return {"error": str(e)}
# A2A endpoints
@app.get("/.well-known/agent.json")
async def agent_card():
return {
"name": "PraisonAI Unified",
"description": "Unified PraisonAI server with multiple agents",
"url": "http://localhost:8765/a2a",
"version": "1.0.0",
"capabilities": {"streaming": True}
}
@app.post("/a2a")
async def a2a_endpoint(request: dict):
method = request.get("method")
params = request.get("params", {})
req_id = request.get("id")
if method == "message/send":
message = params.get("message", {})
parts = message.get("parts", [])
text = "".join(p.get("text", "") for p in parts if p.get("type") == "text")
try:
response = assistant.chat(text)
return {
"jsonrpc": "2.0",
"id": req_id,
"result": {
"message": {
"role": "agent",
"parts": [{"type": "text", "text": response}]
}
}
}
except Exception as e:
return {"jsonrpc": "2.0", "id": req_id, "error": {"code": -32000, "message": str(e)}}
return {"jsonrpc": "2.0", "id": req_id, "error": {"code": -32601, "message": f"Unknown method: {method}"}}
# A2U endpoints
@app.get("/a2u/info")
async def a2u_info():
return {
"name": "A2U Event Stream",
"version": "1.0.0",
"streams": ["events"],
"event_types": ["agent.started", "agent.response", "agent.completed"]
}
# Root endpoint
@app.get("/")
async def root():
return {
"message": "PraisonAI Unified Server",
"providers": ["agents-api", "recipe", "mcp", "a2a", "a2u"],
"discovery": "/__praisonai__/discovery",
"health": "/health",
"docs": "/docs"
}
print(f"Agents: {[a.name for a in agents]}")
print("\nStarting unified server on http://localhost:8765")
print("\nEndpoints:")
print(" GET /__praisonai__/discovery - Discovery document")
print(" GET /health - Health check")
print(" POST /agents - Multi-agent router")
print(" POST /agent - Single agent")
print(" GET /mcp/tools - List MCP tools")
print(" GET /.well-known/agent.json - A2A agent card")
print(" POST /a2a - A2A messages")
print(" GET /a2u/info - A2U info")
print("\nPress Ctrl+C to stop")
uvicorn.run(app, host="127.0.0.1", port=8765)
if __name__ == "__main__":
main()
+2 -2
View File
@@ -3,8 +3,8 @@ class Praisonai < Formula
desc "AI tools for various AI applications"
homepage "https://github.com/MervinPraison/PraisonAI"
url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.7.3.tar.gz"
sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.7.3.tar.gz | shasum -a 256`.split.first
url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.7.4.tar.gz"
sha256 `curl -sL https://github.com/MervinPraison/PraisonAI/archive/refs/tags/v2.7.4.tar.gz | shasum -a 256`.split.first
license "MIT"
depends_on "[email protected]"
@@ -19,3 +19,4 @@ def register_all_checks():
from . import network_checks
from . import performance_checks
from . import selftest_checks
from . import serve_checks
@@ -0,0 +1,426 @@
"""
Doctor checks for serve and endpoints functionality.
Checks for:
- Serve module availability
- Endpoints module availability
- Server connectivity
- Discovery endpoint
- Provider types
"""
import os
from ..models import CheckResult, CheckStatus, CheckCategory, CheckSeverity, DoctorConfig
from ..registry import register_check
@register_check(
id="serve_module",
title="Serve Module",
description="Check if serve module is available",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.MEDIUM,
tags=["serve", "deploy"],
)
def check_serve_module(config: DoctorConfig) -> CheckResult:
"""Check if serve module is available."""
try:
from praisonai.cli.features.serve import ServeHandler
ServeHandler() # Test instantiation
return CheckResult(
id="serve_module",
title="Serve Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message="Serve module available",
metadata={"handler": "ServeHandler"},
)
except ImportError as e:
return CheckResult(
id="serve_module",
title="Serve Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message=f"Serve module not available: {e}",
remediation="pip install praisonai[serve]",
)
except Exception as e:
return CheckResult(
id="serve_module",
title="Serve Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.ERROR,
message=f"Error checking serve module: {e}",
)
@register_check(
id="endpoints_module",
title="Endpoints Module",
description="Check if endpoints module is available",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.MEDIUM,
tags=["endpoints", "deploy"],
)
def check_endpoints_module(config: DoctorConfig) -> CheckResult:
"""Check if endpoints module is available."""
try:
from praisonai.endpoints import (
DiscoveryDocument,
EndpointInfo,
ProviderInfo,
list_provider_types,
)
types = list_provider_types()
return CheckResult(
id="endpoints_module",
title="Endpoints Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message=f"Endpoints module available ({len(types)} provider types)",
metadata={"provider_types": types},
)
except ImportError as e:
return CheckResult(
id="endpoints_module",
title="Endpoints Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message=f"Endpoints module not available: {e}",
remediation="pip install praisonai[serve]",
)
except Exception as e:
return CheckResult(
id="endpoints_module",
title="Endpoints Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.ERROR,
message=f"Error checking endpoints module: {e}",
)
@register_check(
id="endpoints_cli",
title="Endpoints CLI",
description="Check if endpoints CLI handler is available",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.LOW,
tags=["endpoints", "cli"],
)
def check_endpoints_cli(config: DoctorConfig) -> CheckResult:
"""Check if endpoints CLI handler is available."""
try:
from praisonai.cli.features.endpoints import EndpointsHandler
handler = EndpointsHandler()
return CheckResult(
id="endpoints_cli",
title="Endpoints CLI",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message="Endpoints CLI handler available",
metadata={"default_url": handler.DEFAULT_URL},
)
except ImportError as e:
return CheckResult(
id="endpoints_cli",
title="Endpoints CLI",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message=f"Endpoints CLI not available: {e}",
remediation="pip install praisonai",
)
except Exception as e:
return CheckResult(
id="endpoints_cli",
title="Endpoints CLI",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.ERROR,
message=f"Error checking endpoints CLI: {e}",
)
@register_check(
id="server_connectivity",
title="Server Connectivity",
description="Check if a PraisonAI server is reachable",
category=CheckCategory.NETWORK,
severity=CheckSeverity.LOW,
requires_deep=True,
tags=["serve", "network"],
)
def check_server_connectivity(config: DoctorConfig) -> CheckResult:
"""Check if a PraisonAI server is reachable."""
import urllib.request
import urllib.error
url = os.environ.get("PRAISONAI_ENDPOINTS_URL", "http://localhost:8765")
health_url = f"{url}/health"
try:
req = urllib.request.Request(health_url, method="GET")
with urllib.request.urlopen(req, timeout=5) as resp:
if resp.status == 200:
return CheckResult(
id="server_connectivity",
title="Server Connectivity",
category=CheckCategory.NETWORK,
status=CheckStatus.PASS,
message=f"Server reachable at {url}",
metadata={"url": url, "status": resp.status},
)
else:
return CheckResult(
id="server_connectivity",
title="Server Connectivity",
category=CheckCategory.NETWORK,
status=CheckStatus.WARN,
message=f"Server returned status {resp.status}",
metadata={"url": url, "status": resp.status},
)
except urllib.error.URLError as e:
return CheckResult(
id="server_connectivity",
title="Server Connectivity",
category=CheckCategory.NETWORK,
status=CheckStatus.SKIP,
message=f"No server running at {url}",
details=str(e.reason),
remediation="Start a server: praisonai serve unified --port 8765",
)
except Exception as e:
return CheckResult(
id="server_connectivity",
title="Server Connectivity",
category=CheckCategory.NETWORK,
status=CheckStatus.ERROR,
message=f"Error checking server: {e}",
)
@register_check(
id="discovery_endpoint",
title="Discovery Endpoint",
description="Check if discovery endpoint is available",
category=CheckCategory.NETWORK,
severity=CheckSeverity.LOW,
requires_deep=True,
dependencies=["server_connectivity"],
tags=["serve", "discovery"],
)
def check_discovery_endpoint(config: DoctorConfig) -> CheckResult:
"""Check if discovery endpoint is available."""
import urllib.request
import urllib.error
import json
url = os.environ.get("PRAISONAI_ENDPOINTS_URL", "http://localhost:8765")
discovery_url = f"{url}/__praisonai__/discovery"
try:
req = urllib.request.Request(discovery_url, method="GET")
with urllib.request.urlopen(req, timeout=5) as resp:
if resp.status == 200:
data = json.loads(resp.read().decode())
providers = data.get("providers", [])
endpoints = data.get("endpoints", [])
return CheckResult(
id="discovery_endpoint",
title="Discovery Endpoint",
category=CheckCategory.NETWORK,
status=CheckStatus.PASS,
message=f"Discovery available: {len(providers)} providers, {len(endpoints)} endpoints",
metadata={
"url": discovery_url,
"schema_version": data.get("schema_version"),
"server_name": data.get("server_name"),
"providers": [p.get("type") for p in providers],
},
)
else:
return CheckResult(
id="discovery_endpoint",
title="Discovery Endpoint",
category=CheckCategory.NETWORK,
status=CheckStatus.WARN,
message=f"Discovery returned status {resp.status}",
)
except urllib.error.URLError:
return CheckResult(
id="discovery_endpoint",
title="Discovery Endpoint",
category=CheckCategory.NETWORK,
status=CheckStatus.SKIP,
message="No server running (skipped)",
)
except Exception as e:
return CheckResult(
id="discovery_endpoint",
title="Discovery Endpoint",
category=CheckCategory.NETWORK,
status=CheckStatus.ERROR,
message=f"Error checking discovery: {e}",
)
@register_check(
id="a2u_module",
title="A2U Module",
description="Check if A2U server module is available",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.LOW,
tags=["serve", "a2u"],
)
def check_a2u_module(config: DoctorConfig) -> CheckResult:
"""Check if A2U server module is available."""
try:
from praisonai.endpoints.a2u_server import (
A2UEvent,
A2USubscription,
A2UEventBus,
create_a2u_routes,
)
return CheckResult(
id="a2u_module",
title="A2U Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message="A2U server module available",
)
except ImportError as e:
return CheckResult(
id="a2u_module",
title="A2U Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message=f"A2U module not available: {e}",
remediation="pip install praisonai[serve]",
)
except Exception as e:
return CheckResult(
id="a2u_module",
title="A2U Module",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.ERROR,
message=f"Error checking A2U module: {e}",
)
@register_check(
id="provider_adapters",
title="Provider Adapters",
description="Check if all provider adapters are available",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.LOW,
tags=["serve", "providers"],
)
def check_provider_adapters(config: DoctorConfig) -> CheckResult:
"""Check if all provider adapters are available."""
try:
from praisonai.endpoints.providers import (
BaseProvider,
RecipeProvider,
AgentsAPIProvider,
MCPProvider,
ToolsMCPProvider,
A2AProvider,
A2UProvider,
)
providers = [
"BaseProvider",
"RecipeProvider",
"AgentsAPIProvider",
"MCPProvider",
"ToolsMCPProvider",
"A2AProvider",
"A2UProvider",
]
return CheckResult(
id="provider_adapters",
title="Provider Adapters",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message=f"All {len(providers)} provider adapters available",
metadata={"providers": providers},
)
except ImportError as e:
return CheckResult(
id="provider_adapters",
title="Provider Adapters",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message=f"Some provider adapters not available: {e}",
remediation="pip install praisonai[serve]",
)
except Exception as e:
return CheckResult(
id="provider_adapters",
title="Provider Adapters",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.ERROR,
message=f"Error checking provider adapters: {e}",
)
@register_check(
id="fastapi_available",
title="FastAPI Available",
description="Check if FastAPI is available for serving",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.MEDIUM,
tags=["serve", "dependencies"],
)
def check_fastapi_available(config: DoctorConfig) -> CheckResult:
"""Check if FastAPI is available."""
try:
import fastapi
version = getattr(fastapi, "__version__", "unknown")
return CheckResult(
id="fastapi_available",
title="FastAPI Available",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message=f"FastAPI {version} available",
metadata={"version": version},
)
except ImportError:
return CheckResult(
id="fastapi_available",
title="FastAPI Available",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message="FastAPI not installed",
remediation="pip install fastapi uvicorn",
)
@register_check(
id="uvicorn_available",
title="Uvicorn Available",
description="Check if Uvicorn is available for serving",
category=CheckCategory.ENVIRONMENT,
severity=CheckSeverity.MEDIUM,
tags=["serve", "dependencies"],
)
def check_uvicorn_available(config: DoctorConfig) -> CheckResult:
"""Check if Uvicorn is available."""
try:
import uvicorn
version = getattr(uvicorn, "__version__", "unknown")
return CheckResult(
id="uvicorn_available",
title="Uvicorn Available",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.PASS,
message=f"Uvicorn {version} available",
metadata={"version": version},
)
except ImportError:
return CheckResult(
id="uvicorn_available",
title="Uvicorn Available",
category=CheckCategory.ENVIRONMENT,
status=CheckStatus.WARN,
message="Uvicorn not installed",
remediation="pip install uvicorn",
)
+520 -79
View File
@@ -1,11 +1,20 @@
"""
Endpoints CLI Feature Handler
Provides CLI commands for interacting with recipe endpoints:
- list: List available recipe endpoints
Provides CLI commands for interacting with PraisonAI endpoints:
- list: List available endpoints (all provider types)
- describe: Show endpoint details and schema
- invoke: Call an endpoint
- health: Check endpoint server health
- types: List supported provider types
Supported Provider Types:
- recipe: Recipe runner endpoints
- agents-api: Single/multi-agent HTTP API
- mcp: MCP server (stdio, http, sse)
- tools-mcp: Tools exposed as MCP server
- a2a: Agent-to-agent protocol
- a2u: Agent-to-user event stream
All commands use the canonical `praisonai endpoints` prefix.
@@ -14,11 +23,13 @@ Why this feature is valuable:
- Ops: Health checks, monitoring, automation
- Polyglot: Non-Python clients can invoke recipes via HTTP
- Testing: Easy endpoint verification without code
- Unified: Single interface for all server types
Architecture notes:
- Optional extras only (no server deps in core)
- Lazy imports for all HTTP client code
- No impact on praisonaiagents import time
- Backward compatible with existing recipe-only usage
"""
import json
@@ -27,15 +38,27 @@ import sys
from typing import Any, Dict, List, Optional
# Supported provider types
PROVIDER_TYPES = [
"recipe",
"agents-api",
"mcp",
"tools-mcp",
"a2a",
"a2u",
]
class EndpointsHandler:
"""
CLI handler for endpoints operations.
Commands:
- list: List available recipe endpoints
- list: List available endpoints (all provider types)
- describe: Show endpoint details
- invoke: Call an endpoint
- health: Check server health
- types: List supported provider types
"""
# Exit codes
@@ -90,6 +113,8 @@ class EndpointsHandler:
"describe": self.cmd_describe,
"invoke": self.cmd_invoke,
"health": self.cmd_health,
"types": self.cmd_types,
"discovery": self.cmd_discovery,
"help": lambda _: self._print_help() or self.EXIT_SUCCESS,
"--help": lambda _: self._print_help() or self.EXIT_SUCCESS,
"-h": lambda _: self._print_help() or self.EXIT_SUCCESS,
@@ -107,25 +132,30 @@ class EndpointsHandler:
help_text = """
[bold cyan]PraisonAI Endpoints[/bold cyan]
Client CLI for interacting with recipe endpoints.
Unified client CLI for interacting with PraisonAI endpoints.
[bold]Usage:[/bold]
praisonai endpoints <command> [options]
[bold]Commands:[/bold]
list List available recipe endpoints
describe <recipe> Show endpoint details and schema
invoke <recipe> Call an endpoint
list List available endpoints (all types)
describe <name> Show endpoint details and schema
invoke <name> Call an endpoint
health Check server health
types List supported provider types
discovery Show raw discovery document
[bold]Global Options:[/bold]
--url <url> Server URL (default: http://localhost:8765)
--api-key <key> API key (or set PRAISONAI_ENDPOINTS_API_KEY)
--type <type> Filter by provider type (recipe, agents-api, mcp, etc.)
[bold]List Options:[/bold]
--format json Output as JSON
--tags <a,b> Filter by tags (comma-separated)
--url <url> Server URL (default: http://localhost:8765)
[bold]Describe Options:[/bold]
--schema Show input/output schema only
--url <url> Server URL
[bold]Invoke Options:[/bold]
--input <path> Input file path
@@ -133,12 +163,15 @@ Client CLI for interacting with recipe endpoints.
--config k=v Config override (repeatable)
--json Output as JSON
--stream Stream output events (SSE)
--url <url> Server URL
--api-key <key> API key for auth (or set PRAISONAI_ENDPOINTS_API_KEY)
--dry-run Validate without executing
[bold]Health Options:[/bold]
--url <url> Server URL
[bold]Provider Types:[/bold]
recipe Recipe runner endpoints
agents-api Single/multi-agent HTTP API
mcp MCP server (stdio, http, sse)
tools-mcp Tools exposed as MCP server
a2a Agent-to-agent protocol
a2u Agent-to-user event stream
[bold]Environment Variables:[/bold]
PRAISONAI_ENDPOINTS_URL Default server URL
@@ -146,14 +179,16 @@ Client CLI for interacting with recipe endpoints.
[bold]Examples:[/bold]
praisonai endpoints list
praisonai endpoints list --type agents-api
praisonai endpoints list --format json --tags audio,video
praisonai endpoints describe ai-podcast-cleaner
praisonai endpoints describe ai-podcast-cleaner --schema
praisonai endpoints invoke ai-podcast-cleaner --input ./audio.mp3
praisonai endpoints invoke ai-podcast-cleaner --input-json '{"text": "hello"}'
praisonai endpoints invoke ai-podcast-cleaner --input ./audio.mp3 --stream
praisonai endpoints describe my-agent
praisonai endpoints describe my-agent --schema
praisonai endpoints invoke my-agent --input-json '{"query": "hello"}'
praisonai endpoints invoke my-recipe --input ./data.json --stream
praisonai endpoints health
praisonai endpoints health --url http://localhost:8000
praisonai endpoints types
praisonai endpoints discovery
"""
self._print_rich(help_text)
@@ -339,34 +374,96 @@ Client CLI for interacting with recipe endpoints.
except Exception as e:
yield {"error": str(e)}
def _get_provider(self, provider_type: str, url: Optional[str] = None, api_key: Optional[str] = None):
"""Get a provider instance."""
try:
from praisonai.endpoints import get_provider
return get_provider(
provider_type,
base_url=url or self.base_url,
api_key=api_key or self.api_key,
)
except ImportError:
return None
def _try_unified_discovery(self, url: Optional[str] = None) -> Optional[Dict[str, Any]]:
"""Try to get unified discovery document."""
result = self._make_request("GET", "/__praisonai__/discovery", url=url)
if not result.get("error") and result.get("data"):
return result.get("data")
return None
def cmd_list(self, args: List[str]) -> int:
"""List available recipe endpoints."""
"""List available endpoints (all provider types)."""
spec = {
"format": {"default": "table"},
"tags": {"default": None},
"url": {"default": None},
"type": {"default": None},
}
parsed = self._parse_args(args, spec)
# Build query params
path = "/v1/recipes"
if parsed["tags"]:
path += f"?tags={parsed['tags']}"
all_endpoints = []
result = self._make_request("GET", path, url=parsed["url"])
# Try unified discovery first
discovery = self._try_unified_discovery(parsed["url"])
if discovery:
endpoints = discovery.get("endpoints", [])
for ep in endpoints:
# Filter by type if specified
if parsed["type"] and ep.get("provider_type") != parsed["type"]:
continue
# Filter by tags if specified
if parsed["tags"]:
tag_list = [t.strip() for t in parsed["tags"].split(",")]
ep_tags = ep.get("tags", [])
if not any(t in ep_tags for t in tag_list):
continue
all_endpoints.append(ep)
else:
# Fallback: try recipe endpoint (backward compatibility)
if not parsed["type"] or parsed["type"] == "recipe":
path = "/v1/recipes"
if parsed["tags"]:
path += f"?tags={parsed['tags']}"
result = self._make_request("GET", path, url=parsed["url"])
if not result.get("error"):
recipes = result.get("data", {}).get("recipes", [])
for r in recipes:
all_endpoints.append({
"name": r.get("name", ""),
"description": r.get("description", ""),
"provider_type": "recipe",
"version": r.get("version", "1.0.0"),
"tags": r.get("tags", []),
"streaming": ["none", "sse"],
})
# Try agents-api endpoint
if not parsed["type"] or parsed["type"] == "agents-api":
result = self._make_request("GET", "/", url=parsed["url"])
if not result.get("error"):
data = result.get("data", {})
for path in data.get("endpoints", []):
all_endpoints.append({
"name": path.lstrip("/"),
"description": f"Agent endpoint at {path}",
"provider_type": "agents-api",
"streaming": ["none"],
})
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
recipes = result.get("data", {}).get("recipes", [])
if parsed["format"] == "json":
self._print_json(recipes)
if not all_endpoints:
result = self._make_request("GET", "/health", url=parsed["url"])
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
print("No endpoints available.")
return self.EXIT_SUCCESS
if not recipes:
print("No endpoints available.")
if parsed["format"] == "json":
self._print_json(all_endpoints)
return self.EXIT_SUCCESS
try:
@@ -376,63 +473,106 @@ Client CLI for interacting with recipe endpoints.
console = Console()
table = Table(title="Available Endpoints")
table.add_column("Name", style="cyan")
table.add_column("Version", style="green")
table.add_column("Type", style="magenta")
table.add_column("Description")
table.add_column("Streaming", style="green")
table.add_column("Tags", style="yellow")
for r in recipes:
tags = r.get("tags", [])
for ep in all_endpoints:
tags = ep.get("tags", [])
streaming = ep.get("streaming", ["none"])
desc = ep.get("description", "")
table.add_row(
r.get("name", ""),
r.get("version", ""),
(r.get("description", "")[:50] + "...") if len(r.get("description", "")) > 50 else r.get("description", ""),
ep.get("name", ""),
ep.get("provider_type", "unknown"),
(desc[:40] + "...") if len(desc) > 40 else desc,
", ".join(streaming[:2]),
", ".join(tags[:3]) if tags else "",
)
console.print(table)
except ImportError:
for r in recipes:
print(f"{r.get('name')} ({r.get('version')}): {r.get('description', '')}")
for ep in all_endpoints:
print(f"[{ep.get('provider_type', 'unknown')}] {ep.get('name')}: {ep.get('description', '')}")
return self.EXIT_SUCCESS
def cmd_describe(self, args: List[str]) -> int:
"""Describe a recipe endpoint."""
"""Describe an endpoint."""
spec = {
"recipe": {"positional": True, "default": ""},
"name": {"positional": True, "default": ""},
"schema": {"flag": True, "default": False},
"url": {"default": None},
"type": {"default": None},
}
parsed = self._parse_args(args, spec)
if not parsed["recipe"]:
self._print_error("Recipe name required")
if not parsed["name"]:
self._print_error("Endpoint name required")
return self.EXIT_VALIDATION_ERROR
# Get schema if requested
if parsed["schema"]:
path = f"/v1/recipes/{parsed['recipe']}/schema"
else:
path = f"/v1/recipes/{parsed['recipe']}"
endpoint_name = parsed["name"]
endpoint_info = None
result = self._make_request("GET", path, url=parsed["url"])
# Try unified discovery first
discovery = self._try_unified_discovery(parsed["url"])
if discovery:
for ep in discovery.get("endpoints", []):
if ep.get("name") == endpoint_name:
if parsed["type"] and ep.get("provider_type") != parsed["type"]:
continue
endpoint_info = ep
break
if result.get("status") == 404:
self._print_error(f"Endpoint not found: {parsed['recipe']}")
return self.EXIT_NOT_FOUND
if endpoint_info:
if parsed["schema"]:
schema_info = {
"name": endpoint_info.get("name"),
"input_schema": endpoint_info.get("input_schema"),
"output_schema": endpoint_info.get("output_schema"),
}
self._print_json(schema_info)
else:
self._print_json(endpoint_info)
return self.EXIT_SUCCESS
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
# Fallback: try recipe endpoint (backward compatibility)
if not parsed["type"] or parsed["type"] == "recipe":
if parsed["schema"]:
path = f"/v1/recipes/{endpoint_name}/schema"
else:
path = f"/v1/recipes/{endpoint_name}"
result = self._make_request("GET", path, url=parsed["url"])
if not result.get("error") and result.get("status") != 404:
data = result.get("data", {})
# Add provider_type for consistency
data["provider_type"] = "recipe"
self._print_json(data)
return self.EXIT_SUCCESS
data = result.get("data", {})
self._print_json(data)
return self.EXIT_SUCCESS
# Try agents-api
if not parsed["type"] or parsed["type"] == "agents-api":
# Return basic info for agent endpoints
endpoint_info = {
"name": endpoint_name,
"provider_type": "agents-api",
"description": f"Agent endpoint: {endpoint_name}",
"input_schema": {"type": "object", "properties": {"query": {"type": "string"}}},
"streaming": ["none"],
"auth_modes": ["none"],
}
self._print_json(endpoint_info)
return self.EXIT_SUCCESS
self._print_error(f"Endpoint not found: {endpoint_name}")
return self.EXIT_NOT_FOUND
def cmd_invoke(self, args: List[str]) -> int:
"""Invoke a recipe endpoint."""
"""Invoke an endpoint."""
spec = {
"recipe": {"positional": True, "default": ""},
"name": {"positional": True, "default": ""},
"input": {"short": "-i", "default": None},
"input_json": {"default": None},
"config": {"repeatable": True, "default": []},
@@ -441,13 +581,16 @@ Client CLI for interacting with recipe endpoints.
"url": {"default": None},
"api_key": {"default": None},
"dry_run": {"flag": True, "default": False},
"type": {"default": None},
}
parsed = self._parse_args(args, spec)
if not parsed["recipe"]:
self._print_error("Recipe name required")
if not parsed["name"]:
self._print_error("Endpoint name required")
return self.EXIT_VALIDATION_ERROR
endpoint_name = parsed["name"]
# Build input data
input_data = {}
if parsed["input_json"]:
@@ -457,7 +600,15 @@ Client CLI for interacting with recipe endpoints.
self._print_error("Invalid JSON in --input-json")
return self.EXIT_VALIDATION_ERROR
elif parsed["input"]:
input_data = {"input": parsed["input"]}
# Check if input is a file path
if os.path.isfile(parsed["input"]):
try:
with open(parsed["input"]) as f:
input_data = json.load(f)
except (json.JSONDecodeError, IOError):
input_data = {"input": parsed["input"]}
else:
input_data = {"input": parsed["input"]}
# Build config from --config key=value pairs
config = {}
@@ -466,9 +617,39 @@ Client CLI for interacting with recipe endpoints.
k, v = cfg.split("=", 1)
config[k] = v
# Build request body
# Detect provider type from discovery or explicit --type
provider_type = parsed["type"]
if not provider_type:
discovery = self._try_unified_discovery(parsed["url"])
if discovery:
for ep in discovery.get("endpoints", []):
if ep.get("name") == endpoint_name:
provider_type = ep.get("provider_type")
break
# Default to recipe for backward compatibility
if not provider_type:
provider_type = "recipe"
# Route to appropriate invocation method
if provider_type == "recipe":
return self._invoke_recipe(endpoint_name, input_data, config, parsed)
elif provider_type == "agents-api":
return self._invoke_agents_api(endpoint_name, input_data, config, parsed)
elif provider_type in ("mcp", "tools-mcp"):
return self._invoke_mcp(endpoint_name, input_data, config, parsed)
elif provider_type == "a2a":
return self._invoke_a2a(endpoint_name, input_data, config, parsed)
elif provider_type == "a2u":
return self._invoke_a2u(endpoint_name, input_data, config, parsed)
else:
# Fallback to recipe
return self._invoke_recipe(endpoint_name, input_data, config, parsed)
def _invoke_recipe(self, name: str, input_data: Dict, config: Dict, parsed: Dict) -> int:
"""Invoke a recipe endpoint."""
body = {
"recipe": parsed["recipe"],
"recipe": name,
"input": input_data,
"config": config,
"options": {"dry_run": parsed["dry_run"]},
@@ -477,7 +658,6 @@ Client CLI for interacting with recipe endpoints.
if parsed["stream"]:
return self._invoke_stream(body, parsed)
# Sync invocation
result = self._make_request(
"POST",
"/v1/recipes/run",
@@ -486,12 +666,31 @@ Client CLI for interacting with recipe endpoints.
json_data=body,
)
return self._handle_invoke_result(result, name, parsed)
def _invoke_agents_api(self, name: str, input_data: Dict, config: Dict, parsed: Dict) -> int:
"""Invoke an agents-api endpoint."""
path = name if name.startswith("/") else f"/{name}"
# Build body - agents-api expects query field
body = input_data
if "query" not in body and config.get("query"):
body["query"] = config["query"]
result = self._make_request(
"POST",
path,
url=parsed["url"],
api_key=parsed["api_key"],
json_data=body,
)
if result.get("status") == 401:
self._print_error("Authentication required. Use --api-key or set PRAISONAI_ENDPOINTS_API_KEY")
self._print_error("Authentication required")
return self.EXIT_AUTH_ERROR
if result.get("status") == 404:
self._print_error(f"Endpoint not found: {parsed['recipe']}")
self._print_error(f"Endpoint not found: {name}")
return self.EXIT_NOT_FOUND
if result.get("error"):
@@ -503,17 +702,171 @@ Client CLI for interacting with recipe endpoints.
if parsed["json"]:
self._print_json(data)
else:
if data.get("ok"):
self._print_success(f"Endpoint '{parsed['recipe']}' invoked successfully")
print(f" Run ID: {data.get('run_id')}")
print(f" Status: {data.get('status')}")
response = data.get("response", data)
self._print_success(f"Endpoint '{name}' invoked successfully")
print(f" Response: {response}")
return self.EXIT_SUCCESS
def _invoke_mcp(self, name: str, input_data: Dict, config: Dict, parsed: Dict) -> int:
"""Invoke an MCP tool."""
body = {
"tool": name,
"arguments": input_data,
}
result = self._make_request(
"POST",
"/mcp/tools/call",
url=parsed["url"],
api_key=parsed["api_key"],
json_data=body,
)
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
data = result.get("data", {})
if parsed["json"]:
self._print_json(data)
else:
self._print_success(f"Tool '{name}' invoked successfully")
print(f" Result: {data.get('result', data)}")
return self.EXIT_SUCCESS
def _invoke_a2a(self, name: str, input_data: Dict, config: Dict, parsed: Dict) -> int:
"""Invoke an A2A agent."""
import uuid
message = input_data.get("message", input_data.get("query", ""))
if not message and config.get("message"):
message = config["message"]
body = {
"jsonrpc": "2.0",
"method": "message/send",
"id": str(uuid.uuid4()),
"params": {
"message": {
"role": "user",
"parts": [{"type": "text", "text": message}],
},
},
}
result = self._make_request(
"POST",
"/a2a",
url=parsed["url"],
api_key=parsed["api_key"],
json_data=body,
)
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
data = result.get("data", {})
if parsed["json"]:
self._print_json(data)
else:
if "result" in data:
self._print_success(f"A2A message sent to '{name}'")
print(f" Result: {data['result']}")
elif "error" in data:
self._print_error(data["error"].get("message", str(data["error"])))
return self.EXIT_RUNTIME_ERROR
return self.EXIT_SUCCESS
def _invoke_a2u(self, name: str, input_data: Dict, config: Dict, parsed: Dict) -> int:
"""Subscribe to an A2U event stream."""
if parsed["stream"]:
return self._stream_a2u(name, parsed)
body = {
"stream": name,
"filters": input_data.get("filters", []),
}
result = self._make_request(
"POST",
"/a2u/subscribe",
url=parsed["url"],
api_key=parsed["api_key"],
json_data=body,
)
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
data = result.get("data", {})
if parsed["json"]:
self._print_json(data)
else:
self._print_success(f"Subscribed to A2U stream '{name}'")
print(f" Stream URL: {data.get('stream_url')}")
return self.EXIT_SUCCESS
def _stream_a2u(self, name: str, parsed: Dict) -> int:
"""Stream A2U events."""
print(f"Streaming events from '{name}'...")
for event in self._stream_request(
f"/a2u/events/{name}",
url=parsed["url"],
api_key=parsed["api_key"],
):
if event.get("error"):
self._print_error(event["error"])
return self.EXIT_CONNECTION_ERROR
if parsed["json"]:
print(json.dumps(event))
else:
print(f" [{event.get('event', 'event')}] {event.get('data', '')}")
return self.EXIT_SUCCESS
def _handle_invoke_result(self, result: Dict, name: str, parsed: Dict) -> int:
"""Handle common invoke result processing."""
if result.get("status") == 401:
self._print_error("Authentication required. Use --api-key or set PRAISONAI_ENDPOINTS_API_KEY")
return self.EXIT_AUTH_ERROR
if result.get("status") == 404:
self._print_error(f"Endpoint not found: {name}")
return self.EXIT_NOT_FOUND
if result.get("error"):
self._print_error(result["error"].get("message", str(result["error"])))
return self.EXIT_CONNECTION_ERROR
data = result.get("data", {})
if parsed["json"]:
self._print_json(data)
else:
if data.get("ok", True):
self._print_success(f"Endpoint '{name}' invoked successfully")
if data.get("run_id"):
print(f" Run ID: {data.get('run_id')}")
if data.get("status"):
print(f" Status: {data.get('status')}")
if data.get("output"):
print(f" Output: {data.get('output')}")
else:
self._print_error(f"Invocation failed: {data.get('error')}")
return self.EXIT_RUNTIME_ERROR
return self.EXIT_SUCCESS if data.get("ok") else self.EXIT_RUNTIME_ERROR
return self.EXIT_SUCCESS if data.get("ok", True) else self.EXIT_RUNTIME_ERROR
def _invoke_stream(self, body: Dict, parsed: Dict) -> int:
"""Handle streaming invocation."""
@@ -551,9 +904,14 @@ Client CLI for interacting with recipe endpoints.
"""Check endpoint server health."""
spec = {
"url": {"default": None},
"type": {"default": None},
"format": {"default": "table"},
}
parsed = self._parse_args(args, spec)
# Try unified discovery first for richer info
discovery = self._try_unified_discovery(parsed["url"])
result = self._make_request("GET", "/health", url=parsed["url"])
if result.get("error"):
@@ -562,14 +920,97 @@ Client CLI for interacting with recipe endpoints.
data = result.get("data", {})
if data.get("status") == "healthy":
# Merge discovery info
if discovery:
data["schema_version"] = discovery.get("schema_version")
data["providers"] = [p.get("type") for p in discovery.get("providers", [])]
data["endpoint_count"] = len(discovery.get("endpoints", []))
if parsed["format"] == "json":
self._print_json(data)
return self.EXIT_SUCCESS
status = data.get("status", "unknown")
if status in ("healthy", "ok"):
self._print_success("Server healthy")
print(f" Service: {data.get('service')}")
print(f" Version: {data.get('version')}")
if data.get("service"):
print(f" Service: {data.get('service')}")
if data.get("server_name"):
print(f" Server: {data.get('server_name')}")
if data.get("version"):
print(f" Version: {data.get('version')}")
if data.get("schema_version"):
print(f" Schema Version: {data.get('schema_version')}")
if data.get("providers"):
print(f" Providers: {', '.join(data.get('providers'))}")
if data.get("endpoint_count"):
print(f" Endpoints: {data.get('endpoint_count')}")
if data.get("endpoints"):
print(f" Endpoints: {', '.join(data.get('endpoints')[:5])}")
return self.EXIT_SUCCESS
else:
self._print_error(f"Server unhealthy: {data}")
return self.EXIT_RUNTIME_ERROR
def cmd_types(self, args: List[str]) -> int:
"""List supported provider types."""
spec = {
"format": {"default": "table"},
}
parsed = self._parse_args(args, spec)
types_info = [
{"type": "recipe", "description": "Recipe runner endpoints", "capabilities": ["list", "describe", "invoke", "stream"]},
{"type": "agents-api", "description": "Single/multi-agent HTTP API", "capabilities": ["list", "describe", "invoke"]},
{"type": "mcp", "description": "MCP server (stdio, http, sse)", "capabilities": ["list-tools", "call-tool"]},
{"type": "tools-mcp", "description": "Tools exposed as MCP server", "capabilities": ["list-tools", "call-tool"]},
{"type": "a2a", "description": "Agent-to-agent protocol", "capabilities": ["agent-card", "message-send", "stream"]},
{"type": "a2u", "description": "Agent-to-user event stream", "capabilities": ["subscribe", "stream"]},
]
if parsed["format"] == "json":
self._print_json(types_info)
return self.EXIT_SUCCESS
try:
from rich.console import Console
from rich.table import Table
console = Console()
table = Table(title="Supported Provider Types")
table.add_column("Type", style="cyan")
table.add_column("Description")
table.add_column("Capabilities", style="green")
for t in types_info:
table.add_row(
t["type"],
t["description"],
", ".join(t["capabilities"]),
)
console.print(table)
except ImportError:
for t in types_info:
print(f"{t['type']}: {t['description']}")
return self.EXIT_SUCCESS
def cmd_discovery(self, args: List[str]) -> int:
"""Show raw discovery document."""
spec = {
"url": {"default": None},
}
parsed = self._parse_args(args, spec)
discovery = self._try_unified_discovery(parsed["url"])
if discovery:
self._print_json(discovery)
return self.EXIT_SUCCESS
self._print_error("Discovery endpoint not available at this server")
return self.EXIT_NOT_FOUND
def handle_endpoints_command(args: List[str]) -> int:
@@ -0,0 +1,449 @@
"""
Recipes CLI - Commands for managing and running Agent-Recipes.
Provides commands for:
- praison recipes list - List available recipes
- praison recipes info <name> - Show recipe details
- praison recipes run <name> <input> - Run a recipe
- praison recipes doctor <name> - Check recipe dependencies
- praison recipes explain <name> - Show recipe execution plan
- praison recipes init <name> - Initialize recipe in current directory
"""
import os
import sys
import json
import yaml
import argparse
import logging
from pathlib import Path
from typing import Optional, List, Dict, Any
logger = logging.getLogger(__name__)
# Recipe discovery paths in order of precedence
RECIPE_PATHS = [
Path.home() / ".praison" / "templates",
Path.home() / ".config" / "praison" / "templates",
Path.cwd() / ".praison" / "templates",
Path("/Users/praison/Agent-Recipes/agent_recipes/templates"),
]
def find_recipe_paths() -> List[Path]:
"""Find all valid recipe directories."""
paths = []
for p in RECIPE_PATHS:
if p.exists() and p.is_dir():
paths.append(p)
return paths
def list_recipes() -> List[Dict[str, Any]]:
"""List all available recipes."""
recipes = []
seen = set()
for base_path in find_recipe_paths():
for recipe_dir in base_path.iterdir():
if not recipe_dir.is_dir():
continue
template_file = recipe_dir / "TEMPLATE.yaml"
if not template_file.exists():
continue
name = recipe_dir.name
if name in seen:
continue
seen.add(name)
try:
with open(template_file) as f:
template = yaml.safe_load(f)
recipes.append({
"name": name,
"description": template.get("description", ""),
"version": template.get("version", "1.0.0"),
"tags": template.get("tags", []),
"path": str(recipe_dir),
})
except Exception as e:
logger.warning(f"Failed to load recipe {name}: {e}")
return sorted(recipes, key=lambda x: x["name"])
def get_recipe(name: str) -> Optional[Dict[str, Any]]:
"""Get recipe by name."""
for base_path in find_recipe_paths():
recipe_dir = base_path / name
template_file = recipe_dir / "TEMPLATE.yaml"
if template_file.exists():
with open(template_file) as f:
template = yaml.safe_load(f)
template["path"] = str(recipe_dir)
return template
return None
def check_dependencies(recipe: Dict[str, Any]) -> Dict[str, bool]:
"""Check if recipe dependencies are satisfied."""
results = {}
requires = recipe.get("requires", {})
# Check environment variables
for env_var in requires.get("env", []):
results[f"env:{env_var}"] = bool(os.environ.get(env_var))
# Check Python packages
for package in requires.get("packages", []):
try:
__import__(package.replace("-", "_").split("[")[0])
results[f"package:{package}"] = True
except ImportError:
results[f"package:{package}"] = False
# Check external tools
import shutil
for tool in requires.get("external", []):
results[f"external:{tool}"] = shutil.which(tool) is not None
# Check recipe tools
for tool in requires.get("tools", []):
try:
from praisonai_tools.recipe_tools import __all__ as available_tools
# Check if tool class exists
tool_class = tool.replace("_tool", "").title() + "Tool"
results[f"tool:{tool}"] = tool_class in available_tools or tool in str(available_tools)
except ImportError:
results[f"tool:{tool}"] = False
return results
def cmd_list(args: argparse.Namespace) -> int:
"""List available recipes."""
recipes = list_recipes()
if args.json:
print(json.dumps(recipes, indent=2))
return 0
if not recipes:
print("No recipes found.")
return 0
# Group by first tag if available
if args.group:
groups = {}
for r in recipes:
tag = r["tags"][0] if r["tags"] else "other"
if tag not in groups:
groups[tag] = []
groups[tag].append(r)
for group, group_recipes in sorted(groups.items()):
print(f"\n{group.upper()}:")
for r in group_recipes:
print(f" {r['name']:<35} {r['description'][:50]}")
else:
print(f"{'Recipe':<35} {'Description':<50} {'Version'}")
print("-" * 95)
for r in recipes:
desc = r["description"][:47] + "..." if len(r["description"]) > 50 else r["description"]
print(f"{r['name']:<35} {desc:<50} {r['version']}")
print(f"\nTotal: {len(recipes)} recipes")
return 0
def cmd_info(args: argparse.Namespace) -> int:
"""Show recipe details."""
recipe = get_recipe(args.name)
if not recipe:
print(f"Recipe not found: {args.name}")
return 1
if args.json:
print(json.dumps(recipe, indent=2, default=str))
return 0
print(f"Name: {recipe.get('name', args.name)}")
print(f"Version: {recipe.get('version', '1.0.0')}")
print(f"Description: {recipe.get('description', '')}")
print(f"Author: {recipe.get('author', 'unknown')}")
print(f"License: {recipe.get('license', 'unknown')}")
print(f"Path: {recipe.get('path', '')}")
if recipe.get("tags"):
print(f"Tags: {', '.join(recipe['tags'])}")
requires = recipe.get("requires", {})
if requires:
print("\nRequirements:")
if requires.get("env"):
print(f" Environment: {', '.join(requires['env'])}")
if requires.get("packages"):
print(f" Packages: {', '.join(requires['packages'])}")
if requires.get("external"):
print(f" External: {', '.join(requires['external'])}")
if requires.get("tools"):
print(f" Tools: {', '.join(requires['tools'])}")
cli = recipe.get("cli", {})
if cli:
print("\nCLI:")
print(f" Command: {cli.get('command', '')}")
if cli.get("examples"):
print(" Examples:")
for ex in cli["examples"]:
print(f" {ex}")
safety = recipe.get("safety", {})
if safety:
print("\nSafety:")
print(f" Dry-run default: {safety.get('dry_run_default', False)}")
if safety.get("requires_consent"):
print(f" Requires consent: {safety.get('consent_message', 'Yes')}")
if safety.get("legal_disclaimer"):
print(f" Legal disclaimer: {safety.get('legal_disclaimer')}")
return 0
def cmd_doctor(args: argparse.Namespace) -> int:
"""Check recipe dependencies."""
recipe = get_recipe(args.name)
if not recipe:
print(f"Recipe not found: {args.name}")
return 1
print(f"Checking dependencies for: {args.name}")
print("-" * 50)
deps = check_dependencies(recipe)
all_ok = True
for dep, status in sorted(deps.items()):
icon = "✓" if status else "✗"
color = "\033[92m" if status else "\033[91m"
reset = "\033[0m"
print(f" {color}{icon}{reset} {dep}")
if not status:
all_ok = False
print("-" * 50)
if all_ok:
print("All dependencies satisfied!")
return 0
else:
print("\nMissing dependencies. Install with:")
requires = recipe.get("requires", {})
if requires.get("packages"):
missing_pkgs = [p for p in requires["packages"] if not deps.get(f"package:{p}", True)]
if missing_pkgs:
print(f" pip install {' '.join(missing_pkgs)}")
if requires.get("external"):
missing_ext = [e for e in requires["external"] if not deps.get(f"external:{e}", True)]
if missing_ext:
print(f" # Install external tools: {', '.join(missing_ext)}")
if requires.get("env"):
missing_env = [e for e in requires["env"] if not deps.get(f"env:{e}", True)]
if missing_env:
print(f" # Set environment variables: {', '.join(missing_env)}")
return 1
def cmd_run(args: argparse.Namespace) -> int:
"""Run a recipe."""
recipe = get_recipe(args.name)
if not recipe:
print(f"Recipe not found: {args.name}")
return 1
# Check safety requirements
safety = recipe.get("safety", {})
if safety.get("requires_consent") and not args.consent:
print(f"This recipe requires consent: {safety.get('consent_message', '')}")
print("Use --consent flag to acknowledge.")
return 1
if safety.get("legal_disclaimer"):
print(f"LEGAL DISCLAIMER: {safety['legal_disclaimer']}")
# Check dependencies
if not args.skip_checks:
deps = check_dependencies(recipe)
missing = [k for k, v in deps.items() if not v]
if missing:
print(f"Missing dependencies: {', '.join(missing)}")
print("Run 'praison recipes doctor {args.name}' for details.")
if not args.force:
return 1
# Dry run mode
if args.dry_run or (safety.get("dry_run_default") and not args.write):
print(f"DRY RUN: Would execute recipe '{args.name}'")
print(f" Input: {args.input}")
print(f" Output: {args.output or 'default'}")
print("\nUse --write to execute.")
return 0
# Execute recipe
print(f"Running recipe: {args.name}")
print(f"Input: {args.input}")
# TODO: Implement actual recipe execution via praisonaiagents
# For now, show what would be executed
print("\nRecipe execution would involve:")
for tool in recipe.get("requires", {}).get("tools", []):
print(f" - Using {tool}")
print("\nRecipe execution completed (placeholder).")
return 0
def cmd_explain(args: argparse.Namespace) -> int:
"""Explain recipe execution plan."""
recipe = get_recipe(args.name)
if not recipe:
print(f"Recipe not found: {args.name}")
return 1
print(f"Execution Plan for: {args.name}")
print("=" * 50)
print("\n1. Dependency Check:")
for tool in recipe.get("requires", {}).get("tools", []):
print(f" - Load {tool}")
print("\n2. Input Processing:")
inputs = recipe.get("inputs", {})
for name, spec in inputs.items():
print(f" - {name}: {spec.get('type', 'unknown')}")
print("\n3. Execution:")
print(f" - Run recipe workflow")
print("\n4. Output:")
outputs = recipe.get("outputs", {})
for name, spec in outputs.items():
print(f" - {name}: {spec.get('type', 'unknown')}")
return 0
def cmd_init(args: argparse.Namespace) -> int:
"""Initialize recipe in current directory."""
recipe = get_recipe(args.name)
if not recipe:
print(f"Recipe not found: {args.name}")
return 1
target_dir = Path(args.output) if args.output else Path.cwd() / args.name
if target_dir.exists() and not args.force:
print(f"Directory already exists: {target_dir}")
print("Use --force to overwrite.")
return 1
import shutil
source_dir = Path(recipe["path"])
if args.dry_run:
print(f"DRY RUN: Would copy {source_dir} to {target_dir}")
return 0
target_dir.mkdir(parents=True, exist_ok=True)
# Copy recipe files
for item in source_dir.iterdir():
if item.is_file():
shutil.copy2(item, target_dir / item.name)
print(f"Initialized recipe '{args.name}' in {target_dir}")
return 0
def setup_parser(subparsers: argparse._SubParsersAction) -> None:
"""Setup recipes subcommand parser."""
recipes_parser = subparsers.add_parser(
"recipes",
help="Manage and run Agent-Recipes",
description="Commands for managing and running Agent-Recipes templates.",
)
recipes_subparsers = recipes_parser.add_subparsers(dest="recipes_command")
# List command
list_parser = recipes_subparsers.add_parser("list", help="List available recipes")
list_parser.add_argument("--json", action="store_true", help="Output as JSON")
list_parser.add_argument("--group", action="store_true", help="Group by category")
list_parser.set_defaults(func=cmd_list)
# Info command
info_parser = recipes_subparsers.add_parser("info", help="Show recipe details")
info_parser.add_argument("name", help="Recipe name")
info_parser.add_argument("--json", action="store_true", help="Output as JSON")
info_parser.set_defaults(func=cmd_info)
# Doctor command
doctor_parser = recipes_subparsers.add_parser("doctor", help="Check recipe dependencies")
doctor_parser.add_argument("name", help="Recipe name")
doctor_parser.set_defaults(func=cmd_doctor)
# Run command
run_parser = recipes_subparsers.add_parser("run", help="Run a recipe")
run_parser.add_argument("name", help="Recipe name")
run_parser.add_argument("input", nargs="?", help="Input file or value")
run_parser.add_argument("--output", "-o", help="Output directory")
run_parser.add_argument("--dry-run", action="store_true", help="Show what would be done")
run_parser.add_argument("--write", action="store_true", help="Actually execute (for dry-run-default recipes)")
run_parser.add_argument("--force", action="store_true", help="Force execution despite missing deps")
run_parser.add_argument("--consent", action="store_true", help="Acknowledge consent requirements")
run_parser.add_argument("--skip-checks", action="store_true", help="Skip dependency checks")
run_parser.add_argument("--json", action="store_true", help="Output as JSON")
run_parser.set_defaults(func=cmd_run)
# Explain command
explain_parser = recipes_subparsers.add_parser("explain", help="Explain recipe execution plan")
explain_parser.add_argument("name", help="Recipe name")
explain_parser.set_defaults(func=cmd_explain)
# Init command
init_parser = recipes_subparsers.add_parser("init", help="Initialize recipe in directory")
init_parser.add_argument("name", help="Recipe name")
init_parser.add_argument("--output", "-o", help="Target directory")
init_parser.add_argument("--force", action="store_true", help="Overwrite existing")
init_parser.add_argument("--dry-run", action="store_true", help="Show what would be done")
init_parser.set_defaults(func=cmd_init)
def handle_recipes_command(args: argparse.Namespace) -> int:
"""Handle recipes subcommand."""
if not hasattr(args, "func"):
# No subcommand specified, show help
print("Usage: praison recipes <command> [options]")
print("\nCommands:")
print(" list List available recipes")
print(" info Show recipe details")
print(" doctor Check recipe dependencies")
print(" run Run a recipe")
print(" explain Explain recipe execution plan")
print(" init Initialize recipe in directory")
return 0
return args.func(args)
@@ -0,0 +1,812 @@
"""
Serve CLI Feature Handler
Provides CLI commands for launching PraisonAI servers:
- praisonai serve agents - Launch agents as HTTP API
- praisonai serve recipe - Launch recipe runner
- praisonai serve mcp - Launch MCP server
- praisonai serve tools - Launch tools as MCP server
- praisonai serve a2a - Launch A2A server
- praisonai serve a2u - Launch A2U event stream server
- praisonai serve unified - Launch unified server with all providers
All servers include the unified discovery endpoint at /__praisonai__/discovery.
"""
import os
import sys
from typing import Any, Dict, List
class ServeHandler:
"""
CLI handler for serve operations.
Commands:
- agents: Launch agents as HTTP API
- recipe: Launch recipe runner
- mcp: Launch MCP server
- tools: Launch tools as MCP server
- a2a: Launch A2A server
- a2u: Launch A2U event stream server
- unified: Launch unified server
"""
EXIT_SUCCESS = 0
EXIT_GENERAL_ERROR = 1
EXIT_VALIDATION_ERROR = 2
DEFAULT_HOST = "127.0.0.1"
DEFAULT_PORT = 8765
def __init__(self):
"""Initialize the handler."""
pass
def handle(self, args: List[str]) -> int:
"""
Handle serve subcommand.
Args:
args: Command arguments
Returns:
Exit code
"""
if not args:
self._print_help()
return self.EXIT_SUCCESS
command = args[0]
remaining = args[1:]
commands = {
"agents": self.cmd_agents,
"recipe": self.cmd_recipe,
"mcp": self.cmd_mcp,
"tools": self.cmd_tools,
"a2a": self.cmd_a2a,
"a2u": self.cmd_a2u,
"unified": self.cmd_unified,
"help": lambda _: self._print_help() or self.EXIT_SUCCESS,
"--help": lambda _: self._print_help() or self.EXIT_SUCCESS,
"-h": lambda _: self._print_help() or self.EXIT_SUCCESS,
}
if command in commands:
return commands[command](remaining)
else:
self._print_error(f"Unknown command: {command}")
self._print_help()
return self.EXIT_GENERAL_ERROR
def _print_help(self):
"""Print help message."""
help_text = """
[bold cyan]PraisonAI Serve[/bold cyan]
Launch PraisonAI servers with unified discovery support.
[bold]Usage:[/bold]
praisonai serve <command> [options]
[bold]Commands:[/bold]
agents Launch agents as HTTP API
recipe Launch recipe runner server
mcp Launch MCP server (HTTP mode)
tools Launch tools as MCP server
a2a Launch A2A protocol server
a2u Launch A2U event stream server
unified Launch unified server with all providers
[bold]Common Options:[/bold]
--host <host> Server host (default: 127.0.0.1)
--port <port> Server port (default: 8765)
--reload Enable hot reload
--api-key <key> API key for authentication
[bold]Agents Options:[/bold]
--file <path> Agents YAML file (default: agents.yaml)
--path <path> API endpoint path (default: /agents)
[bold]Recipe Options:[/bold]
--config <path> Config file path (serve.yaml)
--preload Preload all recipes
[bold]MCP Options:[/bold]
--transport Transport type: http, sse (default: http)
[bold]Examples:[/bold]
praisonai serve agents --file agents.yaml --port 8000
praisonai serve recipe --port 8765
praisonai serve mcp --transport http --port 8080
praisonai serve tools --port 8081
praisonai serve a2a --port 8082
praisonai serve unified --port 8765
[bold]Discovery:[/bold]
All servers expose /__praisonai__/discovery for unified endpoint discovery.
Use `praisonai endpoints` to interact with any server.
"""
self._print_rich(help_text)
def _print_rich(self, text: str):
"""Print with rich formatting if available."""
try:
from rich import print as rprint
rprint(text)
except ImportError:
import re
plain = re.sub(r'\[/?[^\]]+\]', '', text)
print(plain)
def _print_error(self, message: str):
"""Print error message."""
try:
from rich import print as rprint
rprint(f"[red]Error: {message}[/red]")
except ImportError:
print(f"Error: {message}", file=sys.stderr)
def _print_success(self, message: str):
"""Print success message."""
try:
from rich import print as rprint
rprint(f"[green]✓ {message}[/green]")
except ImportError:
print(f"✓ {message}")
def _parse_args(self, args: List[str], spec: Dict[str, Any]) -> Dict[str, Any]:
"""Parse command arguments based on spec."""
result = {k: v.get("default") for k, v in spec.items()}
i = 0
while i < len(args):
arg = args[i]
if arg.startswith("--"):
key = arg[2:].replace("-", "_")
if key in spec:
if spec[key].get("flag"):
result[key] = True
elif i + 1 < len(args):
value = args[i + 1]
# Type conversion
if spec[key].get("type") == "int":
value = int(value)
result[key] = value
i += 1
i += 1
else:
i += 1
return result
def cmd_agents(self, args: List[str]) -> int:
"""Launch agents as HTTP API."""
spec = {
"file": {"default": "agents.yaml"},
"host": {"default": self.DEFAULT_HOST},
"port": {"default": self.DEFAULT_PORT, "type": "int"},
"path": {"default": "/agents"},
"reload": {"flag": True, "default": False},
"api_key": {"default": None},
}
parsed = self._parse_args(args, spec)
# Check if agents file exists
if not os.path.exists(parsed["file"]):
self._print_error(f"Agents file not found: {parsed['file']}")
return self.EXIT_VALIDATION_ERROR
try:
self._print_success(f"Starting agents server on {parsed['host']}:{parsed['port']}")
print(f" Agents file: {parsed['file']}")
print(f" Endpoint: {parsed['path']}")
print(" Discovery: /__praisonai__/discovery")
# Create and run server
app = self._create_agents_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], parsed["reload"])
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
print("Install with: pip install praisonai[serve]")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_agents_app(self, config: Dict[str, Any]) -> Any:
"""Create FastAPI app for agents."""
from fastapi import FastAPI, HTTPException, Request
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from praisonai.endpoints.discovery import (
create_discovery_document,
EndpointInfo,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
# Load agents from YAML
import yaml
with open(config["file"]) as f:
_ = yaml.safe_load(f) # Validate YAML
# Create discovery document
discovery = create_discovery_document(server_name="praisonai-agents")
discovery.add_provider(ProviderInfo(
type="agents-api",
name="Agents API",
description="Agent HTTP API endpoints",
capabilities=["invoke", "health"],
))
# Create app
app = FastAPI(
title="PraisonAI Agents API",
description="HTTP API for PraisonAI Agents",
)
# Add discovery routes
add_discovery_routes(app, discovery)
# Request model
class AgentQuery(BaseModel):
query: str
# Create endpoint for agents
path = config["path"]
@app.post(path)
async def invoke_agents(request: Request, query_data: AgentQuery = None):
"""Invoke agents with a query."""
if query_data is None:
try:
body = await request.json()
_ = body.get("query", "") # Extract query
except Exception:
raise HTTPException(status_code=400, detail="Invalid request")
try:
# Lazy load and run agents
from praisonai.agents_generator import AgentsGenerator
generator = AgentsGenerator(
agent_file=config["file"],
framework="praisonai",
)
result = generator.generate_crew_and_kickoff()
return {"response": result}
except Exception as e:
return JSONResponse(
{"error": str(e)},
status_code=500,
)
# Add endpoint to discovery
discovery.add_endpoint(EndpointInfo(
name=path.lstrip("/"),
description="Invoke agents workflow",
provider_type="agents-api",
input_schema={"type": "object", "properties": {"query": {"type": "string"}}},
streaming=["none"],
))
# Root endpoint
@app.get("/")
async def root():
return {
"message": "PraisonAI Agents API",
"endpoints": [path],
"discovery": "/__praisonai__/discovery",
}
return app
def cmd_recipe(self, args: List[str]) -> int:
"""Launch recipe runner server."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": self.DEFAULT_PORT, "type": "int"},
"config": {"default": None},
"reload": {"flag": True, "default": False},
"api_key": {"default": None},
"preload": {"flag": True, "default": False},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting recipe server on {parsed['host']}:{parsed['port']}")
print(" Discovery: /__praisonai__/discovery")
from praisonai.recipe.serve import serve, load_config
# Load config
config = load_config(parsed["config"]) if parsed["config"] else {}
if parsed["api_key"]:
config["api_key"] = parsed["api_key"]
serve(
host=parsed["host"],
port=parsed["port"],
reload=parsed["reload"],
config=config,
)
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
print("Install with: pip install praisonai[serve]")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def cmd_mcp(self, args: List[str]) -> int:
"""Launch MCP server."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": 8080, "type": "int"},
"transport": {"default": "http"},
"tools_file": {"default": None},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting MCP server on {parsed['host']}:{parsed['port']}")
print(f" Transport: {parsed['transport']}")
print(" Discovery: /__praisonai__/discovery")
app = self._create_mcp_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], False)
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_mcp_app(self, config: Dict[str, Any]) -> Any:
"""Create FastAPI app for MCP server."""
from fastapi import FastAPI
from praisonai.endpoints.discovery import (
create_discovery_document,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
# Create discovery document
discovery = create_discovery_document(server_name="praisonai-mcp")
discovery.add_provider(ProviderInfo(
type="mcp",
name="MCP Server",
description=f"MCP server ({config['transport']} transport)",
capabilities=["list-tools", "call-tool"],
))
app = FastAPI(
title="PraisonAI MCP Server",
description="MCP protocol server",
)
add_discovery_routes(app, discovery)
# MCP tools endpoint
@app.get("/mcp/tools")
async def list_tools():
"""List available MCP tools."""
# TODO: Load tools from config or registry
return {"tools": []}
@app.post("/mcp/tools/call")
async def call_tool(request_data: dict):
"""Call an MCP tool."""
tool_name = request_data.get("tool")
_ = request_data.get("arguments", {}) # Arguments for tool
# TODO: Execute tool
return {"result": None, "tool": tool_name}
@app.get("/")
async def root():
return {
"message": "PraisonAI MCP Server",
"transport": config["transport"],
"discovery": "/__praisonai__/discovery",
}
return app
def cmd_tools(self, args: List[str]) -> int:
"""Launch tools as MCP server."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": 8081, "type": "int"},
"tools_file": {"default": None},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting tools MCP server on {parsed['host']}:{parsed['port']}")
print(" Discovery: /__praisonai__/discovery")
app = self._create_tools_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], False)
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_tools_app(self, config: Dict[str, Any]) -> Any:
"""Create FastAPI app for tools MCP server."""
from fastapi import FastAPI
from praisonai.endpoints.discovery import (
create_discovery_document,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
discovery = create_discovery_document(server_name="praisonai-tools-mcp")
discovery.add_provider(ProviderInfo(
type="tools-mcp",
name="Tools MCP Server",
description="Python tools exposed as MCP server",
capabilities=["list-tools", "call-tool"],
))
app = FastAPI(
title="PraisonAI Tools MCP Server",
description="Tools exposed as MCP server",
)
add_discovery_routes(app, discovery)
@app.get("/tools")
async def list_tools():
"""List available tools."""
return {"tools": []}
@app.post("/tools/call")
async def call_tool(request_data: dict):
"""Call a tool."""
return {"result": None}
@app.get("/")
async def root():
return {
"message": "PraisonAI Tools MCP Server",
"discovery": "/__praisonai__/discovery",
}
return app
def cmd_a2a(self, args: List[str]) -> int:
"""Launch A2A protocol server."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": 8082, "type": "int"},
"file": {"default": "agents.yaml"},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting A2A server on {parsed['host']}:{parsed['port']}")
print(" Agent Card: /.well-known/agent.json")
print(" Discovery: /__praisonai__/discovery")
app = self._create_a2a_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], False)
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_a2a_app(self, config: Dict[str, Any]) -> Any:
"""Create FastAPI app for A2A server."""
from fastapi import FastAPI
from praisonai.endpoints.discovery import (
create_discovery_document,
EndpointInfo,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
discovery = create_discovery_document(server_name="praisonai-a2a")
discovery.add_provider(ProviderInfo(
type="a2a",
name="A2A Protocol",
description="Agent-to-Agent communication",
capabilities=["agent-card", "message-send"],
))
app = FastAPI(
title="PraisonAI A2A Server",
description="A2A protocol server",
)
add_discovery_routes(app, discovery)
# Agent card
@app.get("/.well-known/agent.json")
async def agent_card():
"""Return agent card for A2A discovery."""
return {
"name": "PraisonAI Agent",
"description": "PraisonAI Agent via A2A protocol",
"version": "1.0.0",
"url": f"http://{config.get('host', 'localhost')}:{config.get('port', 8082)}/a2a",
"capabilities": {
"streaming": True,
"pushNotifications": False,
},
"skills": [],
}
@app.get("/status")
async def status():
"""Return server status."""
return {"status": "ok", "name": "PraisonAI A2A", "version": "1.0.0"}
@app.post("/a2a")
async def a2a_endpoint(request_data: dict):
"""Handle A2A JSON-RPC requests."""
method = request_data.get("method")
params = request_data.get("params", {})
request_id = request_data.get("id")
if method == "message/send":
# Handle message
message = params.get("message", {})
parts = message.get("parts", [])
text = ""
for part in parts:
if part.get("type") == "text":
text += part.get("text", "")
# TODO: Process with agent
return {
"jsonrpc": "2.0",
"id": request_id,
"result": {
"message": {
"role": "assistant",
"parts": [{"type": "text", "text": f"Received: {text}"}],
},
},
}
return {
"jsonrpc": "2.0",
"id": request_id,
"error": {"code": -32601, "message": f"Method not found: {method}"},
}
discovery.add_endpoint(EndpointInfo(
name="a2a",
description="A2A protocol endpoint",
provider_type="a2a",
streaming=["sse"],
))
@app.get("/")
async def root():
return {
"message": "PraisonAI A2A Server",
"agent_card": "/.well-known/agent.json",
"discovery": "/__praisonai__/discovery",
}
return app
def cmd_a2u(self, args: List[str]) -> int:
"""Launch A2U event stream server."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": 8083, "type": "int"},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting A2U server on {parsed['host']}:{parsed['port']}")
print(" Events: /a2u/events/<stream>")
print(" Discovery: /__praisonai__/discovery")
app = self._create_a2u_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], False)
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_a2u_app(self, config: Dict[str, Any]) -> Any:
"""Create FastAPI app for A2U server."""
from fastapi import FastAPI
from praisonai.endpoints.discovery import (
create_discovery_document,
EndpointInfo,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
from praisonai.endpoints.a2u_server import create_a2u_routes
discovery = create_discovery_document(server_name="praisonai-a2u")
discovery.add_provider(ProviderInfo(
type="a2u",
name="A2U Event Stream",
description="Agent-to-User event streaming",
capabilities=["subscribe", "stream"],
))
app = FastAPI(
title="PraisonAI A2U Server",
description="A2U event stream server",
)
add_discovery_routes(app, discovery)
create_a2u_routes(app)
discovery.add_endpoint(EndpointInfo(
name="events",
description="Agent event stream",
provider_type="a2u",
streaming=["sse"],
))
@app.get("/")
async def root():
return {
"message": "PraisonAI A2U Server",
"events": "/a2u/events/<stream>",
"discovery": "/__praisonai__/discovery",
}
return app
def cmd_unified(self, args: List[str]) -> int:
"""Launch unified server with all providers."""
spec = {
"host": {"default": self.DEFAULT_HOST},
"port": {"default": self.DEFAULT_PORT, "type": "int"},
"file": {"default": "agents.yaml"},
"reload": {"flag": True, "default": False},
"api_key": {"default": None},
}
parsed = self._parse_args(args, spec)
try:
self._print_success(f"Starting unified server on {parsed['host']}:{parsed['port']}")
print(" Providers: agents-api, recipe, mcp, a2a, a2u")
print(" Discovery: /__praisonai__/discovery")
app = self._create_unified_app(parsed)
self._run_server(app, parsed["host"], parsed["port"], parsed["reload"])
except ImportError as e:
self._print_error(f"Missing dependency: {e}")
print("Install with: pip install praisonai[serve]")
return self.EXIT_GENERAL_ERROR
except Exception as e:
self._print_error(str(e))
return self.EXIT_GENERAL_ERROR
return self.EXIT_SUCCESS
def _create_unified_app(self, config: Dict[str, Any]) -> Any:
"""Create unified FastAPI app with all providers."""
from fastapi import FastAPI
from praisonai.endpoints.discovery import (
create_discovery_document,
EndpointInfo,
ProviderInfo,
)
from praisonai.endpoints.server import add_discovery_routes
from praisonai.endpoints.a2u_server import create_a2u_routes
discovery = create_discovery_document(server_name="praisonai-unified")
# Add all providers
discovery.add_provider(ProviderInfo(
type="agents-api",
name="Agents API",
capabilities=["invoke", "health"],
))
discovery.add_provider(ProviderInfo(
type="recipe",
name="Recipe Runner",
capabilities=["list", "describe", "invoke", "stream"],
))
discovery.add_provider(ProviderInfo(
type="mcp",
name="MCP Server",
capabilities=["list-tools", "call-tool"],
))
discovery.add_provider(ProviderInfo(
type="a2a",
name="A2A Protocol",
capabilities=["agent-card", "message-send"],
))
discovery.add_provider(ProviderInfo(
type="a2u",
name="A2U Event Stream",
capabilities=["subscribe", "stream"],
))
app = FastAPI(
title="PraisonAI Unified Server",
description="Unified server with all PraisonAI providers",
)
add_discovery_routes(app, discovery)
create_a2u_routes(app)
# Add sample endpoints
discovery.add_endpoint(EndpointInfo(
name="agents",
description="Agents workflow endpoint",
provider_type="agents-api",
))
discovery.add_endpoint(EndpointInfo(
name="events",
description="Agent event stream",
provider_type="a2u",
streaming=["sse"],
))
@app.get("/")
async def root():
return {
"message": "PraisonAI Unified Server",
"providers": ["agents-api", "recipe", "mcp", "a2a", "a2u"],
"discovery": "/__praisonai__/discovery",
}
@app.get("/.well-known/agent.json")
async def agent_card():
return {
"name": "PraisonAI Unified",
"description": "PraisonAI Unified Server",
"version": "1.0.0",
}
return app
def _run_server(self, app: Any, host: str, port: int, reload: bool) -> None:
"""Run the server with uvicorn."""
import uvicorn
uvicorn.run(app, host=host, port=port, reload=reload)
def handle_serve_command(args: List[str]) -> int:
"""Entry point for serve command."""
handler = ServeHandler()
return handler.handle(args)
+7 -1
View File
@@ -1193,11 +1193,17 @@ class PraisonAI:
sys.exit(exit_code)
elif args.command == 'endpoints':
# Endpoints command - client CLI for recipe endpoints
# Endpoints command - unified client CLI for all server types
from .features.endpoints import handle_endpoints_command
exit_code = handle_endpoints_command(unknown_args)
sys.exit(exit_code)
elif args.command == 'serve':
# Serve command - launch PraisonAI servers
from .features.serve import handle_serve_command
exit_code = handle_serve_command(unknown_args)
sys.exit(exit_code)
elif args.command == 'agents':
# Agents command - run multiple agents with custom definitions
if not PRAISONAI_AVAILABLE:
+1 -1
View File
@@ -57,7 +57,7 @@ class CloudDeployer:
file.write("FROM python:3.11-slim\n")
file.write("WORKDIR /app\n")
file.write("COPY . .\n")
file.write("RUN pip install flask praisonai==2.7.3 gunicorn markdown\n")
file.write("RUN pip install flask praisonai==2.7.4 gunicorn markdown\n")
file.write("EXPOSE 8080\n")
file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
@@ -0,0 +1,53 @@
"""
Unified Endpoints Module
Provides unified discovery, provider adapters, and server utilities
for all PraisonAI serve features.
Provider Types:
- recipe: Recipe runner endpoints
- agents-api: Single/multi-agent HTTP API
- mcp: MCP server (stdio, http, sse)
- tools-mcp: Tools exposed as MCP server
- a2a: Agent-to-agent protocol
- a2u: Agent-to-user event stream
Architecture:
- Discovery schema is versioned and consistent across all providers
- Provider adapters abstract protocol differences
- CLI client uses unified discovery for list/describe/invoke/health
"""
# Lazy loading for all exports
_lazy_imports = {
"DiscoveryDocument": ".discovery",
"EndpointInfo": ".discovery",
"ProviderInfo": ".discovery",
"SCHEMA_VERSION": ".discovery",
"create_discovery_document": ".discovery",
"BaseProvider": ".providers.base",
"RecipeProvider": ".providers.recipe",
"AgentsAPIProvider": ".providers.agents_api",
"MCPProvider": ".providers.mcp",
"ToolsMCPProvider": ".providers.tools_mcp",
"A2AProvider": ".providers.a2a",
"A2UProvider": ".providers.a2u",
"ProviderRegistry": ".registry",
"get_provider": ".registry",
"register_provider": ".registry",
"list_provider_types": ".registry",
"get_provider_class": ".registry",
"create_unified_app": ".server",
"add_discovery_routes": ".server",
}
def __getattr__(name: str):
if name in _lazy_imports:
module_path = _lazy_imports[name]
import importlib
module = importlib.import_module(module_path, package=__name__)
return getattr(module, name)
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
def __dir__():
return list(_lazy_imports.keys())
@@ -0,0 +1,410 @@
"""
A2U (Agent-to-User) Event Stream Server
Provides SSE-based event streaming for agent-to-user communication.
"""
import asyncio
import json
import logging
import uuid
from dataclasses import dataclass, field
from datetime import datetime, timezone
from typing import Any, Dict, List, Optional, Set
logger = logging.getLogger(__name__)
@dataclass
class A2UEvent:
"""An event in the A2U stream."""
event_type: str
data: Dict[str, Any]
timestamp: str = field(default_factory=lambda: datetime.now(timezone.utc).isoformat())
event_id: str = field(default_factory=lambda: uuid.uuid4().hex[:12])
def to_sse(self) -> str:
"""Convert to SSE format."""
return f"event: {self.event_type}\ndata: {json.dumps(self.data)}\nid: {self.event_id}\n\n"
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary."""
return {
"event_type": self.event_type,
"data": self.data,
"timestamp": self.timestamp,
"event_id": self.event_id,
}
@dataclass
class A2USubscription:
"""A subscription to an A2U event stream."""
subscription_id: str
stream_name: str
filters: List[str] = field(default_factory=list)
created_at: str = field(default_factory=lambda: datetime.now(timezone.utc).isoformat())
def matches_event(self, event: A2UEvent) -> bool:
"""Check if event matches subscription filters."""
if not self.filters:
return True
return event.event_type in self.filters
class A2UEventBus:
"""
Event bus for A2U event distribution.
Manages subscriptions and broadcasts events to subscribers.
"""
def __init__(self):
"""Initialize the event bus."""
self._subscriptions: Dict[str, A2USubscription] = {}
self._queues: Dict[str, asyncio.Queue] = {}
self._streams: Dict[str, Set[str]] = {} # stream_name -> subscription_ids
def subscribe(
self,
stream_name: str = "events",
filters: Optional[List[str]] = None,
) -> A2USubscription:
"""
Subscribe to an event stream.
Args:
stream_name: Name of the stream to subscribe to
filters: Optional list of event types to filter
Returns:
A2USubscription object
"""
subscription_id = f"sub-{uuid.uuid4().hex[:12]}"
subscription = A2USubscription(
subscription_id=subscription_id,
stream_name=stream_name,
filters=filters or [],
)
self._subscriptions[subscription_id] = subscription
self._queues[subscription_id] = asyncio.Queue()
if stream_name not in self._streams:
self._streams[stream_name] = set()
self._streams[stream_name].add(subscription_id)
logger.debug(f"Created subscription {subscription_id} for stream {stream_name}")
return subscription
def unsubscribe(self, subscription_id: str) -> bool:
"""
Unsubscribe from an event stream.
Args:
subscription_id: ID of the subscription to remove
Returns:
True if unsubscribed, False if not found
"""
if subscription_id not in self._subscriptions:
return False
subscription = self._subscriptions[subscription_id]
# Remove from stream set
if subscription.stream_name in self._streams:
self._streams[subscription.stream_name].discard(subscription_id)
# Clean up
del self._subscriptions[subscription_id]
del self._queues[subscription_id]
logger.debug(f"Removed subscription {subscription_id}")
return True
async def publish(self, event: A2UEvent, stream_name: str = "events") -> int:
"""
Publish an event to a stream.
Args:
event: Event to publish
stream_name: Name of the stream
Returns:
Number of subscribers that received the event
"""
if stream_name not in self._streams:
return 0
count = 0
for sub_id in self._streams[stream_name]:
subscription = self._subscriptions.get(sub_id)
if subscription and subscription.matches_event(event):
await self._queues[sub_id].put(event)
count += 1
logger.debug(f"Published event {event.event_type} to {count} subscribers")
return count
def publish_sync(self, event: A2UEvent, stream_name: str = "events") -> int:
"""
Synchronously publish an event (creates event loop if needed).
Args:
event: Event to publish
stream_name: Name of the stream
Returns:
Number of subscribers that received the event
"""
try:
loop = asyncio.get_event_loop()
if loop.is_running():
# Schedule in running loop
asyncio.ensure_future(self.publish(event, stream_name))
return len(self._streams.get(stream_name, set()))
else:
return loop.run_until_complete(self.publish(event, stream_name))
except RuntimeError:
# No event loop, create one
return asyncio.run(self.publish(event, stream_name))
async def get_events(
self,
subscription_id: str,
timeout: float = 30.0,
):
"""
Async generator for subscription events.
Args:
subscription_id: ID of the subscription
timeout: Timeout for waiting for events
Yields:
A2UEvent objects
"""
if subscription_id not in self._queues:
return
queue = self._queues[subscription_id]
while True:
try:
event = await asyncio.wait_for(queue.get(), timeout=timeout)
yield event
except asyncio.TimeoutError:
# Send keepalive
yield A2UEvent(event_type="keepalive", data={})
# Global event bus instance
_event_bus: Optional[A2UEventBus] = None
def get_event_bus() -> A2UEventBus:
"""Get or create the global event bus."""
global _event_bus
if _event_bus is None:
_event_bus = A2UEventBus()
return _event_bus
def create_a2u_routes(app: Any, event_bus: Optional[A2UEventBus] = None) -> None:
"""
Add A2U routes to a FastAPI/Starlette application.
Args:
app: FastAPI or Starlette application
event_bus: Optional event bus (uses global if not provided)
"""
bus = event_bus or get_event_bus()
try:
from starlette.responses import JSONResponse, StreamingResponse
from starlette.routing import Route
except ImportError:
try:
from fastapi.responses import JSONResponse, StreamingResponse
except ImportError:
raise ImportError("Starlette or FastAPI required for A2U routes")
async def a2u_info(request):
"""GET /a2u/info - Get A2U server info."""
return JSONResponse({
"name": "A2U Event Stream",
"version": "1.0.0",
"streams": list(bus._streams.keys()) or ["events"],
"event_types": [
"agent.started",
"agent.thinking",
"agent.tool_call",
"agent.response",
"agent.completed",
"agent.error",
],
})
async def a2u_subscribe(request):
"""POST /a2u/subscribe - Subscribe to an event stream."""
try:
body = await request.json()
except Exception:
body = {}
stream_name = body.get("stream", "events")
filters = body.get("filters", [])
subscription = bus.subscribe(stream_name, filters)
base_url = str(request.url).rsplit("/", 1)[0]
return JSONResponse({
"subscription_id": subscription.subscription_id,
"stream_name": subscription.stream_name,
"stream_url": f"{base_url}/events/{subscription.subscription_id}",
"created_at": subscription.created_at,
})
async def a2u_unsubscribe(request):
"""POST /a2u/unsubscribe - Unsubscribe from an event stream."""
try:
body = await request.json()
except Exception:
return JSONResponse({"error": "Invalid JSON"}, status_code=400)
subscription_id = body.get("subscription_id")
if not subscription_id:
return JSONResponse({"error": "subscription_id required"}, status_code=400)
if bus.unsubscribe(subscription_id):
return JSONResponse({"status": "unsubscribed"})
else:
return JSONResponse({"error": "Subscription not found"}, status_code=404)
async def a2u_events_stream(request):
"""GET /a2u/events/{stream_name} - Stream events via SSE."""
stream_name = request.path_params.get("stream_name", "events")
# Create subscription for this stream
subscription = bus.subscribe(stream_name)
async def event_generator():
try:
async for event in bus.get_events(subscription.subscription_id):
yield event.to_sse()
finally:
bus.unsubscribe(subscription.subscription_id)
return StreamingResponse(
event_generator(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
"X-Accel-Buffering": "no",
},
)
async def a2u_events_subscription(request):
"""GET /a2u/events/sub/{subscription_id} - Stream events for subscription."""
subscription_id = request.path_params.get("subscription_id")
if subscription_id not in bus._subscriptions:
return JSONResponse({"error": "Subscription not found"}, status_code=404)
async def event_generator():
async for event in bus.get_events(subscription_id):
yield event.to_sse()
return StreamingResponse(
event_generator(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
"X-Accel-Buffering": "no",
},
)
async def a2u_health(request):
"""GET /a2u/health - A2U health check."""
return JSONResponse({
"status": "healthy",
"active_subscriptions": len(bus._subscriptions),
"active_streams": len(bus._streams),
})
# Add routes based on app type
if hasattr(app, 'add_api_route'):
# FastAPI
app.add_api_route("/a2u/info", a2u_info, methods=["GET"])
app.add_api_route("/a2u/subscribe", a2u_subscribe, methods=["POST"])
app.add_api_route("/a2u/unsubscribe", a2u_unsubscribe, methods=["POST"])
app.add_api_route("/a2u/events/{stream_name}", a2u_events_stream, methods=["GET"])
app.add_api_route("/a2u/events/sub/{subscription_id}", a2u_events_subscription, methods=["GET"])
app.add_api_route("/a2u/health", a2u_health, methods=["GET"])
elif hasattr(app, 'routes'):
# Starlette
app.routes.extend([
Route("/a2u/info", a2u_info, methods=["GET"]),
Route("/a2u/subscribe", a2u_subscribe, methods=["POST"]),
Route("/a2u/unsubscribe", a2u_unsubscribe, methods=["POST"]),
Route("/a2u/events/{stream_name}", a2u_events_stream, methods=["GET"]),
Route("/a2u/events/sub/{subscription_id}", a2u_events_subscription, methods=["GET"]),
Route("/a2u/health", a2u_health, methods=["GET"]),
])
# Helper functions for publishing events
def emit_agent_event(
event_type: str,
data: Dict[str, Any],
agent_id: Optional[str] = None,
stream_name: str = "events",
) -> None:
"""
Emit an agent event to the A2U stream.
Args:
event_type: Type of event (e.g., "agent.started", "agent.response")
data: Event data
agent_id: Optional agent ID to include
stream_name: Stream to publish to
"""
if agent_id:
data["agent_id"] = agent_id
event = A2UEvent(event_type=event_type, data=data)
get_event_bus().publish_sync(event, stream_name)
def emit_agent_started(agent_id: str, agent_name: str, **kwargs) -> None:
"""Emit agent.started event."""
emit_agent_event("agent.started", {"agent_name": agent_name, **kwargs}, agent_id)
def emit_agent_thinking(agent_id: str, message: str = "", **kwargs) -> None:
"""Emit agent.thinking event."""
emit_agent_event("agent.thinking", {"message": message, **kwargs}, agent_id)
def emit_agent_tool_call(agent_id: str, tool_name: str, arguments: Dict = None, **kwargs) -> None:
"""Emit agent.tool_call event."""
emit_agent_event("agent.tool_call", {"tool_name": tool_name, "arguments": arguments or {}, **kwargs}, agent_id)
def emit_agent_response(agent_id: str, response: str, **kwargs) -> None:
"""Emit agent.response event."""
emit_agent_event("agent.response", {"response": response, **kwargs}, agent_id)
def emit_agent_completed(agent_id: str, result: Any = None, **kwargs) -> None:
"""Emit agent.completed event."""
emit_agent_event("agent.completed", {"result": result, **kwargs}, agent_id)
def emit_agent_error(agent_id: str, error: str, **kwargs) -> None:
"""Emit agent.error event."""
emit_agent_event("agent.error", {"error": error, **kwargs}, agent_id)
@@ -0,0 +1,165 @@
"""
Unified Discovery Schema
Provides versioned discovery document format for all PraisonAI serve features.
This schema is consistent across all provider types.
"""
from dataclasses import dataclass, field, asdict
from typing import Any, Dict, List, Optional
from enum import Enum
# Schema version - increment on breaking changes
SCHEMA_VERSION = "1.0.0"
class ProviderType(str, Enum):
"""Supported provider types."""
RECIPE = "recipe"
AGENTS_API = "agents-api"
MCP = "mcp"
TOOLS_MCP = "tools-mcp"
A2A = "a2a"
A2U = "a2u"
class StreamingMode(str, Enum):
"""Supported streaming modes."""
NONE = "none"
SSE = "sse"
WEBSOCKET = "websocket"
MCP_STREAM = "mcp-stream"
class AuthMode(str, Enum):
"""Supported authentication modes."""
NONE = "none"
API_KEY = "api-key"
BEARER = "bearer"
JWT = "jwt"
@dataclass
class EndpointInfo:
"""Information about a single endpoint."""
name: str
description: str = ""
provider_type: str = "recipe"
tags: List[str] = field(default_factory=list)
version: str = "1.0.0"
# Schema information
input_schema: Optional[Dict[str, Any]] = None
output_schema: Optional[Dict[str, Any]] = None
# Capabilities
streaming: List[str] = field(default_factory=lambda: ["none"])
auth_modes: List[str] = field(default_factory=lambda: ["none"])
# Metadata
examples: List[Dict[str, Any]] = field(default_factory=list)
metadata: Dict[str, Any] = field(default_factory=dict)
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary."""
return asdict(self)
@dataclass
class ProviderInfo:
"""Information about a provider."""
type: str
name: str
version: str = "1.0.0"
description: str = ""
capabilities: List[str] = field(default_factory=list)
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary."""
return asdict(self)
@dataclass
class DiscoveryDocument:
"""
Unified discovery document for PraisonAI servers.
This document is served at /__praisonai__/discovery and provides
a consistent interface for discovering endpoints across all provider types.
"""
schema_version: str = SCHEMA_VERSION
server_name: str = "praisonai"
server_version: str = "1.0.0"
# Provider information
providers: List[ProviderInfo] = field(default_factory=list)
# Endpoint information
endpoints: List[EndpointInfo] = field(default_factory=list)
# Server capabilities
auth_modes: List[str] = field(default_factory=lambda: ["none"])
streaming_modes: List[str] = field(default_factory=lambda: ["none", "sse"])
# Metadata
metadata: Dict[str, Any] = field(default_factory=dict)
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary for JSON serialization."""
return {
"schema_version": self.schema_version,
"server_name": self.server_name,
"server_version": self.server_version,
"providers": [p.to_dict() for p in self.providers],
"endpoints": [e.to_dict() for e in self.endpoints],
"auth_modes": self.auth_modes,
"streaming_modes": self.streaming_modes,
"metadata": self.metadata,
}
def add_provider(self, provider: ProviderInfo) -> None:
"""Add a provider to the discovery document."""
self.providers.append(provider)
def add_endpoint(self, endpoint: EndpointInfo) -> None:
"""Add an endpoint to the discovery document."""
self.endpoints.append(endpoint)
def get_endpoints_by_type(self, provider_type: str) -> List[EndpointInfo]:
"""Get endpoints filtered by provider type."""
return [e for e in self.endpoints if e.provider_type == provider_type]
def get_endpoint_by_name(self, name: str) -> Optional[EndpointInfo]:
"""Get an endpoint by name."""
for e in self.endpoints:
if e.name == name:
return e
return None
def create_discovery_document(
server_name: str = "praisonai",
server_version: Optional[str] = None,
) -> DiscoveryDocument:
"""
Create a new discovery document.
Args:
server_name: Name of the server
server_version: Version of the server (auto-detected if not provided)
Returns:
DiscoveryDocument instance
"""
if server_version is None:
try:
from praisonai.version import __version__
server_version = __version__
except ImportError:
server_version = "unknown"
return DiscoveryDocument(
server_name=server_name,
server_version=server_version,
)
@@ -0,0 +1,28 @@
"""
Provider Adapters
Each provider adapter implements the BaseProvider interface
and handles protocol-specific details for discovery and invocation.
"""
# Lazy loading for all exports
_lazy_imports = {
"BaseProvider": ".base",
"RecipeProvider": ".recipe",
"AgentsAPIProvider": ".agents_api",
"MCPProvider": ".mcp",
"ToolsMCPProvider": ".tools_mcp",
"A2AProvider": ".a2a",
"A2UProvider": ".a2u",
}
def __getattr__(name: str):
if name in _lazy_imports:
module_path = _lazy_imports[name]
import importlib
module = importlib.import_module(module_path, package=__name__)
return getattr(module, name)
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
def __dir__():
return list(_lazy_imports.keys())
@@ -0,0 +1,253 @@
"""
A2A Provider
Provider adapter for Agent-to-Agent protocol endpoints.
"""
from typing import Any, Dict, Iterator, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class A2AProvider(BaseProvider):
"""
Provider adapter for A2A (Agent-to-Agent) protocol endpoints.
A2A enables agent-to-agent communication following the A2A protocol spec.
Supports discovery via /.well-known/agent.json and message passing.
"""
provider_type = "a2a"
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="A2A Protocol",
description="Agent-to-Agent communication protocol",
capabilities=["agent-card", "message-send", "task-management"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available A2A agents."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
endpoints = []
for ep in data.get("endpoints", []):
if ep.get("provider_type") == self.provider_type:
endpoints.append(EndpointInfo(
name=ep.get("name", ""),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
streaming=ep.get("streaming", ["sse"]),
auth_modes=ep.get("auth_modes", ["none"]),
))
return endpoints
# Fallback: try agent card
result = self._make_request("GET", "/.well-known/agent.json")
if result.get("error"):
return []
agent_card = result.get("data", {})
return [EndpointInfo(
name=agent_card.get("name", "a2a-agent"),
description=agent_card.get("description", "A2A Agent"),
provider_type=self.provider_type,
version=agent_card.get("version", "1.0.0"),
streaming=["sse"] if agent_card.get("capabilities", {}).get("streaming") else ["none"],
auth_modes=["none"],
metadata={"agent_card": agent_card},
)]
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about an A2A agent."""
# Get agent card
result = self._make_request("GET", "/.well-known/agent.json")
if result.get("error"):
return None
agent_card = result.get("data", {})
return EndpointInfo(
name=agent_card.get("name", name),
description=agent_card.get("description", ""),
provider_type=self.provider_type,
version=agent_card.get("version", "1.0.0"),
streaming=["sse"] if agent_card.get("capabilities", {}).get("streaming") else ["none"],
auth_modes=["none"],
metadata={
"agent_card": agent_card,
"skills": agent_card.get("skills", []),
"capabilities": agent_card.get("capabilities", {}),
},
)
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Send a message to an A2A agent."""
import uuid
# Build A2A message
message = input_data.get("message", "") if input_data else ""
if not message and config:
message = config.get("message", config.get("query", ""))
body = {
"jsonrpc": "2.0",
"method": "message/send",
"id": str(uuid.uuid4()),
"params": {
"message": {
"role": "user",
"parts": [{"type": "text", "text": message}],
},
},
}
result = self._make_request("POST", "/a2a", json_data=body)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
# Handle JSON-RPC response
if "result" in data:
return InvokeResult(
ok=True,
status="success",
data=data["result"],
)
elif "error" in data:
return InvokeResult(
ok=False,
status="error",
error=data["error"].get("message", str(data["error"])),
)
return InvokeResult(
ok=True,
status="success",
data=data,
)
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""Send a streaming message to an A2A agent."""
import json
import urllib.request
import uuid
message = input_data.get("message", "") if input_data else ""
if not message and config:
message = config.get("message", config.get("query", ""))
body = {
"jsonrpc": "2.0",
"method": "message/stream",
"id": str(uuid.uuid4()),
"params": {
"message": {
"role": "user",
"parts": [{"type": "text", "text": message}],
},
},
}
full_url = f"{self.base_url}/a2a"
headers = {
"Content-Type": "application/json",
"Accept": "text/event-stream",
}
if self.api_key:
headers["X-API-Key"] = self.api_key
data = json.dumps(body).encode("utf-8")
try:
req = urllib.request.Request(full_url, data=data, headers=headers, method="POST")
with urllib.request.urlopen(req, timeout=300) as response:
buffer = ""
for line in response:
line = line.decode("utf-8")
buffer += line
if buffer.endswith("\n\n"):
event_type = "message"
event_data = ""
for part in buffer.strip().split("\n"):
if part.startswith("event:"):
event_type = part[6:].strip()
elif part.startswith("data:"):
event_data = part[5:].strip()
if event_data:
try:
yield {"event": event_type, "data": json.loads(event_data)}
except json.JSONDecodeError:
yield {"event": event_type, "data": event_data}
buffer = ""
except Exception as e:
yield {"event": "error", "data": {"error": str(e)}}
def health(self) -> HealthResult:
"""Check A2A server health."""
# Try status endpoint first
result = self._make_request("GET", "/status")
if not result.get("error"):
data = result.get("data", {})
return HealthResult(
healthy=data.get("status") == "ok",
status=data.get("status", "unknown"),
server_name=data.get("name", "A2A Agent"),
server_version=data.get("version"),
provider_type=self.provider_type,
)
# Fallback: try agent card
result = self._make_request("GET", "/.well-known/agent.json")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
agent_card = result.get("data", {})
return HealthResult(
healthy=True,
status="healthy",
server_name=agent_card.get("name", "A2A Agent"),
server_version=agent_card.get("version"),
provider_type=self.provider_type,
)
@@ -0,0 +1,208 @@
"""
A2U Provider
Provider adapter for Agent-to-User event stream endpoints.
"""
from typing import Any, Dict, Iterator, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class A2UProvider(BaseProvider):
"""
Provider adapter for A2U (Agent-to-User) event stream endpoints.
A2U provides a protocol surface for streaming agent events to users,
typically via SSE or WebSocket. This enables real-time UI updates.
"""
provider_type = "a2u"
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="A2U Event Stream",
description="Agent-to-User event streaming protocol",
capabilities=["event-stream", "subscribe", "unsubscribe"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available A2U event streams."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
endpoints = []
for ep in data.get("endpoints", []):
if ep.get("provider_type") == self.provider_type:
endpoints.append(EndpointInfo(
name=ep.get("name", ""),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
streaming=ep.get("streaming", ["sse"]),
auth_modes=ep.get("auth_modes", ["none"]),
))
return endpoints
# Fallback: check for events endpoint
result = self._make_request("GET", "/a2u/info")
if result.get("error"):
# Return default event stream endpoint
return [EndpointInfo(
name="events",
description="Agent event stream",
provider_type=self.provider_type,
streaming=["sse"],
auth_modes=["none"],
)]
data = result.get("data", {})
streams = data.get("streams", [{"name": "events", "description": "Agent event stream"}])
endpoints = []
for stream in streams:
endpoints.append(EndpointInfo(
name=stream.get("name", "events"),
description=stream.get("description", "Event stream"),
provider_type=self.provider_type,
streaming=["sse", "websocket"],
auth_modes=["none"],
))
return endpoints
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about an A2U event stream."""
endpoints = self.list_endpoints()
for ep in endpoints:
if ep.name == name:
return ep
# Return default info
return EndpointInfo(
name=name,
description=f"Event stream: {name}",
provider_type=self.provider_type,
streaming=["sse"],
auth_modes=["none"],
metadata={
"event_types": [
"agent.started",
"agent.thinking",
"agent.tool_call",
"agent.response",
"agent.completed",
"agent.error",
],
},
)
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Subscribe to an A2U event stream (returns subscription info)."""
body = {
"stream": name,
"filters": input_data.get("filters", []) if input_data else [],
}
result = self._make_request("POST", "/a2u/subscribe", json_data=body)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
return InvokeResult(
ok=True,
status="subscribed",
data={
"subscription_id": data.get("subscription_id"),
"stream_url": data.get("stream_url", f"{self.base_url}/a2u/events/{name}"),
},
)
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""Stream events from an A2U endpoint."""
import json
import urllib.request
stream_path = f"/a2u/events/{name}"
full_url = f"{self.base_url}{stream_path}"
headers = {"Accept": "text/event-stream"}
if self.api_key:
headers["X-API-Key"] = self.api_key
try:
req = urllib.request.Request(full_url, headers=headers, method="GET")
with urllib.request.urlopen(req, timeout=300) as response:
buffer = ""
for line in response:
line = line.decode("utf-8")
buffer += line
if buffer.endswith("\n\n"):
event_type = "event"
event_data = ""
for part in buffer.strip().split("\n"):
if part.startswith("event:"):
event_type = part[6:].strip()
elif part.startswith("data:"):
event_data = part[5:].strip()
if event_data:
try:
yield {"event": event_type, "data": json.loads(event_data)}
except json.JSONDecodeError:
yield {"event": event_type, "data": event_data}
buffer = ""
except Exception as e:
yield {"event": "error", "data": {"error": str(e)}}
def health(self) -> HealthResult:
"""Check A2U server health."""
result = self._make_request("GET", "/health")
if result.get("error"):
# Try a2u-specific health
result = self._make_request("GET", "/a2u/health")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
data = result.get("data", {})
return HealthResult(
healthy=True,
status="healthy",
server_name=data.get("server_name", "A2U Server"),
server_version=data.get("version"),
provider_type=self.provider_type,
)
@@ -0,0 +1,171 @@
"""
Agents API Provider
Provider adapter for agents-as-API endpoints (single agent or multi-agent router).
"""
from typing import Any, Dict, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class AgentsAPIProvider(BaseProvider):
"""
Provider adapter for agents-as-API endpoints.
Connects to FastAPI-based agent servers launched via Agent.launch() or Agents.launch()
and provides unified discovery and invocation interface.
"""
provider_type = "agents-api"
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="Agents API",
description="Agent HTTP API endpoints",
capabilities=["list", "describe", "invoke", "health"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available agent endpoints."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
endpoints = []
for ep in data.get("endpoints", []):
if ep.get("provider_type") == self.provider_type:
endpoints.append(EndpointInfo(
name=ep.get("name", ""),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
version=ep.get("version", "1.0.0"),
streaming=ep.get("streaming", ["none"]),
auth_modes=ep.get("auth_modes", ["none"]),
))
return endpoints
# Fallback: try root endpoint for endpoint list
result = self._make_request("GET", "/")
if result.get("error"):
return []
data = result.get("data", {})
endpoint_paths = data.get("endpoints", [])
endpoints = []
for path in endpoint_paths:
endpoints.append(EndpointInfo(
name=path.lstrip("/"),
description=f"Agent endpoint at {path}",
provider_type=self.provider_type,
streaming=["none"],
auth_modes=["none"],
))
return endpoints
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about an agent endpoint."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
for ep in data.get("endpoints", []):
if ep.get("name") == name:
return EndpointInfo(
name=ep.get("name", name),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
version=ep.get("version", "1.0.0"),
input_schema=ep.get("input_schema"),
output_schema=ep.get("output_schema"),
streaming=ep.get("streaming", ["none"]),
auth_modes=ep.get("auth_modes", ["none"]),
metadata=ep.get("metadata", {}),
)
# Fallback: return basic info
return EndpointInfo(
name=name,
description=f"Agent endpoint: {name}",
provider_type=self.provider_type,
input_schema={"type": "object", "properties": {"query": {"type": "string"}}},
streaming=["none"],
auth_modes=["none"],
)
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Invoke an agent endpoint."""
# Normalize endpoint path
path = name if name.startswith("/") else f"/{name}"
# Build request body
body = input_data or {}
if "query" not in body and config and "query" in config:
body["query"] = config["query"]
result = self._make_request("POST", path, json_data=body)
if result.get("status") == 401:
return InvokeResult(
ok=False,
status="auth_error",
error="Authentication required",
)
if result.get("status") == 404:
return InvokeResult(
ok=False,
status="not_found",
error=f"Endpoint not found: {name}",
)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
return InvokeResult(
ok=True,
status="success",
data=data.get("response", data),
)
def health(self) -> HealthResult:
"""Check agent server health."""
result = self._make_request("GET", "/health")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
data = result.get("data", {})
return HealthResult(
healthy=data.get("status") == "ok",
status=data.get("status", "unknown"),
server_name="PraisonAI Agents API",
provider_type=self.provider_type,
metadata={"endpoints": data.get("endpoints", [])},
)
@@ -0,0 +1,231 @@
"""
Base Provider Interface
Abstract base class for all provider adapters.
Each provider implements discovery, invocation, and health check methods.
"""
from abc import ABC, abstractmethod
from dataclasses import dataclass
from typing import Any, Dict, Iterator, List, Optional
from ..discovery import EndpointInfo, ProviderInfo
@dataclass
class InvokeResult:
"""Result from invoking an endpoint."""
ok: bool
status: str = "success"
data: Optional[Any] = None
error: Optional[str] = None
metadata: Optional[Dict[str, Any]] = None
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary."""
return {
"ok": self.ok,
"status": self.status,
"data": self.data,
"error": self.error,
"metadata": self.metadata,
}
@dataclass
class HealthResult:
"""Result from health check."""
healthy: bool
status: str = "healthy"
server_name: Optional[str] = None
server_version: Optional[str] = None
provider_type: Optional[str] = None
metadata: Optional[Dict[str, Any]] = None
def to_dict(self) -> Dict[str, Any]:
"""Convert to dictionary."""
return {
"healthy": self.healthy,
"status": self.status,
"server_name": self.server_name,
"server_version": self.server_version,
"provider_type": self.provider_type,
"metadata": self.metadata,
}
class BaseProvider(ABC):
"""
Abstract base class for provider adapters.
Each provider type (recipe, agents-api, mcp, etc.) implements this interface
to provide consistent discovery and invocation behavior.
"""
# Provider type identifier
provider_type: str = "base"
def __init__(
self,
base_url: str = "http://localhost:8765",
api_key: Optional[str] = None,
timeout: float = 30.0,
):
"""
Initialize the provider.
Args:
base_url: Base URL of the server
api_key: Optional API key for authentication
timeout: Request timeout in seconds
"""
self.base_url = base_url.rstrip("/")
self.api_key = api_key
self.timeout = timeout
@abstractmethod
def get_provider_info(self) -> ProviderInfo:
"""
Get provider information.
Returns:
ProviderInfo with provider details
"""
pass
@abstractmethod
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""
List available endpoints.
Args:
tags: Optional list of tags to filter by
Returns:
List of EndpointInfo objects
"""
pass
@abstractmethod
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""
Get detailed information about an endpoint.
Args:
name: Endpoint name
Returns:
EndpointInfo or None if not found
"""
pass
@abstractmethod
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""
Invoke an endpoint.
Args:
name: Endpoint name
input_data: Input data for the endpoint
config: Optional configuration overrides
stream: Whether to stream the response
Returns:
InvokeResult with response data
"""
pass
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""
Invoke an endpoint with streaming.
Args:
name: Endpoint name
input_data: Input data for the endpoint
config: Optional configuration overrides
Yields:
Stream events as dictionaries
"""
# Default implementation - subclasses can override
result = self.invoke(name, input_data, config, stream=True)
if result.ok:
yield {"event": "complete", "data": result.data}
else:
yield {"event": "error", "data": {"error": result.error}}
@abstractmethod
def health(self) -> HealthResult:
"""
Check server health.
Returns:
HealthResult with health status
"""
pass
def _make_request(
self,
method: str,
path: str,
json_data: Optional[Dict] = None,
) -> Dict[str, Any]:
"""
Make HTTP request to server.
Args:
method: HTTP method
path: URL path
json_data: Optional JSON body
Returns:
Response data dictionary
"""
import json
import urllib.request
import urllib.error
full_url = f"{self.base_url}{path}"
headers = {"Content-Type": "application/json"}
if self.api_key:
headers["X-API-Key"] = self.api_key
data = None
if json_data:
data = json.dumps(json_data).encode("utf-8")
try:
req = urllib.request.Request(
full_url,
data=data,
headers=headers,
method=method,
)
with urllib.request.urlopen(req, timeout=self.timeout) as response:
body = response.read().decode("utf-8")
return {"status": response.status, "data": json.loads(body) if body else {}}
except urllib.error.HTTPError as e:
body = e.read().decode("utf-8") if e.fp else ""
try:
error_data = json.loads(body) if body else {}
except json.JSONDecodeError:
error_data = {"message": body}
return {"status": e.code, "error": error_data}
except urllib.error.URLError as e:
return {"status": 0, "error": {"message": f"Connection error: {e.reason}"}}
except Exception as e:
return {"status": 0, "error": {"message": str(e)}}
@@ -0,0 +1,263 @@
"""
MCP Provider
Provider adapter for MCP server endpoints (stdio, http, sse).
"""
from typing import Any, Dict, Iterator, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class MCPProvider(BaseProvider):
"""
Provider adapter for MCP server endpoints.
Supports multiple MCP transport types:
- stdio: Standard input/output (requires subprocess)
- http: HTTP-based MCP server
- sse: Server-Sent Events MCP gateway
"""
provider_type = "mcp"
def __init__(
self,
base_url: str = "http://localhost:8765",
api_key: Optional[str] = None,
timeout: float = 30.0,
transport: str = "http",
command: Optional[str] = None,
args: Optional[List[str]] = None,
env: Optional[Dict[str, str]] = None,
):
"""
Initialize MCP provider.
Args:
base_url: Base URL for HTTP/SSE transport
api_key: Optional API key
timeout: Request timeout
transport: Transport type (stdio, http, sse)
command: Command for stdio transport
args: Command arguments for stdio transport
env: Environment variables for stdio transport
"""
super().__init__(base_url, api_key, timeout)
self.transport = transport
self.command = command
self.args = args or []
self.env = env or {}
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="MCP Server",
description=f"MCP server ({self.transport} transport)",
capabilities=["list-tools", "call-tool", "list-resources"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available MCP tools as endpoints."""
if self.transport == "stdio":
return self._list_stdio_tools(tags)
else:
return self._list_http_tools(tags)
def _list_stdio_tools(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List tools from stdio MCP server."""
try:
from praisonaiagents.mcp import MCP
mcp = MCP(command=self.command, args=self.args, env=self.env)
tools = mcp.get_tools()
endpoints = []
for tool in tools:
tool_name = getattr(tool, 'name', str(tool))
tool_desc = getattr(tool, 'description', '')
endpoints.append(EndpointInfo(
name=tool_name,
description=tool_desc,
provider_type=self.provider_type,
streaming=["mcp-stream"],
auth_modes=["none"],
metadata={"transport": "stdio"},
))
return endpoints
except Exception:
return []
def _list_http_tools(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List tools from HTTP MCP server."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
endpoints = []
for ep in data.get("endpoints", []):
if ep.get("provider_type") == self.provider_type:
endpoints.append(EndpointInfo(
name=ep.get("name", ""),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
streaming=ep.get("streaming", ["mcp-stream"]),
auth_modes=ep.get("auth_modes", ["none"]),
))
return endpoints
# Fallback: try MCP tools endpoint
result = self._make_request("GET", "/mcp/tools")
if result.get("error"):
return []
tools = result.get("data", {}).get("tools", [])
endpoints = []
for tool in tools:
endpoints.append(EndpointInfo(
name=tool.get("name", ""),
description=tool.get("description", ""),
provider_type=self.provider_type,
input_schema=tool.get("inputSchema"),
streaming=["mcp-stream"],
auth_modes=["none"],
))
return endpoints
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about an MCP tool."""
endpoints = self.list_endpoints()
for ep in endpoints:
if ep.name == name:
return ep
return None
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Invoke an MCP tool."""
if self.transport == "stdio":
return self._invoke_stdio(name, input_data)
else:
return self._invoke_http(name, input_data)
def _invoke_stdio(self, name: str, input_data: Optional[Dict[str, Any]] = None) -> InvokeResult:
"""Invoke tool via stdio MCP."""
try:
from praisonaiagents.mcp import MCP
mcp = MCP(command=self.command, args=self.args, env=self.env)
tools = mcp.get_tools()
for tool in tools:
tool_name = getattr(tool, 'name', str(tool))
if tool_name == name:
if callable(tool):
result = tool(**(input_data or {}))
return InvokeResult(
ok=True,
status="success",
data=result,
)
return InvokeResult(
ok=False,
status="not_found",
error=f"Tool not found: {name}",
)
except Exception as e:
return InvokeResult(
ok=False,
status="error",
error=str(e),
)
def _invoke_http(self, name: str, input_data: Optional[Dict[str, Any]] = None) -> InvokeResult:
"""Invoke tool via HTTP MCP."""
body = {
"tool": name,
"arguments": input_data or {},
}
result = self._make_request("POST", "/mcp/tools/call", json_data=body)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
return InvokeResult(
ok=True,
status="success",
data=data.get("result", data),
)
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""Invoke MCP tool with streaming."""
# MCP streaming is tool-specific; for now, wrap sync call
result = self.invoke(name, input_data, config)
if result.ok:
yield {"event": "result", "data": result.data}
else:
yield {"event": "error", "data": {"error": result.error}}
def health(self) -> HealthResult:
"""Check MCP server health."""
if self.transport == "stdio":
# For stdio, check if command exists
import shutil
if self.command and shutil.which(self.command):
return HealthResult(
healthy=True,
status="available",
server_name=self.command,
provider_type=self.provider_type,
metadata={"transport": "stdio"},
)
return HealthResult(
healthy=False,
status="unavailable",
provider_type=self.provider_type,
metadata={"error": f"Command not found: {self.command}"},
)
# HTTP health check
result = self._make_request("GET", "/health")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
data = result.get("data", {})
return HealthResult(
healthy=True,
status="healthy",
server_name=data.get("server_name", "MCP Server"),
server_version=data.get("version"),
provider_type=self.provider_type,
)
@@ -0,0 +1,206 @@
"""
Recipe Provider
Provider adapter for recipe runner endpoints.
"""
from typing import Any, Dict, Iterator, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class RecipeProvider(BaseProvider):
"""
Provider adapter for recipe runner endpoints.
Connects to recipe runner server (Starlette-based) and provides
unified discovery and invocation interface.
"""
provider_type = "recipe"
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="Recipe Runner",
description="Recipe execution endpoints",
capabilities=["list", "describe", "invoke", "stream", "validate"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available recipe endpoints."""
path = "/v1/recipes"
if tags:
path += f"?tags={','.join(tags)}"
result = self._make_request("GET", path)
if result.get("error"):
return []
recipes = result.get("data", {}).get("recipes", [])
endpoints = []
for r in recipes:
endpoints.append(EndpointInfo(
name=r.get("name", ""),
description=r.get("description", ""),
provider_type=self.provider_type,
tags=r.get("tags", []),
version=r.get("version", "1.0.0"),
streaming=["none", "sse"],
auth_modes=["none", "api-key"],
))
return endpoints
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about a recipe endpoint."""
result = self._make_request("GET", f"/v1/recipes/{name}")
if result.get("status") == 404 or result.get("error"):
return None
data = result.get("data", {})
# Get schema
schema_result = self._make_request("GET", f"/v1/recipes/{name}/schema")
schema_data = schema_result.get("data", {}) if not schema_result.get("error") else {}
return EndpointInfo(
name=data.get("name", name),
description=data.get("description", ""),
provider_type=self.provider_type,
tags=data.get("tags", []),
version=data.get("version", "1.0.0"),
input_schema=schema_data.get("input_schema"),
output_schema=schema_data.get("output_schema"),
streaming=["none", "sse"],
auth_modes=["none", "api-key"],
metadata=data.get("metadata", {}),
)
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Invoke a recipe endpoint."""
body = {
"recipe": name,
"input": input_data or {},
"config": config or {},
}
result = self._make_request("POST", "/v1/recipes/run", json_data=body)
if result.get("status") == 401:
return InvokeResult(
ok=False,
status="auth_error",
error="Authentication required",
)
if result.get("status") == 404:
return InvokeResult(
ok=False,
status="not_found",
error=f"Recipe not found: {name}",
)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
return InvokeResult(
ok=data.get("ok", True),
status=data.get("status", "success"),
data=data.get("output"),
metadata={"run_id": data.get("run_id")},
)
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""Invoke a recipe endpoint with streaming."""
import json
import urllib.request
body = {
"recipe": name,
"input": input_data or {},
"config": config or {},
}
full_url = f"{self.base_url}/v1/recipes/stream"
headers = {
"Content-Type": "application/json",
"Accept": "text/event-stream",
}
if self.api_key:
headers["X-API-Key"] = self.api_key
data = json.dumps(body).encode("utf-8")
try:
req = urllib.request.Request(full_url, data=data, headers=headers, method="POST")
with urllib.request.urlopen(req, timeout=300) as response:
buffer = ""
for line in response:
line = line.decode("utf-8")
buffer += line
if buffer.endswith("\n\n"):
event_type = "message"
event_data = ""
for part in buffer.strip().split("\n"):
if part.startswith("event:"):
event_type = part[6:].strip()
elif part.startswith("data:"):
event_data = part[5:].strip()
if event_data:
try:
yield {"event": event_type, "data": json.loads(event_data)}
except json.JSONDecodeError:
yield {"event": event_type, "data": event_data}
buffer = ""
except Exception as e:
yield {"event": "error", "data": {"error": str(e)}}
def health(self) -> HealthResult:
"""Check recipe server health."""
result = self._make_request("GET", "/health")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
data = result.get("data", {})
return HealthResult(
healthy=data.get("status") == "healthy",
status=data.get("status", "unknown"),
server_name=data.get("service"),
server_version=data.get("version"),
provider_type=self.provider_type,
)
@@ -0,0 +1,150 @@
"""
Tools MCP Provider
Provider adapter for tools exposed as MCP server.
"""
from typing import Any, Dict, Iterator, List, Optional
from .base import BaseProvider, InvokeResult, HealthResult
from ..discovery import EndpointInfo, ProviderInfo
class ToolsMCPProvider(BaseProvider):
"""
Provider adapter for tools exposed as MCP server.
This provider connects to a ToolsMCPServer that exposes Python functions
as MCP tools.
"""
provider_type = "tools-mcp"
def get_provider_info(self) -> ProviderInfo:
"""Get provider information."""
return ProviderInfo(
type=self.provider_type,
name="Tools MCP Server",
description="Python tools exposed as MCP server",
capabilities=["list-tools", "call-tool"],
)
def list_endpoints(self, tags: Optional[List[str]] = None) -> List[EndpointInfo]:
"""List available tools as endpoints."""
# Try unified discovery first
result = self._make_request("GET", "/__praisonai__/discovery")
if not result.get("error") and result.get("data"):
data = result.get("data", {})
endpoints = []
for ep in data.get("endpoints", []):
if ep.get("provider_type") == self.provider_type:
endpoints.append(EndpointInfo(
name=ep.get("name", ""),
description=ep.get("description", ""),
provider_type=self.provider_type,
tags=ep.get("tags", []),
input_schema=ep.get("input_schema"),
streaming=ep.get("streaming", ["none"]),
auth_modes=ep.get("auth_modes", ["none"]),
))
return endpoints
# Fallback: try tools endpoint
result = self._make_request("GET", "/tools")
if result.get("error"):
return []
tools = result.get("data", {}).get("tools", [])
endpoints = []
for tool in tools:
endpoints.append(EndpointInfo(
name=tool.get("name", ""),
description=tool.get("description", ""),
provider_type=self.provider_type,
input_schema=tool.get("input_schema"),
streaming=["none"],
auth_modes=["none"],
))
return endpoints
def describe_endpoint(self, name: str) -> Optional[EndpointInfo]:
"""Get detailed information about a tool."""
endpoints = self.list_endpoints()
for ep in endpoints:
if ep.name == name:
return ep
return None
def invoke(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
stream: bool = False,
) -> InvokeResult:
"""Invoke a tool."""
body = {
"tool": name,
"arguments": input_data or {},
}
result = self._make_request("POST", "/tools/call", json_data=body)
if result.get("status") == 404:
return InvokeResult(
ok=False,
status="not_found",
error=f"Tool not found: {name}",
)
if result.get("error"):
return InvokeResult(
ok=False,
status="error",
error=result["error"].get("message", str(result["error"])),
)
data = result.get("data", {})
return InvokeResult(
ok=True,
status="success",
data=data.get("result", data),
)
def invoke_stream(
self,
name: str,
input_data: Optional[Dict[str, Any]] = None,
config: Optional[Dict[str, Any]] = None,
) -> Iterator[Dict[str, Any]]:
"""Invoke tool with streaming (wraps sync call)."""
result = self.invoke(name, input_data, config)
if result.ok:
yield {"event": "result", "data": result.data}
else:
yield {"event": "error", "data": {"error": result.error}}
def health(self) -> HealthResult:
"""Check tools server health."""
result = self._make_request("GET", "/health")
if result.get("error"):
return HealthResult(
healthy=False,
status="unhealthy",
provider_type=self.provider_type,
metadata={"error": result["error"].get("message", str(result["error"]))},
)
data = result.get("data", {})
return HealthResult(
healthy=True,
status="healthy",
server_name=data.get("server_name", "Tools MCP Server"),
server_version=data.get("version"),
provider_type=self.provider_type,
)
@@ -0,0 +1,131 @@
"""
Provider Registry
Central registry for provider adapters.
"""
from typing import Dict, List, Optional, Type
from .providers.base import BaseProvider
# Global provider registry
_providers: Dict[str, Type[BaseProvider]] = {}
def register_provider(provider_type: str, provider_class: Type[BaseProvider]) -> None:
"""
Register a provider class.
Args:
provider_type: Provider type identifier
provider_class: Provider class to register
"""
_providers[provider_type] = provider_class
def get_provider(
provider_type: str,
base_url: str = "http://localhost:8765",
api_key: Optional[str] = None,
**kwargs,
) -> Optional[BaseProvider]:
"""
Get a provider instance by type.
Args:
provider_type: Provider type identifier
base_url: Base URL for the provider
api_key: Optional API key
**kwargs: Additional provider-specific arguments
Returns:
Provider instance or None if not found
"""
# Lazy register built-in providers
_ensure_providers_registered()
if provider_type not in _providers:
return None
return _providers[provider_type](base_url=base_url, api_key=api_key, **kwargs)
def list_provider_types() -> List[str]:
"""
List all registered provider types.
Returns:
List of provider type identifiers
"""
_ensure_providers_registered()
return list(_providers.keys())
def get_provider_class(provider_type: str) -> Optional[Type[BaseProvider]]:
"""
Get a provider class by type.
Args:
provider_type: Provider type identifier
Returns:
Provider class or None if not found
"""
_ensure_providers_registered()
return _providers.get(provider_type)
def _ensure_providers_registered() -> None:
"""Ensure built-in providers are registered."""
if _providers:
return
# Lazy import and register built-in providers
from .providers.recipe import RecipeProvider
from .providers.agents_api import AgentsAPIProvider
from .providers.mcp import MCPProvider
from .providers.tools_mcp import ToolsMCPProvider
from .providers.a2a import A2AProvider
from .providers.a2u import A2UProvider
register_provider("recipe", RecipeProvider)
register_provider("agents-api", AgentsAPIProvider)
register_provider("mcp", MCPProvider)
register_provider("tools-mcp", ToolsMCPProvider)
register_provider("a2a", A2AProvider)
register_provider("a2u", A2UProvider)
class ProviderRegistry:
"""
Provider registry class for managing provider instances.
This class provides a more object-oriented interface to the provider registry.
"""
def __init__(self):
"""Initialize the registry."""
_ensure_providers_registered()
def get(
self,
provider_type: str,
base_url: str = "http://localhost:8765",
api_key: Optional[str] = None,
**kwargs,
) -> Optional[BaseProvider]:
"""Get a provider instance."""
return get_provider(provider_type, base_url, api_key, **kwargs)
def register(self, provider_type: str, provider_class: Type[BaseProvider]) -> None:
"""Register a provider class."""
register_provider(provider_type, provider_class)
def list_types(self) -> List[str]:
"""List all provider types."""
return list_provider_types()
def get_class(self, provider_type: str) -> Optional[Type[BaseProvider]]:
"""Get a provider class."""
return get_provider_class(provider_type)
+161
View File
@@ -0,0 +1,161 @@
"""
Unified Server Utilities
Provides utilities for adding discovery endpoints to any server.
"""
from typing import Any, List, Optional
from .discovery import (
DiscoveryDocument,
EndpointInfo,
ProviderInfo,
SCHEMA_VERSION,
create_discovery_document,
)
def add_discovery_routes(
app: Any,
discovery: DiscoveryDocument,
path: str = "/__praisonai__/discovery",
) -> None:
"""
Add unified discovery routes to a Starlette/FastAPI application.
Args:
app: Starlette or FastAPI application
discovery: DiscoveryDocument to serve
path: Path for discovery endpoint
"""
try:
from starlette.responses import JSONResponse
from starlette.routing import Route
except ImportError:
# Try FastAPI
try:
from fastapi.responses import JSONResponse
except ImportError:
raise ImportError("Starlette or FastAPI required for discovery routes")
async def discovery_handler(request):
"""Return discovery document."""
return JSONResponse(discovery.to_dict())
async def health_handler(request):
"""Return health status with discovery info."""
return JSONResponse({
"status": "healthy",
"schema_version": SCHEMA_VERSION,
"server_name": discovery.server_name,
"server_version": discovery.server_version,
"providers": [p.type for p in discovery.providers],
"endpoint_count": len(discovery.endpoints),
})
# Add routes based on app type
if hasattr(app, 'add_api_route'):
# FastAPI
app.add_api_route(path, discovery_handler, methods=["GET"])
if not _has_route(app, "/health"):
app.add_api_route("/health", health_handler, methods=["GET"])
elif hasattr(app, 'routes'):
# Starlette
app.routes.append(Route(path, discovery_handler, methods=["GET"]))
if not _has_route(app, "/health"):
app.routes.append(Route("/health", health_handler, methods=["GET"]))
def _has_route(app: Any, path: str) -> bool:
"""Check if app already has a route at path."""
if hasattr(app, 'routes'):
for route in app.routes:
if hasattr(route, 'path') and route.path == path:
return True
return False
def create_unified_app(
providers: Optional[List[str]] = None,
server_name: str = "praisonai",
host: str = "127.0.0.1",
port: int = 8765,
cors_origins: Optional[List[str]] = None,
api_key: Optional[str] = None,
) -> Any:
"""
Create a unified server application with discovery support.
Args:
providers: List of provider types to enable
server_name: Server name for discovery
host: Server host
port: Server port
cors_origins: CORS allowed origins
api_key: Optional API key for authentication
Returns:
FastAPI application
"""
try:
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
except ImportError:
raise ImportError("FastAPI required. Install with: pip install fastapi")
# Create discovery document
discovery = create_discovery_document(server_name=server_name)
# Create FastAPI app
app = FastAPI(
title=f"{server_name} Unified API",
description="Unified PraisonAI endpoints server",
version=discovery.server_version,
)
# Add CORS if configured
if cors_origins:
app.add_middleware(
CORSMiddleware,
allow_origins=cors_origins,
allow_methods=["*"],
allow_headers=["*"],
)
# Add discovery routes
add_discovery_routes(app, discovery)
# Store discovery document for later modification
app.state.discovery = discovery
return app
def register_endpoint_to_discovery(
app: Any,
endpoint: EndpointInfo,
) -> None:
"""
Register an endpoint to the app's discovery document.
Args:
app: FastAPI/Starlette app with discovery
endpoint: EndpointInfo to register
"""
if hasattr(app, 'state') and hasattr(app.state, 'discovery'):
app.state.discovery.add_endpoint(endpoint)
def register_provider_to_discovery(
app: Any,
provider: ProviderInfo,
) -> None:
"""
Register a provider to the app's discovery document.
Args:
app: FastAPI/Starlette app with discovery
provider: ProviderInfo to register
"""
if hasattr(app, 'state') and hasattr(app.state, 'discovery'):
app.state.discovery.add_provider(provider)
+1 -1
View File
@@ -1 +1 @@
__version__ = "2.7.3"
__version__ = "2.7.4"
@@ -0,0 +1,470 @@
"""
Unit tests for unified endpoints module.
Tests discovery schema, provider adapters, registry, and CLI.
"""
import pytest
from unittest.mock import patch, MagicMock
class TestDiscoverySchema:
"""Tests for discovery schema."""
def test_discovery_document_creation(self):
"""Test creating a discovery document."""
from praisonai.endpoints.discovery import (
DiscoveryDocument,
create_discovery_document,
SCHEMA_VERSION,
)
doc = create_discovery_document(server_name="test-server")
assert doc.server_name == "test-server"
assert doc.schema_version == SCHEMA_VERSION
assert isinstance(doc.providers, list)
assert isinstance(doc.endpoints, list)
def test_endpoint_info_creation(self):
"""Test creating endpoint info."""
from praisonai.endpoints.discovery import EndpointInfo
endpoint = EndpointInfo(
name="test-endpoint",
description="Test endpoint",
provider_type="recipe",
tags=["test", "demo"],
)
assert endpoint.name == "test-endpoint"
assert endpoint.provider_type == "recipe"
assert "test" in endpoint.tags
def test_provider_info_creation(self):
"""Test creating provider info."""
from praisonai.endpoints.discovery import ProviderInfo
provider = ProviderInfo(
type="agents-api",
name="Agents API",
capabilities=["invoke", "health"],
)
assert provider.type == "agents-api"
assert "invoke" in provider.capabilities
def test_discovery_document_to_dict(self):
"""Test converting discovery document to dict."""
from praisonai.endpoints.discovery import (
DiscoveryDocument,
EndpointInfo,
ProviderInfo,
)
doc = DiscoveryDocument(server_name="test")
doc.add_provider(ProviderInfo(type="recipe", name="Recipe"))
doc.add_endpoint(EndpointInfo(name="test-ep", provider_type="recipe"))
data = doc.to_dict()
assert data["server_name"] == "test"
assert len(data["providers"]) == 1
assert len(data["endpoints"]) == 1
assert data["providers"][0]["type"] == "recipe"
def test_get_endpoints_by_type(self):
"""Test filtering endpoints by type."""
from praisonai.endpoints.discovery import DiscoveryDocument, EndpointInfo
doc = DiscoveryDocument()
doc.add_endpoint(EndpointInfo(name="ep1", provider_type="recipe"))
doc.add_endpoint(EndpointInfo(name="ep2", provider_type="agents-api"))
doc.add_endpoint(EndpointInfo(name="ep3", provider_type="recipe"))
recipe_eps = doc.get_endpoints_by_type("recipe")
assert len(recipe_eps) == 2
assert all(ep.provider_type == "recipe" for ep in recipe_eps)
def test_get_endpoint_by_name(self):
"""Test getting endpoint by name."""
from praisonai.endpoints.discovery import DiscoveryDocument, EndpointInfo
doc = DiscoveryDocument()
doc.add_endpoint(EndpointInfo(name="my-endpoint", description="Test"))
ep = doc.get_endpoint_by_name("my-endpoint")
assert ep is not None
assert ep.name == "my-endpoint"
missing = doc.get_endpoint_by_name("nonexistent")
assert missing is None
class TestProviderRegistry:
"""Tests for provider registry."""
def test_list_provider_types(self):
"""Test listing provider types."""
from praisonai.endpoints.registry import list_provider_types
types = list_provider_types()
assert "recipe" in types
assert "agents-api" in types
assert "mcp" in types
assert "a2a" in types
assert "a2u" in types
def test_get_provider(self):
"""Test getting a provider instance."""
from praisonai.endpoints.registry import get_provider
provider = get_provider("recipe", base_url="http://localhost:8000")
assert provider is not None
assert provider.provider_type == "recipe"
assert provider.base_url == "http://localhost:8000"
def test_get_unknown_provider(self):
"""Test getting unknown provider returns None."""
from praisonai.endpoints.registry import get_provider
provider = get_provider("unknown-type")
assert provider is None
def test_provider_registry_class(self):
"""Test ProviderRegistry class."""
from praisonai.endpoints.registry import ProviderRegistry
registry = ProviderRegistry()
assert "recipe" in registry.list_types()
provider = registry.get("agents-api")
assert provider is not None
class TestRecipeProvider:
"""Tests for recipe provider."""
def test_provider_info(self):
"""Test recipe provider info."""
from praisonai.endpoints.providers.recipe import RecipeProvider
provider = RecipeProvider()
info = provider.get_provider_info()
assert info.type == "recipe"
assert "invoke" in info.capabilities
@patch('praisonai.endpoints.providers.base.BaseProvider._make_request')
def test_list_endpoints(self, mock_request):
"""Test listing recipe endpoints."""
from praisonai.endpoints.providers.recipe import RecipeProvider
mock_request.return_value = {
"data": {
"recipes": [
{"name": "recipe1", "description": "Test 1"},
{"name": "recipe2", "description": "Test 2"},
]
}
}
provider = RecipeProvider()
endpoints = provider.list_endpoints()
assert len(endpoints) == 2
assert endpoints[0].name == "recipe1"
@patch('praisonai.endpoints.providers.base.BaseProvider._make_request')
def test_health_check(self, mock_request):
"""Test recipe health check."""
from praisonai.endpoints.providers.recipe import RecipeProvider
mock_request.return_value = {
"data": {"status": "healthy", "service": "recipe-runner"}
}
provider = RecipeProvider()
health = provider.health()
assert health.healthy is True
assert health.status == "healthy"
class TestAgentsAPIProvider:
"""Tests for agents-api provider."""
def test_provider_info(self):
"""Test agents-api provider info."""
from praisonai.endpoints.providers.agents_api import AgentsAPIProvider
provider = AgentsAPIProvider()
info = provider.get_provider_info()
assert info.type == "agents-api"
@patch('praisonai.endpoints.providers.base.BaseProvider._make_request')
def test_invoke(self, mock_request):
"""Test invoking agents-api endpoint."""
from praisonai.endpoints.providers.agents_api import AgentsAPIProvider
mock_request.return_value = {
"status": 200,
"data": {"response": "Hello!"}
}
provider = AgentsAPIProvider()
result = provider.invoke("test-agent", {"query": "Hi"})
assert result.ok is True
assert result.data == "Hello!"
class TestA2AProvider:
"""Tests for A2A provider."""
def test_provider_info(self):
"""Test A2A provider info."""
from praisonai.endpoints.providers.a2a import A2AProvider
provider = A2AProvider()
info = provider.get_provider_info()
assert info.type == "a2a"
assert "agent-card" in info.capabilities
class TestA2UProvider:
"""Tests for A2U provider."""
def test_provider_info(self):
"""Test A2U provider info."""
from praisonai.endpoints.providers.a2u import A2UProvider
provider = A2UProvider()
info = provider.get_provider_info()
assert info.type == "a2u"
assert "subscribe" in info.capabilities
class TestEndpointsCLI:
"""Tests for endpoints CLI."""
def test_handler_initialization(self):
"""Test CLI handler initialization."""
from praisonai.cli.features.endpoints import EndpointsHandler
handler = EndpointsHandler()
assert handler.base_url == "http://localhost:8765"
def test_help_command(self):
"""Test help command."""
from praisonai.cli.features.endpoints import EndpointsHandler
handler = EndpointsHandler()
result = handler.handle(["help"])
assert result == 0
def test_types_command(self):
"""Test types command."""
from praisonai.cli.features.endpoints import EndpointsHandler
handler = EndpointsHandler()
result = handler.cmd_types(["--format", "json"])
assert result == 0
def test_parse_args(self):
"""Test argument parsing."""
from praisonai.cli.features.endpoints import EndpointsHandler
handler = EndpointsHandler()
spec = {
"name": {"positional": True, "default": ""},
"url": {"default": None},
"json": {"flag": True, "default": False},
}
parsed = handler._parse_args(["my-endpoint", "--url", "http://test", "--json"], spec)
assert parsed["name"] == "my-endpoint"
assert parsed["url"] == "http://test"
assert parsed["json"] is True
class TestServeCLI:
"""Tests for serve CLI."""
def test_handler_initialization(self):
"""Test serve handler initialization."""
from praisonai.cli.features.serve import ServeHandler
handler = ServeHandler()
assert handler.DEFAULT_PORT == 8765
def test_help_command(self):
"""Test help command."""
from praisonai.cli.features.serve import ServeHandler
handler = ServeHandler()
result = handler.handle(["help"])
assert result == 0
def test_parse_args(self):
"""Test argument parsing."""
from praisonai.cli.features.serve import ServeHandler
handler = ServeHandler()
spec = {
"host": {"default": "127.0.0.1"},
"port": {"default": 8765, "type": "int"},
"reload": {"flag": True, "default": False},
}
parsed = handler._parse_args(["--host", "0.0.0.0", "--port", "9000", "--reload"], spec)
assert parsed["host"] == "0.0.0.0"
assert parsed["port"] == 9000
assert parsed["reload"] is True
class TestA2UServer:
"""Tests for A2U server components."""
def test_event_creation(self):
"""Test A2U event creation."""
from praisonai.endpoints.a2u_server import A2UEvent
event = A2UEvent(event_type="test", data={"key": "value"})
assert event.event_type == "test"
assert event.data["key"] == "value"
assert event.event_id is not None
def test_event_to_sse(self):
"""Test converting event to SSE format."""
from praisonai.endpoints.a2u_server import A2UEvent
event = A2UEvent(event_type="message", data={"text": "hello"})
sse = event.to_sse()
assert "event: message" in sse
assert "data:" in sse
assert "hello" in sse
def test_subscription_creation(self):
"""Test A2U subscription creation."""
from praisonai.endpoints.a2u_server import A2USubscription
sub = A2USubscription(
subscription_id="sub-123",
stream_name="events",
filters=["agent.started", "agent.completed"],
)
assert sub.subscription_id == "sub-123"
assert sub.stream_name == "events"
assert len(sub.filters) == 2
def test_subscription_matches_event(self):
"""Test subscription event matching."""
from praisonai.endpoints.a2u_server import A2USubscription, A2UEvent
sub = A2USubscription(
subscription_id="sub-123",
stream_name="events",
filters=["agent.started"],
)
event1 = A2UEvent(event_type="agent.started", data={})
event2 = A2UEvent(event_type="agent.completed", data={})
assert sub.matches_event(event1) is True
assert sub.matches_event(event2) is False
def test_subscription_matches_all_without_filters(self):
"""Test subscription matches all events without filters."""
from praisonai.endpoints.a2u_server import A2USubscription, A2UEvent
sub = A2USubscription(
subscription_id="sub-123",
stream_name="events",
)
event = A2UEvent(event_type="any.event", data={})
assert sub.matches_event(event) is True
def test_event_bus_subscribe(self):
"""Test event bus subscription."""
from praisonai.endpoints.a2u_server import A2UEventBus
bus = A2UEventBus()
sub = bus.subscribe("events", ["agent.started"])
assert sub.subscription_id.startswith("sub-")
assert sub.stream_name == "events"
def test_event_bus_unsubscribe(self):
"""Test event bus unsubscription."""
from praisonai.endpoints.a2u_server import A2UEventBus
bus = A2UEventBus()
sub = bus.subscribe("events")
result = bus.unsubscribe(sub.subscription_id)
assert result is True
result = bus.unsubscribe("nonexistent")
assert result is False
def test_emit_helpers(self):
"""Test emit helper functions."""
from praisonai.endpoints.a2u_server import (
emit_agent_started,
emit_agent_thinking,
emit_agent_response,
emit_agent_completed,
emit_agent_error,
)
# These should not raise
emit_agent_started("agent-1", "TestAgent")
emit_agent_thinking("agent-1", "Processing...")
emit_agent_response("agent-1", "Hello!")
emit_agent_completed("agent-1", {"result": "done"})
emit_agent_error("agent-1", "Something went wrong")
class TestServerUtilities:
"""Tests for server utilities."""
def test_add_discovery_routes(self):
"""Test adding discovery routes to app."""
from praisonai.endpoints.server import add_discovery_routes
from praisonai.endpoints.discovery import create_discovery_document
# Mock FastAPI app
mock_app = MagicMock()
mock_app.add_api_route = MagicMock()
discovery = create_discovery_document()
add_discovery_routes(mock_app, discovery)
# Should have added routes
assert mock_app.add_api_route.called
if __name__ == "__main__":
pytest.main([__file__, "-v"])