PraisonAI 3.0.0__py3-none-any.whl

praisonai/endpoints/providers/recipe.py

@@ -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,
+        )

praisonai/endpoints/providers/tools_mcp.py

@@ -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,
+        )

praisonai/endpoints/registry.py

@@ -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)

praisonai/endpoints/server.py

@@ -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)

praisonai/inbuilt_tools/__init__.py

@@ -0,0 +1,24 @@
+# Only try to import autogen_tools if either CrewAI or AG2 is available
+CREWAI_AVAILABLE = False
+AUTOGEN_AVAILABLE = False
+PRAISONAI_TOOLS_AVAILABLE = False
+
+try:
+    from crewai import Agent, Task, Crew
+    CREWAI_AVAILABLE = True
+except ImportError:
+    pass
+
+try:
+    import autogen
+    AUTOGEN_AVAILABLE = True
+except ImportError:
+    pass
+
+# Only try to import tools if a framework is available
+if CREWAI_AVAILABLE or AUTOGEN_AVAILABLE:
+    try:
+        from .autogen_tools import *
+        PRAISONAI_TOOLS_AVAILABLE = True
+    except ImportError:
+        pass