machinaos 0.0.1

package/server/models/nodes.py

@@ -0,0 +1,455 @@
+"""Pydantic models for node validation with discriminated unions.
+
+This module provides type-safe node validation using Pydantic v2 discriminated unions.
+Benefits:
+- O(1) hash map lookup vs O(n) sequential validation
+- Better error messages (identifies exact expected variant)
+- Type safety with IDE autocompletion
+- Automatic API documentation generation
+"""
+
+from typing import Literal, Union, Annotated, Optional, Dict, Any, List
+from pydantic import BaseModel, Field, TypeAdapter
+from datetime import datetime
+
+from constants import (
+    AI_CHAT_MODEL_TYPES,
+    AI_AGENT_TYPES,
+    AI_MEMORY_TYPES,
+    GOOGLE_MAPS_TYPES,
+    ANDROID_SERVICE_NODE_TYPES,
+    ANDROID_SETUP_TYPES,
+    WHATSAPP_TYPES,
+    CHAT_TYPES,
+    CODE_EXECUTOR_TYPES,
+    HTTP_TYPES,
+    TEXT_TYPES,
+    WORKFLOW_CONTROL_TYPES,
+    TRIGGER_TYPES,
+)
+
+
+# =============================================================================
+# BASE MODELS
+# =============================================================================
+
+class BaseNodeParams(BaseModel):
+    """Base class for all node parameters."""
+    model_config = {"extra": "allow"}  # Allow extra fields for flexibility
+
+
+# =============================================================================
+# AI NODE MODELS
+# =============================================================================
+
+class AIChatModelParams(BaseNodeParams):
+    """Parameters for AI chat model nodes (OpenAI, Anthropic, Gemini, OpenRouter, Groq, Cerebras)."""
+    type: Literal["openaiChatModel", "anthropicChatModel", "geminiChatModel", "openrouterChatModel", "groqChatModel", "cerebrasChatModel"]
+    prompt: str = ""
+    model: str = ""
+    temperature: float = Field(default=0.7, ge=0.0, le=2.0)
+    max_tokens: Optional[int] = Field(default=1000, alias="maxTokens")
+    system_prompt: Optional[str] = Field(default="", alias="systemMessage")
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+class AIAgentParams(BaseNodeParams):
+    """Parameters for AI agent node."""
+    type: Literal["aiAgent"]
+    prompt: str = ""
+    provider: Literal["openai", "anthropic", "gemini"] = "openai"
+    model: str = ""
+    temperature: float = Field(default=0.7, ge=0.0, le=2.0)
+    max_tokens: Optional[int] = Field(default=1000, alias="maxTokens")
+    system_message: Optional[str] = Field(default="You are a helpful assistant", alias="systemMessage")
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+class ChatAgentParams(BaseNodeParams):
+    """Parameters for chat agent node (skill-based)."""
+    type: Literal["chatAgent"]
+    provider: Literal["openai", "anthropic", "gemini", "groq", "openrouter", "cerebras"] = "openai"
+    model: str = ""
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+class SimpleMemoryParams(BaseNodeParams):
+    """Parameters for simple memory node."""
+    type: Literal["simpleMemory"]
+    session_id: str = Field(default="default", alias="sessionId")
+    memory_type: Literal["buffer", "window"] = Field(default="buffer", alias="memoryType")
+    window_size: int = Field(default=10, alias="windowSize", ge=1, le=100)
+    clear_on_run: bool = Field(default=False, alias="clearOnRun")
+
+
+# =============================================================================
+# GOOGLE MAPS NODE MODELS
+# =============================================================================
+
+class CreateMapParams(BaseNodeParams):
+    """Parameters for create map node."""
+    type: Literal["createMap"]
+    center_lat: float = Field(default=0.0, alias="centerLat")
+    center_lng: float = Field(default=0.0, alias="centerLng")
+    zoom: int = Field(default=10, ge=1, le=20)
+    map_type: Literal["roadmap", "satellite", "terrain", "hybrid"] = Field(default="roadmap", alias="mapType")
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+class AddLocationsParams(BaseNodeParams):
+    """Parameters for add locations (geocoding) node."""
+    type: Literal["addLocations"]
+    address: str = ""
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+class ShowNearbyPlacesParams(BaseNodeParams):
+    """Parameters for show nearby places node."""
+    type: Literal["showNearbyPlaces"]
+    latitude: float = 0.0
+    longitude: float = 0.0
+    radius: int = Field(default=1000, ge=1, le=50000)
+    place_type: str = Field(default="restaurant", alias="placeType")
+    api_key: Optional[str] = Field(default=None, alias="apiKey")
+
+
+# =============================================================================
+# ANDROID NODE MODELS
+# =============================================================================
+
+class AndroidDeviceSetupParams(BaseNodeParams):
+    """Parameters for Android device setup node."""
+    type: Literal["androidDeviceSetup"]
+    connection_type: Literal["local", "remote"] = Field(default="local", alias="connection_type")
+    device_id: str = Field(default="", alias="device_id")
+    websocket_url: str = Field(default="", alias="websocket_url")
+    port: int = Field(default=8888, ge=1, le=65535)
+    auto_forward: bool = Field(default=True, alias="auto_forward")
+    target_device_id: str = Field(default="", alias="target_device_id")
+
+
+class AndroidServiceParams(BaseNodeParams):
+    """Parameters for Android service nodes (battery, network, wifi, etc.)."""
+    type: str  # Validated separately against ANDROID_SERVICE_NODE_TYPES
+    service_id: str = Field(default="battery", alias="service_id")
+    action: str = Field(default="status")
+    parameters: Dict[str, Any] = Field(default_factory=dict)
+    android_host: str = Field(default="localhost", alias="android_host")
+    android_port: int = Field(default=8888, alias="android_port", ge=1, le=65535)
+
+
+# =============================================================================
+# WHATSAPP NODE MODELS
+# =============================================================================
+
+class WhatsAppSendParams(BaseNodeParams):
+    """Parameters for WhatsApp send message node."""
+    type: Literal["whatsappSend"]
+    phone_number: str = Field(default="", alias="phoneNumber")
+    message: str = ""
+    message_type: Literal["text", "image", "video", "audio", "document"] = Field(default="text", alias="messageType")
+
+
+class WhatsAppConnectParams(BaseNodeParams):
+    """Parameters for WhatsApp connect node."""
+    type: Literal["whatsappConnect"]
+
+
+class WhatsAppReceiveParams(BaseNodeParams):
+    """Parameters for WhatsApp receive (trigger) node."""
+    type: Literal["whatsappReceive"]
+    message_type_filter: str = Field(default="all", alias="messageTypeFilter")
+    sender_filter: str = Field(default="all", alias="filter")
+    ignore_own: bool = Field(default=True, alias="ignoreOwnMessages")
+    include_media: bool = Field(default=False, alias="includeMediaData")
+
+
+# =============================================================================
+# CODE EXECUTOR NODE MODELS
+# =============================================================================
+
+class PythonExecutorParams(BaseNodeParams):
+    """Parameters for Python code executor node."""
+    type: Literal["pythonExecutor"]
+    code: str = ""
+    timeout: int = Field(default=30, ge=1, le=300)
+
+
+class JavaScriptExecutorParams(BaseNodeParams):
+    """Parameters for JavaScript code executor node."""
+    type: Literal["javascriptExecutor"]
+    code: str = ""
+    timeout: int = Field(default=30, ge=1, le=300)
+
+
+# =============================================================================
+# HTTP NODE MODELS
+# =============================================================================
+
+class HttpRequestParams(BaseNodeParams):
+    """Parameters for HTTP request node."""
+    type: Literal["httpRequest"]
+    url: str = ""
+    method: Literal["GET", "POST", "PUT", "DELETE", "PATCH"] = "GET"
+    headers: Dict[str, str] = Field(default_factory=dict)
+    body: str = ""
+    timeout: int = Field(default=30, ge=1, le=300)
+
+
+class WebhookTriggerParams(BaseNodeParams):
+    """Parameters for webhook trigger node."""
+    type: Literal["webhookTrigger"]
+    path: str = ""
+    method_filter: str = Field(default="all", alias="methodFilter")
+    require_auth: bool = Field(default=False, alias="requireAuth")
+
+
+class WebhookResponseParams(BaseNodeParams):
+    """Parameters for webhook response node."""
+    type: Literal["webhookResponse"]
+    status_code: int = Field(default=200, alias="statusCode", ge=100, le=599)
+    content_type: str = Field(default="application/json", alias="contentType")
+    body: str = ""
+
+
+# =============================================================================
+# WORKFLOW CONTROL NODE MODELS
+# =============================================================================
+
+class StartNodeParams(BaseNodeParams):
+    """Parameters for start node."""
+    type: Literal["start"]
+    initial_data: str = Field(default="{}", alias="initialData")
+
+
+class CronSchedulerParams(BaseNodeParams):
+    """Parameters for cron scheduler node."""
+    type: Literal["cronScheduler"]
+    frequency: Literal["seconds", "minutes", "hours", "days", "weeks", "months", "once"] = "minutes"
+    interval: int = Field(default=5, ge=1)
+    interval_minutes: int = Field(default=5, alias="intervalMinutes", ge=1)
+    interval_hours: int = Field(default=1, alias="intervalHours", ge=1)
+    daily_time: str = Field(default="09:00", alias="dailyTime")
+    weekly_time: str = Field(default="09:00", alias="weeklyTime")
+    weekday: str = Field(default="1")
+    monthly_time: str = Field(default="09:00", alias="monthlyTime")
+    month_day: str = Field(default="1", alias="monthDay")
+
+
+class WorkflowTriggerParams(BaseNodeParams):
+    """Parameters for workflow trigger node."""
+    type: Literal["workflowTrigger"]
+
+
+# =============================================================================
+# TEXT NODE MODELS
+# =============================================================================
+
+class TextGeneratorParams(BaseNodeParams):
+    """Parameters for text generator node."""
+    type: Literal["textGenerator"]
+    template: str = ""
+
+
+class FileHandlerParams(BaseNodeParams):
+    """Parameters for file handler node."""
+    type: Literal["fileHandler"]
+    action: Literal["read", "write", "append", "delete"] = "read"
+    file_path: str = Field(default="", alias="filePath")
+    content: str = ""
+
+
+# =============================================================================
+# CHAT NODE MODELS
+# =============================================================================
+
+class ChatSendParams(BaseNodeParams):
+    """Parameters for chat send node."""
+    type: Literal["chatSend"]
+    message: str = ""
+
+
+class ChatHistoryParams(BaseNodeParams):
+    """Parameters for chat history node."""
+    type: Literal["chatHistory"]
+    limit: int = Field(default=50, ge=1, le=1000)
+
+
+# =============================================================================
+# DISCRIMINATED UNION - All Node Types
+# =============================================================================
+
+# AI Nodes
+AINodeParams = Annotated[
+    Union[AIChatModelParams, AIAgentParams, ChatAgentParams, SimpleMemoryParams],
+    Field(discriminator="type")
+]
+
+# Maps Nodes
+MapsNodeParams = Annotated[
+    Union[CreateMapParams, AddLocationsParams, ShowNearbyPlacesParams],
+    Field(discriminator="type")
+]
+
+# WhatsApp Nodes
+WhatsAppNodeParams = Annotated[
+    Union[WhatsAppSendParams, WhatsAppConnectParams, WhatsAppReceiveParams],
+    Field(discriminator="type")
+]
+
+# Code Executor Nodes
+CodeNodeParams = Annotated[
+    Union[PythonExecutorParams, JavaScriptExecutorParams],
+    Field(discriminator="type")
+]
+
+# HTTP Nodes
+HttpNodeParams = Annotated[
+    Union[HttpRequestParams, WebhookTriggerParams, WebhookResponseParams],
+    Field(discriminator="type")
+]
+
+# Workflow Control Nodes
+WorkflowNodeParams = Annotated[
+    Union[StartNodeParams, CronSchedulerParams, WorkflowTriggerParams],
+    Field(discriminator="type")
+]
+
+# Text Nodes
+TextNodeParams = Annotated[
+    Union[TextGeneratorParams, FileHandlerParams],
+    Field(discriminator="type")
+]
+
+# Chat Nodes
+ChatNodeParams = Annotated[
+    Union[ChatSendParams, ChatHistoryParams],
+    Field(discriminator="type")
+]
+
+# Master union of all known node types (excluding Android service which has dynamic types)
+KnownNodeParams = Annotated[
+    Union[
+        # AI
+        AIChatModelParams, AIAgentParams, ChatAgentParams, SimpleMemoryParams,
+        # Maps
+        CreateMapParams, AddLocationsParams, ShowNearbyPlacesParams,
+        # Android Setup
+        AndroidDeviceSetupParams,
+        # WhatsApp
+        WhatsAppSendParams, WhatsAppConnectParams, WhatsAppReceiveParams,
+        # Code
+        PythonExecutorParams, JavaScriptExecutorParams,
+        # HTTP
+        HttpRequestParams, WebhookTriggerParams, WebhookResponseParams,
+        # Workflow
+        StartNodeParams, CronSchedulerParams, WorkflowTriggerParams,
+        # Text
+        TextGeneratorParams, FileHandlerParams,
+        # Chat
+        ChatSendParams, ChatHistoryParams,
+    ],
+    Field(discriminator="type")
+]
+
+
+# =============================================================================
+# EXECUTION REQUEST MODELS
+# =============================================================================
+
+class NodeExecutionRequest(BaseModel):
+    """Request model for node execution."""
+    node_id: str = Field(..., alias="nodeId", min_length=1)
+    node_type: str = Field(..., alias="nodeType", min_length=1)
+    parameters: Dict[str, Any] = Field(default_factory=dict)
+    nodes: Optional[List[Dict[str, Any]]] = None
+    edges: Optional[List[Dict[str, Any]]] = None
+    session_id: str = Field(default="default", alias="sessionId")
+
+    model_config = {"populate_by_name": True}
+
+
+class NodeExecutionResponse(BaseModel):
+    """Response model for node execution."""
+    success: bool
+    node_id: str
+    node_type: str
+    result: Optional[Dict[str, Any]] = None
+    error: Optional[str] = None
+    execution_time: float
+    timestamp: str = Field(default_factory=lambda: datetime.now().isoformat())
+
+
+# =============================================================================
+# VALIDATION HELPERS
+# =============================================================================
+
+# Create TypeAdapter for discriminated union (created once at module level for performance)
+_known_node_adapter = TypeAdapter(KnownNodeParams)
+
+# Set of all known node types for fallback detection
+_all_known_types: set = None  # Lazy initialized
+
+
+def _get_all_known_types() -> set:
+    """Lazily get all known node types."""
+    global _all_known_types
+    if _all_known_types is None:
+        _all_known_types = get_all_node_types()
+    return _all_known_types
+
+
+def validate_node_params(node_type: str, params: Dict[str, Any]) -> BaseNodeParams:
+    """Validate node parameters using the appropriate model.
+
+    Uses Pydantic v2 discriminated unions for O(1) type lookup instead of
+    sequential if/elif validation. The discriminator field 'type' routes
+    to the correct model automatically.
+
+    For known node types, validation errors are raised (e.g., temperature > 2.0).
+    For unknown node types, falls back to BaseNodeParams (allows extensibility).
+
+    Args:
+        node_type: The node type string
+        params: The parameters dictionary
+
+    Returns:
+        Validated parameters model (specific subclass based on node_type)
+
+    Raises:
+        ValidationError: If validation fails for a known node type
+    """
+    # Add type to params for discriminator
+    params_with_type = {"type": node_type, **params}
+
+    # Handle Android service nodes separately (dynamic types not in KnownNodeParams)
+    if node_type in ANDROID_SERVICE_NODE_TYPES:
+        return AndroidServiceParams(**params_with_type)
+
+    # Check if this is a known type
+    if node_type in _get_all_known_types():
+        # Known type - use discriminated union (O(1) lookup)
+        # ValidationError will be raised if params are invalid
+        return _known_node_adapter.validate_python(params_with_type)
+    else:
+        # Unknown type - fall back to base model (allows extensibility)
+        return BaseNodeParams(**params_with_type)
+
+
+def get_all_node_types() -> set:
+    """Get set of all known node types."""
+    return (
+        AI_CHAT_MODEL_TYPES |
+        AI_AGENT_TYPES |
+        AI_MEMORY_TYPES |
+        GOOGLE_MAPS_TYPES |
+        ANDROID_SERVICE_NODE_TYPES |
+        ANDROID_SETUP_TYPES |
+        WHATSAPP_TYPES |
+        CHAT_TYPES |
+        CODE_EXECUTOR_TYPES |
+        HTTP_TYPES |
+        TEXT_TYPES |
+        WORKFLOW_CONTROL_TYPES |
+        TRIGGER_TYPES
+    )

