kubiya-control-plane-api 0.9.15__py3-none-any.whl

control_plane_api/app/models/workflow.py

@@ -0,0 +1,161 @@
+import enum
+
+from sqlalchemy import (
+    Column,
+    String,
+    DateTime,
+    Text,
+    Boolean,
+    Integer,
+    ForeignKey,
+    CheckConstraint,
+    Index,
+    UniqueConstraint,
+    text,
+)
+from sqlalchemy.dialects.postgresql import UUID, JSONB, ARRAY
+from sqlalchemy.orm import relationship
+
+from control_plane_api.app.database import Base
+
+
+class WorkflowStatus(str, enum.Enum):
+    """Workflow status enumeration"""
+
+    PENDING = "pending"
+    RUNNING = "running"
+    PAUSED = "paused"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    CANCELLED = "cancelled"
+
+
+class Workflow(Base):
+    """Workflow model for storing workflow information"""
+
+    __tablename__ = "workflows"
+
+    # Primary columns
+    id = Column(
+        UUID(as_uuid=True),
+        primary_key=True,
+        server_default=text("gen_random_uuid()"),
+        nullable=False,
+    )
+    workspace_id = Column(
+        UUID(as_uuid=True),
+        ForeignKey("workspaces.id", ondelete="CASCADE"),
+        nullable=False,
+    )
+    name = Column(String(255), nullable=False)
+    description = Column(Text, nullable=True)
+    slug = Column(String(255), nullable=True)
+
+    # Workflow definition
+    definition = Column(JSONB, nullable=False)
+    definition_format = Column(
+        String(10),
+        nullable=True,
+        server_default=text("'json'::character varying"),
+    )
+
+    # Status and trigger configuration
+    status = Column(
+        String(50),
+        nullable=True,
+        server_default=text("'draft'::character varying"),
+    )
+    trigger_type = Column(String(50), nullable=True)
+    trigger_config = Column(JSONB, nullable=True)
+
+    # Upstash integration
+    upstash_schedule_id = Column(String(255), nullable=True)
+    upstash_webhook_id = Column(String(255), nullable=True)
+
+    # User tracking
+    created_by = Column(
+        UUID(as_uuid=True),
+        ForeignKey("user_profiles.id", ondelete="SET NULL"),
+        nullable=True,
+    )
+    updated_by = Column(
+        UUID(as_uuid=True),
+        ForeignKey("user_profiles.id", ondelete="SET NULL"),
+        nullable=True,
+    )
+
+    # Timestamps
+    published_at = Column(DateTime(timezone=True), nullable=True)
+    created_at = Column(
+        DateTime(timezone=True),
+        nullable=True,
+        server_default=text("now()"),
+    )
+    updated_at = Column(
+        DateTime(timezone=True),
+        nullable=True,
+        server_default=text("now()"),
+    )
+
+    # Test workflow configuration
+    is_test = Column(Boolean, nullable=True, server_default=text("false"))
+    expires_at = Column(DateTime(timezone=True), nullable=True)
+
+    # Owner information
+    owner_id = Column(Text, nullable=True)
+    owner_email = Column(Text, nullable=True)
+
+    # Versioning
+    version = Column(Integer, nullable=True, server_default=text("1"))
+
+    # Tags and metadata
+    tags = Column(ARRAY(Text), nullable=True, server_default=text("'{}'::text[]"))
+    metadata_ = Column("metadata", JSONB, nullable=True, server_default=text("'{}'::jsonb"))
+
+    # Soft delete
+    deleted_at = Column(DateTime(timezone=True), nullable=True)
+
+    # Constraints
+    __table_args__ = (
+        UniqueConstraint("workspace_id", "slug", name="unique_workflow_slug_per_workspace"),
+        UniqueConstraint("slug", name="workflows_slug_key"),
+        CheckConstraint(
+            "status::text = ANY (ARRAY['draft'::character varying::text, 'published'::character varying::text, 'archived'::character varying::text])",
+            name="workflows_status_check",
+        ),
+        CheckConstraint(
+            "trigger_type::text = ANY (ARRAY['manual'::character varying::text, 'webhook'::character varying::text, 'schedule'::character varying::text])",
+            name="workflows_trigger_type_check",
+        ),
+        CheckConstraint(
+            "definition_format::text = ANY (ARRAY['json'::character varying::text, 'yaml'::character varying::text])",
+            name="workflows_definition_format_check",
+        ),
+        Index("idx_workflows_slug", "slug"),
+        Index("idx_workflows_status", "status"),
+        Index("idx_workflows_workspace_id", "workspace_id"),
+        Index(
+            "idx_workflows_is_test_expires",
+            "is_test",
+            "expires_at",
+            postgresql_where=text("is_test = true AND expires_at IS NOT NULL"),
+        ),
+    )
+
+    # Relationships
+    workspace = relationship("Workspace", back_populates="workflows", viewonly=True)
+    creator = relationship(
+        "UserProfile",
+        foreign_keys=[created_by],
+        back_populates="created_workflows",
+        viewonly=True,
+    )
+    updater = relationship(
+        "UserProfile",
+        foreign_keys=[updated_by],
+        back_populates="updated_workflows",
+        viewonly=True,
+    )
+
+    def __repr__(self):
+        return f"<Workflow(id={self.id}, name={self.name}, status={self.status})>"

