kubiya-control-plane-api 0.1.0__py3-none-any.whl → 0.3.4__py3-none-any.whl

control_plane_api/README.md

@@ -0,0 +1,266 @@
+# Kubiya Control Plane API
+
+A multi-tenant AI agent orchestration and management platform built with FastAPI, Temporal, and PostgreSQL.
+
+## Features
+
+- **Multi-tenant Architecture**: Manage multiple projects, teams, and agents
+- **Workflow Orchestration**: Temporal-based workflow execution
+- **Flexible Agent Runtime**: Support for multiple agent types and skills
+- **Policy Enforcement**: OPA-based policy engine for agent governance
+- **Scalable Workers**: Distributed worker architecture for agent execution
+- **Context Management**: Environment and team-specific context handling
+- **LLM Integration**: Support for multiple LLM providers via LiteLLM
+- **Comprehensive APIs**: RESTful APIs for all platform features
+
+## Installation
+
+### Using pip
+
+```bash
+pip install kubiya-control_plane_api
+```
+
+### From source
+
+```bash
+git clone https://github.com/kubiyabot/agent-control-plane.git
+cd agent-control-plane/control_plane_api
+pip install -e .
+```
+
+### With development dependencies
+
+```bash
+pip install kubiya-control_plane_api[dev]
+```
+
+## Quick Start
+
+### 1. Set up environment variables
+
+Create a `.env` file or set the following environment variables:
+
+```bash
+# Database Configuration
+DATABASE_URL=postgresql://user:password@localhost:5432/control_plane
+
+# Supabase (for serverless deployments)
+SUPABASE_URL=your-supabase-url
+SUPABASE_KEY=your-supabase-key
+
+# Temporal Configuration
+TEMPORAL_HOST=localhost:7233
+TEMPORAL_NAMESPACE=default
+
+# API Configuration
+API_TITLE="Agent Control Plane"
+API_VERSION="1.0.0"
+ENVIRONMENT=development
+LOG_LEVEL=info
+
+# Security
+SECRET_KEY=your-secret-key-here
+
+# Optional: Kubiya Integration
+KUBIYA_API_KEY=your-kubiya-api-key
+KUBIYA_API_URL=https://api.kubiya.ai
+```
+
+### 2. Run database migrations
+
+```bash
+alembic upgrade head
+```
+
+### 3. Start the API server
+
+```bash
+uvicorn app.main:app --host 0.0.0.0 --port 7777 --reload
+```
+
+The API will be available at `http://localhost:7777`
+
+### 4. Access the API documentation
+
+Open your browser and navigate to:
+- Swagger UI: `http://localhost:7777/api/docs`
+- ReDoc: `http://localhost:7777/api/redoc`
+
+### 5. Start a worker (optional)
+
+To process agent execution workflows:
+
+```bash
+python worker.py
+```
+
+## Package Structure
+
+```
+app/
+├── activities/          # Temporal activities
+├── lib/                # Client libraries (Kubiya, Redis, Temporal, etc.)
+├── middleware/         # FastAPI middleware (auth, etc.)
+├── models/             # SQLAlchemy models
+├── policies/           # OPA policy files (.rego)
+├── routers/            # FastAPI route handlers
+├── services/           # Business logic services
+├── skills/           # Agent skills
+├── workflows/          # Temporal workflows
+├── config.py           # Configuration management
+├── database.py         # Database connection
+└── main.py            # FastAPI application entry point
+```
+
+## API Endpoints
+
+### Core Resources
+
+- **Projects**: `/api/v1/projects` - Multi-project management
+- **Environments**: `/api/v1/environments` - Environment configuration
+- **Agents**: `/api/v1/agents` - Agent management
+- **Teams**: `/api/v1/teams` - Team management
+- **Workflows**: `/api/v1/workflows` - Workflow definitions
+- **Executions**: `/api/v1/executions` - Execution tracking
+- **Workers**: `/api/v1/workers` - Worker registration and management
+
+### Skills and Policies
+
+- **Skills**: `/api/v1/skills` - Tool sets and definitions
+- **Policies**: `/api/v1/policies` - Policy management and enforcement
+
+### Integration
+
+- **Secrets**: `/api/v1/secrets` - Secrets management (proxies to Kubiya)
+- **Integrations**: `/api/v1/integrations` - Third-party integrations
+- **Models**: `/api/v1/models` - LLM model configuration
+
+### Utilities
+
+- **Health**: `/api/health` - Health check endpoint
+- **Task Planning**: `/api/v1/task-planning` - AI-powered task planning
+
+## Configuration
+
+The application uses Pydantic Settings for configuration management. All settings can be configured via environment variables or a `.env` file.
+
+### Key Configuration Options
+
+- `DATABASE_URL`: PostgreSQL connection string
+- `SUPABASE_URL`, `SUPABASE_KEY`: Supabase configuration for serverless
+- `TEMPORAL_HOST`: Temporal server address
+- `KUBIYA_API_KEY`: Kubiya platform API key
+- `SECRET_KEY`: Secret key for JWT token signing
+- `LOG_LEVEL`: Logging level (debug, info, warning, error)
+- `ENVIRONMENT`: Deployment environment (development, staging, production)
+
+## Development
+
+### Install development dependencies
+
+```bash
+pip install -e ".[dev]"
+```
+
+### Run tests
+
+```bash
+# All tests
+pytest
+
+# Unit tests only
+pytest -m unit
+
+# Integration tests
+pytest -m integration
+
+# With coverage
+pytest --cov=app --cov-report=html
+```
+
+### Code formatting
+
+```bash
+# Format code with black
+black .
+
+# Lint with ruff
+ruff check .
+```
+
+### Database migrations
+
+```bash
+# Create a new migration
+alembic revision --autogenerate -m "Description of changes"
+
+# Apply migrations
+alembic upgrade head
+
+# Rollback migration
+alembic downgrade -1
+```
+
+## Deployment
+
+### Docker
+
+```bash
+docker build -t control_plane_api .
+docker run -p 7777:7777 --env-file .env control_plane_api
+```
+
+### Vercel (Serverless)
+
+The API is configured for Vercel deployment with `vercel.json` and the Mangum adapter.
+
+```bash
+vercel deploy
+```
+
+## Architecture
+
+### Workflow Orchestration
+
+The platform uses Temporal for reliable workflow execution:
+- Durable execution with automatic retries
+- Activity-based task decomposition
+- Support for long-running workflows
+- Built-in observability and monitoring
+
+### Multi-tenancy
+
+- **Projects**: Top-level isolation boundary
+- **Environments**: Isolated execution contexts within projects
+- **Teams**: Collaborative agent groups
+- **Agents**: Individual agent instances
+
+### Worker Architecture
+
+Workers pull tasks from environment-specific queues and execute agent workflows using Temporal.
+
+## Contributing
+
+1. Fork the repository
+2. Create a feature branch
+3. Make your changes
+4. Run tests and linting
+5. Submit a pull request
+
+## License
+
+MIT License - see LICENSE file for details
+
+## Support
+
+For issues and questions:
+- GitHub Issues: https://github.com/kubiyabot/agent-control-plane/issues
+- Email: support@kubiya.ai
+
+## Links
+
+- [Documentation](https://github.com/kubiyabot/agent-control-plane/blob/main/README.md)
+- [GitHub Repository](https://github.com/kubiyabot/agent-control-plane)
+- [Kubiya Platform](https://kubiya.ai)
+

control_plane_api/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.3.4"

control_plane_api/alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration.

control_plane_api/alembic/env.py

@@ -0,0 +1,98 @@
+from logging.config import fileConfig
+
+from sqlalchemy import engine_from_config
+from sqlalchemy import pool
+
+from alembic import context
+
+# Import application settings and models
+import sys
+from pathlib import Path
+
+# Add parent directory to path so control_plane_api is importable
+# env.py location: /app/control_plane_api/alembic/env.py
+# We need: /app (which is parent.parent.parent)
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent.parent))
+
+from control_plane_api.app.config import settings
+from control_plane_api.app.database import Base
+from control_plane_api.app.models import (
+    Project, Agent, Team, Workflow, Session,
+    Execution, UserPresence, Environment, AgentEnvironment, TeamEnvironment
+)
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Override sqlalchemy.url with our DATABASE_URL from settings
+if settings.database_url:
+    # Escape % characters for ConfigParser (% -> %%)
+    escaped_url = settings.database_url.replace("%", "%%")
+    config.set_main_option("sqlalchemy.url", escaped_url)
+
+# Interpret the config file for Python logging.
+# This line sets up loggers basically.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+target_metadata = Base.metadata
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+
+    Calls to context.execute() here emit the given string to the
+    script output.
+
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode.
+
+    In this scenario we need to create an Engine
+    and associate a connection with the context.
+
+    """
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        context.configure(
+            connection=connection, target_metadata=target_metadata
+        )
+
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

control_plane_api/alembic/script.py.mako

@@ -0,0 +1,28 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, Sequence[str], None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    ${downgrades if downgrades else "pass"}

control_plane_api/alembic/versions/1382bec74309_initial_migration_with_all_models.py

@@ -0,0 +1,251 @@
+"""Initial migration with all models
+
+Revision ID: 1382bec74309
+Revises: 
+Create Date: 2025-10-28 17:21:33.097680
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '1382bec74309'
+down_revision: Union[str, Sequence[str], None] = None
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('environments',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('name', sa.String(), nullable=False),
+    sa.Column('display_name', sa.String(), nullable=True),
+    sa.Column('description', sa.String(), nullable=True),
+    sa.Column('tags', sa.JSON(), nullable=False),
+    sa.Column('settings', sa.JSON(), nullable=False),
+    sa.Column('status', sa.Enum('PENDING', 'PROVISIONING', 'ACTIVE', 'INACTIVE', 'ERROR', name='environmentstatus'), nullable=False),
+    sa.Column('worker_token', sa.String(), nullable=True),
+    sa.Column('provisioning_workflow_id', sa.String(), nullable=True),
+    sa.Column('provisioned_at', sa.DateTime(), nullable=True),
+    sa.Column('error_message', sa.String(), nullable=True),
+    sa.Column('temporal_namespace_id', sa.String(), nullable=True),
+    sa.Column('execution_environment', sa.JSON(), nullable=False),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.Column('created_by', sa.String(), nullable=True),
+    sa.Column('updated_by', sa.String(), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_environments_name'), 'environments', ['name'], unique=False)
+    op.create_index(op.f('ix_environments_organization_id'), 'environments', ['organization_id'], unique=False)
+    op.create_index(op.f('ix_environments_status'), 'environments', ['status'], unique=False)
+    op.create_table('executions',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('execution_type', sa.Enum('agent', 'team', 'workflow', name='executiontype'), nullable=False),
+    sa.Column('entity_id', sa.String(), nullable=False),
+    sa.Column('entity_name', sa.String(), nullable=True),
+    sa.Column('runner_name', sa.String(), nullable=True),
+    sa.Column('user_id', sa.String(), nullable=True),
+    sa.Column('user_email', sa.String(), nullable=True),
+    sa.Column('user_name', sa.String(), nullable=True),
+    sa.Column('user_avatar', sa.String(), nullable=True),
+    sa.Column('prompt', sa.Text(), nullable=False),
+    sa.Column('system_prompt', sa.Text(), nullable=True),
+    sa.Column('config', sa.JSON(), nullable=True),
+    sa.Column('status', sa.Enum('pending', 'running', 'waiting_for_input', 'completed', 'failed', 'cancelled', name='executionstatus'), nullable=False),
+    sa.Column('response', sa.Text(), nullable=True),
+    sa.Column('error_message', sa.Text(), nullable=True),
+    sa.Column('usage', sa.JSON(), nullable=True),
+    sa.Column('execution_metadata', sa.JSON(), nullable=True),
+    sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('started_at', sa.DateTime(timezone=True), nullable=True),
+    sa.Column('completed_at', sa.DateTime(timezone=True), nullable=True),
+    sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_executions_organization_id'), 'executions', ['organization_id'], unique=False)
+    op.create_index(op.f('ix_executions_user_id'), 'executions', ['user_id'], unique=False)
+    op.create_table('projects',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('name', sa.String(), nullable=False),
+    sa.Column('description', sa.Text(), nullable=True),
+    sa.Column('goals', sa.Text(), nullable=True),
+    sa.Column('status', sa.Enum('active', 'archived', 'paused', name='projectstatus'), nullable=False),
+    sa.Column('restrict_to_environment', sa.Boolean(), nullable=False),
+    sa.Column('default_model', sa.String(), nullable=True),
+    sa.Column('default_settings', sa.JSON(), nullable=False),
+    sa.Column('is_default', sa.Boolean(), nullable=False),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_projects_name'), 'projects', ['name'], unique=False)
+    op.create_index(op.f('ix_projects_organization_id'), 'projects', ['organization_id'], unique=False)
+    op.create_table('teams',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('name', sa.String(), nullable=False),
+    sa.Column('description', sa.Text(), nullable=True),
+    sa.Column('status', sa.Enum('active', 'inactive', 'archived', name='teamstatus'), nullable=False),
+    sa.Column('configuration', sa.JSON(), nullable=False),
+    sa.Column('skill_ids', sa.ARRAY(sa.UUID(as_uuid=False)), nullable=False),
+    sa.Column('execution_environment', sa.JSON(), nullable=False),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('organization_id', 'name', name='uq_team_org_name')
+    )
+    op.create_index(op.f('ix_teams_name'), 'teams', ['name'], unique=False)
+    op.create_index(op.f('ix_teams_organization_id'), 'teams', ['organization_id'], unique=False)
+    op.create_table('user_presence',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('user_id', sa.String(), nullable=False),
+    sa.Column('user_email', sa.String(), nullable=True),
+    sa.Column('user_name', sa.String(), nullable=True),
+    sa.Column('user_avatar', sa.String(), nullable=True),
+    sa.Column('agent_id', sa.String(), nullable=True),
+    sa.Column('session_id', sa.String(), nullable=True),
+    sa.Column('execution_id', sa.String(), nullable=True),
+    sa.Column('is_active', sa.Boolean(), nullable=False),
+    sa.Column('is_typing', sa.Boolean(), nullable=False),
+    sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.Column('last_active_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index('idx_presence_lookup', 'user_presence', ['agent_id', 'is_active', 'last_active_at'], unique=False)
+    op.create_index('idx_presence_user', 'user_presence', ['user_id', 'is_active'], unique=False)
+    op.create_index(op.f('ix_user_presence_agent_id'), 'user_presence', ['agent_id'], unique=False)
+    op.create_index(op.f('ix_user_presence_execution_id'), 'user_presence', ['execution_id'], unique=False)
+    op.create_index(op.f('ix_user_presence_session_id'), 'user_presence', ['session_id'], unique=False)
+    op.create_index(op.f('ix_user_presence_user_id'), 'user_presence', ['user_id'], unique=False)
+    op.create_table('agents',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('name', sa.String(), nullable=False),
+    sa.Column('description', sa.Text(), nullable=True),
+    sa.Column('status', sa.Enum('idle', 'running', 'paused', 'completed', 'failed', 'stopped', name='agentstatus'), nullable=False),
+    sa.Column('capabilities', sa.JSON(), nullable=False),
+    sa.Column('configuration', sa.JSON(), nullable=False),
+    sa.Column('model_id', sa.String(), nullable=True),
+    sa.Column('model_config', sa.JSON(), nullable=False),
+    sa.Column('runtime', sa.Enum('default', 'claude_code', name='runtimetype'), server_default='default', nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('team_id', sa.String(), nullable=True),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.Column('last_active_at', sa.DateTime(), nullable=True),
+    sa.Column('state', sa.JSON(), nullable=False),
+    sa.Column('error_message', sa.Text(), nullable=True),
+    sa.ForeignKeyConstraint(['team_id'], ['teams.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_agents_name'), 'agents', ['name'], unique=False)
+    op.create_index(op.f('ix_agents_organization_id'), 'agents', ['organization_id'], unique=False)
+    op.create_index(op.f('ix_agents_runtime'), 'agents', ['runtime'], unique=False)
+    op.create_table('team_environments',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('team_id', sa.String(), nullable=False),
+    sa.Column('environment_id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('assigned_at', sa.DateTime(), nullable=False),
+    sa.Column('assigned_by', sa.String(), nullable=True),
+    sa.ForeignKeyConstraint(['environment_id'], ['environments.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['team_id'], ['teams.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_team_environments_organization_id'), 'team_environments', ['organization_id'], unique=False)
+    op.create_table('workflows',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('name', sa.String(), nullable=False),
+    sa.Column('description', sa.Text(), nullable=True),
+    sa.Column('status', sa.Enum('pending', 'running', 'paused', 'completed', 'failed', 'cancelled', name='workflowstatus'), nullable=False),
+    sa.Column('steps', sa.JSON(), nullable=False),
+    sa.Column('current_step', sa.String(), nullable=True),
+    sa.Column('configuration', sa.JSON(), nullable=False),
+    sa.Column('team_id', sa.String(), nullable=True),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.Column('started_at', sa.DateTime(), nullable=True),
+    sa.Column('completed_at', sa.DateTime(), nullable=True),
+    sa.Column('state', sa.JSON(), nullable=False),
+    sa.Column('error_message', sa.Text(), nullable=True),
+    sa.ForeignKeyConstraint(['team_id'], ['teams.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_workflows_name'), 'workflows', ['name'], unique=False)
+    op.create_table('agent_environments',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('agent_id', sa.String(), nullable=False),
+    sa.Column('environment_id', sa.String(), nullable=False),
+    sa.Column('organization_id', sa.String(), nullable=False),
+    sa.Column('assigned_at', sa.DateTime(), nullable=False),
+    sa.Column('assigned_by', sa.String(), nullable=True),
+    sa.ForeignKeyConstraint(['agent_id'], ['agents.id'], ondelete='CASCADE'),
+    sa.ForeignKeyConstraint(['environment_id'], ['environments.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_agent_environments_organization_id'), 'agent_environments', ['organization_id'], unique=False)
+    op.create_table('sessions',
+    sa.Column('id', sa.String(), nullable=False),
+    sa.Column('agent_id', sa.String(), nullable=False),
+    sa.Column('user_id', sa.String(), nullable=True),
+    sa.Column('user_email', sa.String(), nullable=True),
+    sa.Column('user_name', sa.String(), nullable=True),
+    sa.Column('user_avatar', sa.String(), nullable=True),
+    sa.Column('messages', sa.JSON(), nullable=False),
+    sa.Column('context', sa.JSON(), nullable=False),
+    sa.Column('session_metadata', sa.JSON(), nullable=False),
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=False),
+    sa.Column('expires_at', sa.DateTime(), nullable=True),
+    sa.Column('last_active_at', sa.DateTime(), nullable=False),
+    sa.ForeignKeyConstraint(['agent_id'], ['agents.id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_sessions_user_id'), 'sessions', ['user_id'], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f('ix_sessions_user_id'), table_name='sessions')
+    op.drop_table('sessions')
+    op.drop_index(op.f('ix_agent_environments_organization_id'), table_name='agent_environments')
+    op.drop_table('agent_environments')
+    op.drop_index(op.f('ix_workflows_name'), table_name='workflows')
+    op.drop_table('workflows')
+    op.drop_index(op.f('ix_team_environments_organization_id'), table_name='team_environments')
+    op.drop_table('team_environments')
+    op.drop_index(op.f('ix_agents_runtime'), table_name='agents')
+    op.drop_index(op.f('ix_agents_organization_id'), table_name='agents')
+    op.drop_index(op.f('ix_agents_name'), table_name='agents')
+    op.drop_table('agents')
+    op.drop_index(op.f('ix_user_presence_user_id'), table_name='user_presence')
+    op.drop_index(op.f('ix_user_presence_session_id'), table_name='user_presence')
+    op.drop_index(op.f('ix_user_presence_execution_id'), table_name='user_presence')
+    op.drop_index(op.f('ix_user_presence_agent_id'), table_name='user_presence')
+    op.drop_index('idx_presence_user', table_name='user_presence')
+    op.drop_index('idx_presence_lookup', table_name='user_presence')
+    op.drop_table('user_presence')
+    op.drop_index(op.f('ix_teams_organization_id'), table_name='teams')
+    op.drop_index(op.f('ix_teams_name'), table_name='teams')
+    op.drop_table('teams')
+    op.drop_index(op.f('ix_projects_organization_id'), table_name='projects')
+    op.drop_index(op.f('ix_projects_name'), table_name='projects')
+    op.drop_table('projects')
+    op.drop_index(op.f('ix_executions_user_id'), table_name='executions')
+    op.drop_index(op.f('ix_executions_organization_id'), table_name='executions')
+    op.drop_table('executions')
+    op.drop_index(op.f('ix_environments_status'), table_name='environments')
+    op.drop_index(op.f('ix_environments_organization_id'), table_name='environments')
+    op.drop_index(op.f('ix_environments_name'), table_name='environments')
+    op.drop_table('environments')
+    # ### end Alembic commands ###