package/server/package.json

@@ -0,0 +1,9 @@
+{
+  "name": "machina-backend",
+  "private": true,
+  "version": "0.0.0",
+  "scripts": {
+    "start": "python -m uvicorn main:app --host 0.0.0.0 --port ${PORT:-3010}",
+    "dev": "python -m uvicorn main:app --host 127.0.0.1 --port ${PORT:-3010} --reload --reload-dir . --reload-exclude \"*.pyc\" --reload-exclude \"__pycache__\" --log-level warning"
+  }
+}

package/server/pyproject.toml

@@ -0,0 +1,72 @@
+[project]
+name = "machinaos-server"
+version = "0.1.0"
+description = "MachinaOS workflow automation backend"
+requires-python = ">=3.11"
+dependencies = [
+    # Core
+    "fastapi>=0.115.0",
+    "uvicorn[standard]>=0.32.0",
+    "pydantic>=2.10.0",
+    "pydantic-settings>=2.6.0",
+
+    # Database
+    "sqlmodel>=0.0.22",
+    "aiosqlite>=0.20.0",
+
+    # HTTP/WebSocket
+    "httpx>=0.28.0",
+    "aiohttp>=3.11.0",
+    "websockets>=14.0",
+
+    # Google Maps
+    "googlemaps>=4.10.0",
+
+    # Logging & Serialization
+    "structlog>=24.1.0",
+    "orjson>=3.9.0",
+
+    # YAML parsing (for SKILL.md files)
+    "pyyaml>=6.0",
+
+    # Auth
+    "bcrypt>=4.2.0",
+    "python-jose[cryptography]>=3.3.0",
+    "email-validator>=2.0.0",
+
+    # AI - All providers (directly imported in ai.py)
+    "langchain-core>=0.3.0",
+    "langchain-openai>=0.2.0",
+    "langchain-anthropic>=0.3.0",
+    "langchain-google-genai>=2.0.0",
+    "langchain-groq>=0.2.0",
+    "langgraph>=0.2.0",
+
+    # Dependency Injection
+    "dependency-injector>=4.40.0",
+
+    # Temporal (distributed execution)
+    "temporalio>=1.7.0",
+
+    # Utils
+    "python-dotenv>=1.0.0",
+    "apscheduler>=3.10.0",
+    "qrcode[pil]>=8.0",
+    "pytz>=2024.1",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0.0",
+    "ruff>=0.8.0",
+]
+extra-ai = [
+    "langchain-cerebras>=0.1.0",
+]
+docs = [
+    "beautifulsoup4>=4.12.0",
+    "pypdf>=5.0.0",
+]
+
+[tool.uv]
+index-url = "https://pypi.org/simple"