control_plane_api/app/models/workspace.py

@@ -0,0 +1,50 @@
+"""Workspace model"""
+from sqlalchemy import Column, String, DateTime, Text, Integer, CheckConstraint, text
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from sqlalchemy.orm import relationship
+
+from control_plane_api.app.database import Base
+
+
+class Workspace(Base):
+    """Model for workspaces"""
+
+    __tablename__ = "workspaces"
+
+    # Primary columns
+    id = Column(UUID(as_uuid=True), primary_key=True, server_default=text("gen_random_uuid()"), nullable=False)
+    name = Column(String(255), nullable=False)
+    description = Column(Text, nullable=True)
+    slug = Column(Text, nullable=True)
+    logo_url = Column(Text, nullable=True)
+
+    # User tracking
+    created_by = Column(UUID(as_uuid=True), nullable=True)
+    owner_email = Column(Text, nullable=True)
+
+    # Workspace type and settings
+    type = Column(Text, nullable=False, server_default=text("'hobby'::text"))
+    settings = Column(JSONB, nullable=True, server_default=text("'{}'::jsonb"))
+    member_count = Column(Integer, nullable=True, server_default=text("1"))
+
+    # Stripe integration
+    stripe_customer_id = Column(Text, nullable=True)
+    stripe_subscription_id = Column(Text, nullable=True)
+
+    # Timestamps
+    created_at = Column(DateTime(timezone=True), nullable=True, server_default=text("now()"))
+    updated_at = Column(DateTime(timezone=True), nullable=True, server_default=text("now()"))
+
+    # Constraints
+    __table_args__ = (
+        CheckConstraint(
+            "type = ANY (ARRAY['hobby'::text, 'enterprise'::text])",
+            name="workspaces_type_check"
+        ),
+    )
+
+    # Relationships
+    workflows = relationship("Workflow", back_populates="workspace")
+
+    def __repr__(self):
+        return f"<Workspace(id={self.id}, name={self.name}, type={self.type})>"

control_plane_api/app/observability/__init__.py