package/server/requirements.txt

@@ -0,0 +1,83 @@
+# Core Framework
+fastapi>=0.110.0
+uvicorn[standard]>=0.27.0
+
+# Performance optimization
+uvloop>=0.19.0;sys_platform!='win32'
+orjson>=3.10.0
+
+# Database & ORM (Python 3.13 compatible versions)
+sqlmodel>=0.0.18
+sqlalchemy[asyncio]>=2.0.35
+aiosqlite>=0.20.0
+
+# Settings & Configuration
+pydantic>=2.6.0
+pydantic-settings>=2.2.0
+python-dotenv>=1.0.0
+
+# Dependency Injection
+dependency-injector>=4.41.0
+
+# Caching (Redis) - Python 3.13 compatible
+redis>=5.0.0
+
+# HTTP Client
+httpx>=0.27.0
+
+# WebSocket Client (for WhatsApp RPC)
+websockets>=12.0
+
+# Async HTTP Client (for Android relay WebSocket)
+aiohttp>=3.9.0
+
+# QR Code Generation (for WhatsApp QR display)
+qrcode[pil]>=7.4.0
+
+# Logging
+structlog>=23.2.0
+
+# AI Providers (LangChain) - compatible versions
+# langchain-anthropic 1.3+ requires langchain-core 1.2.6+ for ModelProfile class
+langchain-core>=1.2.6
+langchain-openai>=1.0.0
+langchain-anthropic>=1.3.0
+# Gemini 3+ requires >=3.1.0 for thought_signature support in tool calling
+langchain-google-genai>=3.1.0
+langchain-groq>=0.1.0
+langchain-cerebras>=0.1.0
+
+# LangGraph - State machine for AI agents
+langgraph>=1.0.6
+
+# Temporal - Durable workflow execution (optional, requires TEMPORAL_ENABLED=true)
+temporalio>=1.21.1
+
+# Google Services
+googlemaps>=4.10.0
+
+# Scheduling
+APScheduler>=3.10.0
+
+# Web Search
+duckduckgo-search>=8.0.0
+
+# Authentication
+bcrypt>=4.1.0
+python-jose[cryptography]>=3.3.0
+email-validator>=2.0.0
+
+# Timezone support
+pytz>=2024.1
+
+# Document Processing Nodes
+beautifulsoup4>=4.12.0
+langchain-text-splitters>=0.3.0
+langchain-huggingface>=0.1.0
+chromadb>=0.5.0
+qdrant-client>=1.12.0
+sentence-transformers>=3.0.0
+pypdf>=4.0.0
+# marker-pdf>=1.0.0  # Optional: GPU OCR (requires CUDA)
+# unstructured>=0.16.0  # Optional: Multi-format document parsing
+

package/server/routers/__init__.py

@@ -0,0 +1 @@
+# FastAPI route modules