@@ -0,0 +1,177 @@
+"""
+OpenTelemetry observability module for distributed tracing and Prometheus metrics.
+
+This module provides:
+- Tracer initialization and configuration
+- OTLP exporter setup
+- Custom span creation utilities
+- Trace context propagation helpers
+- Automatic endpoint instrumentation decorators
+- Prometheus metrics collection and reporting
+
+Falls back gracefully when OpenTelemetry is not available.
+"""
+
+from control_plane_api.app.observability.optional import HAS_OPENTELEMETRY
+
+# Prometheus metrics - always available (required dependency)
+from control_plane_api.app.observability.metrics import (
+    # Core
+    record_http_request,
+    normalize_endpoint,
+    get_metrics_response,
+    # Original metrics
+    update_active_tasks,
+    update_task_failures,
+    # HIGH priority
+    update_execution_duration,
+    update_worker_queue_depth,
+    update_llm_requests,
+    update_llm_latency,
+    update_llm_tokens,
+    # MEDIUM priority
+    update_streaming_connections,
+    update_tool_execution_duration,
+    update_tool_executions,
+    update_execution_wait_time,
+    update_webhook_requests,
+    # LOWER priority
+    update_executions_by_org,
+    update_agents_active,
+    update_scheduled_jobs,
+    update_conversation_turns,
+)
+
+if HAS_OPENTELEMETRY:
+    try:
+        from control_plane_api.app.observability.tracing import (
+            setup_telemetry,
+            get_tracer,
+            create_span_with_context,
+            get_current_trace_id,
+            get_current_span_id,
+            add_span_event,
+            add_span_error,
+            shutdown_telemetry,
+        )
+        from control_plane_api.app.observability.decorators import (
+            instrument_endpoint,
+        )
+        from control_plane_api.app.observability.context_logging import (
+            trace_operation,
+            log_db_query,
+            log_llm_call,
+            log_worker_operation,
+            log_workflow_step,
+            log_external_api_call,
+            log_business_event,
+            log_state_change,
+            log_cache_operation,
+            log_validation_result,
+        )
+    except ImportError:
+        HAS_OPENTELEMETRY = False
+
+if not HAS_OPENTELEMETRY:
+    # No-op implementations
+    async def setup_telemetry() -> None:
+        """No-op setup when OpenTelemetry is not available."""
+        pass
+
+    async def shutdown_telemetry() -> None:
+        """No-op shutdown when OpenTelemetry is not available."""
+        pass
+
+    def get_tracer(name: str, version: str = ""):
+        """No-op tracer."""
+        from control_plane_api.app.observability.optional import get_tracer as _get_tracer
+        return _get_tracer(name, version)
+
+    def create_span_with_context(*args, **kwargs):
+        """No-op span creator."""
+        from contextlib import nullcontext
+        return nullcontext()
+
+    def get_current_trace_id() -> str:
+        """No-op trace ID getter."""
+        return ""
+
+    def get_current_span_id() -> str:
+        """No-op span ID getter."""
+        return ""
+
+    def add_span_event(*args, **kwargs):
+        """No-op event adder."""
+        pass
+
+    def add_span_error(*args, **kwargs):
+        """No-op error adder."""
+        pass
+
+    def instrument_endpoint(func):
+        """No-op decorator."""
+        return func
+
+    def trace_operation(*args, **kwargs):
+        """No-op operation tracer."""
+        from contextlib import nullcontext
+        return nullcontext()
+
+    # All logging helpers are no-ops
+    log_db_query = trace_operation
+    log_llm_call = trace_operation
+    log_worker_operation = trace_operation
+    log_workflow_step = trace_operation
+    log_external_api_call = trace_operation
+    log_business_event = trace_operation
+    log_state_change = trace_operation
+    log_cache_operation = trace_operation
+    log_validation_result = trace_operation
+
+__all__ = [
+    # Core tracing
+    "setup_telemetry",
+    "get_tracer",
+    "create_span_with_context",
+    "get_current_trace_id",
+    "get_current_span_id",
+    "add_span_event",
+    "add_span_error",
+    "shutdown_telemetry",
+    "instrument_endpoint",
+    # Context logging helpers
+    "trace_operation",
+    "log_db_query",
+    "log_llm_call",
+    "log_worker_operation",
+    "log_workflow_step",
+    "log_external_api_call",
+    "log_business_event",
+    "log_state_change",
+    "log_cache_operation",
+    "log_validation_result",
+    # Prometheus metrics - core
+    "record_http_request",
+    "normalize_endpoint",
+    "get_metrics_response",
+    # Prometheus metrics - original
+    "update_active_tasks",
+    "update_task_failures",
+    # Prometheus metrics - HIGH priority
+    "update_execution_duration",
+    "update_worker_queue_depth",
+    "update_llm_requests",
+    "update_llm_latency",
+    "update_llm_tokens",
+    # Prometheus metrics - MEDIUM priority
+    "update_streaming_connections",
+    "update_tool_execution_duration",
+    "update_tool_executions",
+    "update_execution_wait_time",
+    "update_webhook_requests",
+    # Prometheus metrics - LOWER priority
+    "update_executions_by_org",
+    "update_agents_active",
+    "update_scheduled_jobs",
+    "update_conversation_turns",
+]