kubiya-control-plane-api 0.9.15__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- control_plane_api/LICENSE +676 -0
- control_plane_api/README.md +350 -0
- control_plane_api/__init__.py +4 -0
- control_plane_api/__version__.py +8 -0
- control_plane_api/alembic/README +1 -0
- control_plane_api/alembic/env.py +121 -0
- control_plane_api/alembic/script.py.mako +28 -0
- control_plane_api/alembic/versions/2613c65c3dbe_initial_database_setup.py +32 -0
- control_plane_api/alembic/versions/2df520d4927d_merge_heads.py +28 -0
- control_plane_api/alembic/versions/43abf98d6a01_add_paused_status_to_executions.py +73 -0
- control_plane_api/alembic/versions/6289854264cb_merge_multiple_heads.py +28 -0
- control_plane_api/alembic/versions/6a4d4dc3d8dc_generate_execution_transitions.py +50 -0
- control_plane_api/alembic/versions/87d11cf0a783_add_disconnected_status_to_worker_.py +44 -0
- control_plane_api/alembic/versions/add_ephemeral_queue_support.py +85 -0
- control_plane_api/alembic/versions/add_model_type_to_llm_models.py +31 -0
- control_plane_api/alembic/versions/add_plan_executions_table.py +114 -0
- control_plane_api/alembic/versions/add_trace_span_tables.py +154 -0
- control_plane_api/alembic/versions/add_user_info_to_traces.py +36 -0
- control_plane_api/alembic/versions/adjusting_foreign_keys.py +32 -0
- control_plane_api/alembic/versions/b4983d976db2_initial_tables.py +1128 -0
- control_plane_api/alembic/versions/d181a3b40e71_rename_custom_metadata_to_metadata_in_.py +50 -0
- control_plane_api/alembic/versions/df9117888e82_add_missing_columns.py +82 -0
- control_plane_api/alembic/versions/f25de6ad895a_missing_migrations.py +34 -0
- control_plane_api/alembic/versions/f71305fb69b9_fix_ephemeral_queue_deletion_foreign_key.py +54 -0
- control_plane_api/alembic/versions/mark_local_exec_queues_as_ephemeral.py +68 -0
- control_plane_api/alembic.ini +148 -0
- control_plane_api/api/index.py +12 -0
- control_plane_api/app/__init__.py +11 -0
- control_plane_api/app/activities/__init__.py +20 -0
- control_plane_api/app/activities/agent_activities.py +384 -0
- control_plane_api/app/activities/plan_generation_activities.py +499 -0
- control_plane_api/app/activities/team_activities.py +424 -0
- control_plane_api/app/activities/temporal_cloud_activities.py +588 -0
- control_plane_api/app/config/__init__.py +35 -0
- control_plane_api/app/config/api_config.py +469 -0
- control_plane_api/app/config/config_loader.py +224 -0
- control_plane_api/app/config/model_pricing.py +323 -0
- control_plane_api/app/config/storage_config.py +159 -0
- control_plane_api/app/config.py +115 -0
- control_plane_api/app/controllers/__init__.py +0 -0
- control_plane_api/app/controllers/execution_environment_controller.py +1315 -0
- control_plane_api/app/database.py +135 -0
- control_plane_api/app/exceptions.py +408 -0
- control_plane_api/app/lib/__init__.py +11 -0
- control_plane_api/app/lib/environment.py +65 -0
- control_plane_api/app/lib/event_bus/__init__.py +17 -0
- control_plane_api/app/lib/event_bus/base.py +136 -0
- control_plane_api/app/lib/event_bus/manager.py +335 -0
- control_plane_api/app/lib/event_bus/providers/__init__.py +6 -0
- control_plane_api/app/lib/event_bus/providers/http_provider.py +166 -0
- control_plane_api/app/lib/event_bus/providers/nats_provider.py +324 -0
- control_plane_api/app/lib/event_bus/providers/redis_provider.py +233 -0
- control_plane_api/app/lib/event_bus/providers/websocket_provider.py +497 -0
- control_plane_api/app/lib/job_executor.py +330 -0
- control_plane_api/app/lib/kubiya_client.py +293 -0
- control_plane_api/app/lib/litellm_pricing.py +166 -0
- control_plane_api/app/lib/mcp_validation.py +163 -0
- control_plane_api/app/lib/nats/__init__.py +13 -0
- control_plane_api/app/lib/nats/credentials_manager.py +288 -0
- control_plane_api/app/lib/nats/listener.py +374 -0
- control_plane_api/app/lib/planning_prompt_builder.py +153 -0
- control_plane_api/app/lib/planning_tools/__init__.py +41 -0
- control_plane_api/app/lib/planning_tools/agents.py +409 -0
- control_plane_api/app/lib/planning_tools/agno_toolkit.py +836 -0
- control_plane_api/app/lib/planning_tools/base.py +119 -0
- control_plane_api/app/lib/planning_tools/cognitive_memory_tools.py +403 -0
- control_plane_api/app/lib/planning_tools/context_graph_tools.py +545 -0
- control_plane_api/app/lib/planning_tools/environments.py +218 -0
- control_plane_api/app/lib/planning_tools/knowledge.py +204 -0
- control_plane_api/app/lib/planning_tools/models.py +93 -0
- control_plane_api/app/lib/planning_tools/planning_service.py +646 -0
- control_plane_api/app/lib/planning_tools/resources.py +242 -0
- control_plane_api/app/lib/planning_tools/teams.py +334 -0
- control_plane_api/app/lib/policy_enforcer_client.py +1016 -0
- control_plane_api/app/lib/redis_client.py +803 -0
- control_plane_api/app/lib/sqlalchemy_utils.py +486 -0
- control_plane_api/app/lib/state_transition_tools/__init__.py +7 -0
- control_plane_api/app/lib/state_transition_tools/execution_context.py +388 -0
- control_plane_api/app/lib/storage/__init__.py +20 -0
- control_plane_api/app/lib/storage/base_provider.py +274 -0
- control_plane_api/app/lib/storage/provider_factory.py +157 -0
- control_plane_api/app/lib/storage/vercel_blob_provider.py +468 -0
- control_plane_api/app/lib/supabase.py +71 -0
- control_plane_api/app/lib/supabase_utils.py +138 -0
- control_plane_api/app/lib/task_planning/__init__.py +138 -0
- control_plane_api/app/lib/task_planning/agent_factory.py +308 -0
- control_plane_api/app/lib/task_planning/agents.py +389 -0
- control_plane_api/app/lib/task_planning/cache.py +218 -0
- control_plane_api/app/lib/task_planning/entity_resolver.py +273 -0
- control_plane_api/app/lib/task_planning/helpers.py +293 -0
- control_plane_api/app/lib/task_planning/hooks.py +474 -0
- control_plane_api/app/lib/task_planning/models.py +503 -0
- control_plane_api/app/lib/task_planning/plan_validator.py +166 -0
- control_plane_api/app/lib/task_planning/planning_workflow.py +2911 -0
- control_plane_api/app/lib/task_planning/runner.py +656 -0
- control_plane_api/app/lib/task_planning/streaming_hook.py +213 -0
- control_plane_api/app/lib/task_planning/workflow.py +424 -0
- control_plane_api/app/lib/templating/__init__.py +88 -0
- control_plane_api/app/lib/templating/compiler.py +278 -0
- control_plane_api/app/lib/templating/engine.py +178 -0
- control_plane_api/app/lib/templating/parsers/__init__.py +29 -0
- control_plane_api/app/lib/templating/parsers/base.py +96 -0
- control_plane_api/app/lib/templating/parsers/env.py +85 -0
- control_plane_api/app/lib/templating/parsers/graph.py +112 -0
- control_plane_api/app/lib/templating/parsers/secret.py +87 -0
- control_plane_api/app/lib/templating/parsers/simple.py +81 -0
- control_plane_api/app/lib/templating/resolver.py +366 -0
- control_plane_api/app/lib/templating/types.py +214 -0
- control_plane_api/app/lib/templating/validator.py +201 -0
- control_plane_api/app/lib/temporal_client.py +232 -0
- control_plane_api/app/lib/temporal_credentials_cache.py +178 -0
- control_plane_api/app/lib/temporal_credentials_service.py +203 -0
- control_plane_api/app/lib/validation/__init__.py +24 -0
- control_plane_api/app/lib/validation/runtime_validation.py +388 -0
- control_plane_api/app/main.py +531 -0
- control_plane_api/app/middleware/__init__.py +10 -0
- control_plane_api/app/middleware/auth.py +645 -0
- control_plane_api/app/middleware/exception_handler.py +267 -0
- control_plane_api/app/middleware/prometheus_middleware.py +173 -0
- control_plane_api/app/middleware/rate_limiting.py +384 -0
- control_plane_api/app/middleware/request_id.py +202 -0
- control_plane_api/app/models/__init__.py +40 -0
- control_plane_api/app/models/agent.py +90 -0
- control_plane_api/app/models/analytics.py +206 -0
- control_plane_api/app/models/associations.py +107 -0
- control_plane_api/app/models/auth_user.py +73 -0
- control_plane_api/app/models/context.py +161 -0
- control_plane_api/app/models/custom_integration.py +99 -0
- control_plane_api/app/models/environment.py +64 -0
- control_plane_api/app/models/execution.py +125 -0
- control_plane_api/app/models/execution_transition.py +50 -0
- control_plane_api/app/models/job.py +159 -0
- control_plane_api/app/models/llm_model.py +78 -0
- control_plane_api/app/models/orchestration.py +66 -0
- control_plane_api/app/models/plan_execution.py +102 -0
- control_plane_api/app/models/presence.py +49 -0
- control_plane_api/app/models/project.py +61 -0
- control_plane_api/app/models/project_management.py +85 -0
- control_plane_api/app/models/session.py +29 -0
- control_plane_api/app/models/skill.py +155 -0
- control_plane_api/app/models/system_tables.py +43 -0
- control_plane_api/app/models/task_planning.py +372 -0
- control_plane_api/app/models/team.py +86 -0
- control_plane_api/app/models/trace.py +257 -0
- control_plane_api/app/models/user_profile.py +54 -0
- control_plane_api/app/models/worker.py +221 -0
- control_plane_api/app/models/workflow.py +161 -0
- control_plane_api/app/models/workspace.py +50 -0
- control_plane_api/app/observability/__init__.py +177 -0
- control_plane_api/app/observability/context_logging.py +475 -0
- control_plane_api/app/observability/decorators.py +337 -0
- control_plane_api/app/observability/local_span_processor.py +702 -0
- control_plane_api/app/observability/metrics.py +303 -0
- control_plane_api/app/observability/middleware.py +246 -0
- control_plane_api/app/observability/optional.py +115 -0
- control_plane_api/app/observability/tracing.py +382 -0
- control_plane_api/app/policies/README.md +149 -0
- control_plane_api/app/policies/approved_users.rego +62 -0
- control_plane_api/app/policies/business_hours.rego +51 -0
- control_plane_api/app/policies/rate_limiting.rego +100 -0
- control_plane_api/app/policies/tool_enforcement/README.md +336 -0
- control_plane_api/app/policies/tool_enforcement/bash_command_validation.rego +71 -0
- control_plane_api/app/policies/tool_enforcement/business_hours_enforcement.rego +82 -0
- control_plane_api/app/policies/tool_enforcement/mcp_tool_allowlist.rego +58 -0
- control_plane_api/app/policies/tool_enforcement/production_safeguards.rego +80 -0
- control_plane_api/app/policies/tool_enforcement/role_based_tool_access.rego +44 -0
- control_plane_api/app/policies/tool_restrictions.rego +86 -0
- control_plane_api/app/routers/__init__.py +4 -0
- control_plane_api/app/routers/agents.py +382 -0
- control_plane_api/app/routers/agents_v2.py +1598 -0
- control_plane_api/app/routers/analytics.py +1310 -0
- control_plane_api/app/routers/auth.py +59 -0
- control_plane_api/app/routers/client_config.py +57 -0
- control_plane_api/app/routers/context_graph.py +561 -0
- control_plane_api/app/routers/context_manager.py +577 -0
- control_plane_api/app/routers/custom_integrations.py +490 -0
- control_plane_api/app/routers/enforcer.py +132 -0
- control_plane_api/app/routers/environment_context.py +252 -0
- control_plane_api/app/routers/environments.py +761 -0
- control_plane_api/app/routers/execution_environment.py +847 -0
- control_plane_api/app/routers/executions/__init__.py +28 -0
- control_plane_api/app/routers/executions/router.py +286 -0
- control_plane_api/app/routers/executions/services/__init__.py +22 -0
- control_plane_api/app/routers/executions/services/demo_worker_health.py +156 -0
- control_plane_api/app/routers/executions/services/status_service.py +420 -0
- control_plane_api/app/routers/executions/services/test_worker_health.py +480 -0
- control_plane_api/app/routers/executions/services/worker_health.py +514 -0
- control_plane_api/app/routers/executions/streaming/__init__.py +22 -0
- control_plane_api/app/routers/executions/streaming/deduplication.py +352 -0
- control_plane_api/app/routers/executions/streaming/event_buffer.py +353 -0
- control_plane_api/app/routers/executions/streaming/event_formatter.py +964 -0
- control_plane_api/app/routers/executions/streaming/history_loader.py +588 -0
- control_plane_api/app/routers/executions/streaming/live_source.py +693 -0
- control_plane_api/app/routers/executions/streaming/streamer.py +849 -0
- control_plane_api/app/routers/executions.py +4888 -0
- control_plane_api/app/routers/health.py +165 -0
- control_plane_api/app/routers/health_v2.py +394 -0
- control_plane_api/app/routers/integration_templates.py +496 -0
- control_plane_api/app/routers/integrations.py +287 -0
- control_plane_api/app/routers/jobs.py +1809 -0
- control_plane_api/app/routers/metrics.py +517 -0
- control_plane_api/app/routers/models.py +82 -0
- control_plane_api/app/routers/models_v2.py +628 -0
- control_plane_api/app/routers/plan_executions.py +1481 -0
- control_plane_api/app/routers/plan_generation_async.py +304 -0
- control_plane_api/app/routers/policies.py +669 -0
- control_plane_api/app/routers/presence.py +234 -0
- control_plane_api/app/routers/projects.py +987 -0
- control_plane_api/app/routers/runners.py +379 -0
- control_plane_api/app/routers/runtimes.py +172 -0
- control_plane_api/app/routers/secrets.py +171 -0
- control_plane_api/app/routers/skills.py +1010 -0
- control_plane_api/app/routers/skills_definitions.py +140 -0
- control_plane_api/app/routers/storage.py +456 -0
- control_plane_api/app/routers/task_planning.py +611 -0
- control_plane_api/app/routers/task_queues.py +650 -0
- control_plane_api/app/routers/team_context.py +274 -0
- control_plane_api/app/routers/teams.py +1747 -0
- control_plane_api/app/routers/templates.py +248 -0
- control_plane_api/app/routers/traces.py +571 -0
- control_plane_api/app/routers/websocket_client.py +479 -0
- control_plane_api/app/routers/websocket_executions_status.py +437 -0
- control_plane_api/app/routers/websocket_gateway.py +323 -0
- control_plane_api/app/routers/websocket_traces.py +576 -0
- control_plane_api/app/routers/worker_queues.py +2555 -0
- control_plane_api/app/routers/worker_websocket.py +419 -0
- control_plane_api/app/routers/workers.py +1004 -0
- control_plane_api/app/routers/workflows.py +204 -0
- control_plane_api/app/runtimes/__init__.py +6 -0
- control_plane_api/app/runtimes/validation.py +344 -0
- control_plane_api/app/schemas/__init__.py +1 -0
- control_plane_api/app/schemas/job_schemas.py +302 -0
- control_plane_api/app/schemas/mcp_schemas.py +311 -0
- control_plane_api/app/schemas/template_schemas.py +133 -0
- control_plane_api/app/schemas/trace_schemas.py +168 -0
- control_plane_api/app/schemas/worker_queue_observability_schemas.py +165 -0
- control_plane_api/app/services/__init__.py +1 -0
- control_plane_api/app/services/agno_planning_strategy.py +233 -0
- control_plane_api/app/services/agno_service.py +838 -0
- control_plane_api/app/services/claude_code_planning_service.py +203 -0
- control_plane_api/app/services/context_graph_client.py +224 -0
- control_plane_api/app/services/custom_integration_service.py +415 -0
- control_plane_api/app/services/integration_resolution_service.py +345 -0
- control_plane_api/app/services/litellm_service.py +394 -0
- control_plane_api/app/services/plan_generator.py +79 -0
- control_plane_api/app/services/planning_strategy.py +66 -0
- control_plane_api/app/services/planning_strategy_factory.py +118 -0
- control_plane_api/app/services/policy_service.py +615 -0
- control_plane_api/app/services/state_transition_service.py +755 -0
- control_plane_api/app/services/storage_service.py +593 -0
- control_plane_api/app/services/temporal_cloud_provisioning.py +150 -0
- control_plane_api/app/services/toolsets/context_graph_skill.py +432 -0
- control_plane_api/app/services/trace_retention.py +354 -0
- control_plane_api/app/services/worker_queue_metrics_service.py +190 -0
- control_plane_api/app/services/workflow_cancellation_manager.py +135 -0
- control_plane_api/app/services/workflow_operations_service.py +611 -0
- control_plane_api/app/skills/__init__.py +100 -0
- control_plane_api/app/skills/base.py +239 -0
- control_plane_api/app/skills/builtin/__init__.py +37 -0
- control_plane_api/app/skills/builtin/agent_communication/__init__.py +8 -0
- control_plane_api/app/skills/builtin/agent_communication/skill.py +246 -0
- control_plane_api/app/skills/builtin/code_ingestion/__init__.py +4 -0
- control_plane_api/app/skills/builtin/code_ingestion/skill.py +267 -0
- control_plane_api/app/skills/builtin/cognitive_memory/__init__.py +4 -0
- control_plane_api/app/skills/builtin/cognitive_memory/skill.py +174 -0
- control_plane_api/app/skills/builtin/contextual_awareness/__init__.py +4 -0
- control_plane_api/app/skills/builtin/contextual_awareness/skill.py +387 -0
- control_plane_api/app/skills/builtin/data_visualization/__init__.py +4 -0
- control_plane_api/app/skills/builtin/data_visualization/skill.py +154 -0
- control_plane_api/app/skills/builtin/docker/__init__.py +4 -0
- control_plane_api/app/skills/builtin/docker/skill.py +104 -0
- control_plane_api/app/skills/builtin/file_generation/__init__.py +4 -0
- control_plane_api/app/skills/builtin/file_generation/skill.py +94 -0
- control_plane_api/app/skills/builtin/file_system/__init__.py +4 -0
- control_plane_api/app/skills/builtin/file_system/skill.py +110 -0
- control_plane_api/app/skills/builtin/knowledge_api/__init__.py +5 -0
- control_plane_api/app/skills/builtin/knowledge_api/skill.py +124 -0
- control_plane_api/app/skills/builtin/python/__init__.py +4 -0
- control_plane_api/app/skills/builtin/python/skill.py +92 -0
- control_plane_api/app/skills/builtin/remote_filesystem/__init__.py +5 -0
- control_plane_api/app/skills/builtin/remote_filesystem/skill.py +170 -0
- control_plane_api/app/skills/builtin/shell/__init__.py +4 -0
- control_plane_api/app/skills/builtin/shell/skill.py +161 -0
- control_plane_api/app/skills/builtin/slack/__init__.py +3 -0
- control_plane_api/app/skills/builtin/slack/skill.py +302 -0
- control_plane_api/app/skills/builtin/workflow_executor/__init__.py +4 -0
- control_plane_api/app/skills/builtin/workflow_executor/skill.py +469 -0
- control_plane_api/app/skills/business_intelligence.py +189 -0
- control_plane_api/app/skills/config.py +63 -0
- control_plane_api/app/skills/loaders/__init__.py +14 -0
- control_plane_api/app/skills/loaders/base.py +73 -0
- control_plane_api/app/skills/loaders/filesystem_loader.py +199 -0
- control_plane_api/app/skills/registry.py +125 -0
- control_plane_api/app/utils/helpers.py +12 -0
- control_plane_api/app/utils/workflow_executor.py +354 -0
- control_plane_api/app/workflows/__init__.py +11 -0
- control_plane_api/app/workflows/agent_execution.py +520 -0
- control_plane_api/app/workflows/agent_execution_with_skills.py +223 -0
- control_plane_api/app/workflows/namespace_provisioning.py +326 -0
- control_plane_api/app/workflows/plan_generation.py +254 -0
- control_plane_api/app/workflows/team_execution.py +442 -0
- control_plane_api/scripts/seed_models.py +240 -0
- control_plane_api/scripts/validate_existing_tool_names.py +492 -0
- control_plane_api/shared/__init__.py +8 -0
- control_plane_api/shared/version.py +17 -0
- control_plane_api/test_deduplication.py +274 -0
- control_plane_api/test_executor_deduplication_e2e.py +309 -0
- control_plane_api/test_job_execution_e2e.py +283 -0
- control_plane_api/test_real_integration.py +193 -0
- control_plane_api/version.py +38 -0
- control_plane_api/worker/__init__.py +0 -0
- control_plane_api/worker/activities/__init__.py +0 -0
- control_plane_api/worker/activities/agent_activities.py +1585 -0
- control_plane_api/worker/activities/approval_activities.py +234 -0
- control_plane_api/worker/activities/job_activities.py +199 -0
- control_plane_api/worker/activities/runtime_activities.py +1167 -0
- control_plane_api/worker/activities/skill_activities.py +282 -0
- control_plane_api/worker/activities/team_activities.py +479 -0
- control_plane_api/worker/agent_runtime_server.py +370 -0
- control_plane_api/worker/binary_manager.py +333 -0
- control_plane_api/worker/config/__init__.py +31 -0
- control_plane_api/worker/config/worker_config.py +273 -0
- control_plane_api/worker/control_plane_client.py +1491 -0
- control_plane_api/worker/examples/analytics_integration_example.py +362 -0
- control_plane_api/worker/health_monitor.py +159 -0
- control_plane_api/worker/metrics.py +237 -0
- control_plane_api/worker/models/__init__.py +1 -0
- control_plane_api/worker/models/error_events.py +105 -0
- control_plane_api/worker/models/inputs.py +89 -0
- control_plane_api/worker/runtimes/__init__.py +35 -0
- control_plane_api/worker/runtimes/agent_runtime/runtime.py +485 -0
- control_plane_api/worker/runtimes/agno/__init__.py +34 -0
- control_plane_api/worker/runtimes/agno/config.py +248 -0
- control_plane_api/worker/runtimes/agno/hooks.py +385 -0
- control_plane_api/worker/runtimes/agno/mcp_builder.py +195 -0
- control_plane_api/worker/runtimes/agno/runtime.py +1063 -0
- control_plane_api/worker/runtimes/agno/utils.py +163 -0
- control_plane_api/worker/runtimes/base.py +979 -0
- control_plane_api/worker/runtimes/claude_code/__init__.py +38 -0
- control_plane_api/worker/runtimes/claude_code/cleanup.py +184 -0
- control_plane_api/worker/runtimes/claude_code/client_pool.py +529 -0
- control_plane_api/worker/runtimes/claude_code/config.py +829 -0
- control_plane_api/worker/runtimes/claude_code/hooks.py +482 -0
- control_plane_api/worker/runtimes/claude_code/litellm_proxy.py +1702 -0
- control_plane_api/worker/runtimes/claude_code/mcp_builder.py +467 -0
- control_plane_api/worker/runtimes/claude_code/mcp_discovery.py +558 -0
- control_plane_api/worker/runtimes/claude_code/runtime.py +1546 -0
- control_plane_api/worker/runtimes/claude_code/tool_mapper.py +403 -0
- control_plane_api/worker/runtimes/claude_code/utils.py +149 -0
- control_plane_api/worker/runtimes/factory.py +173 -0
- control_plane_api/worker/runtimes/model_utils.py +107 -0
- control_plane_api/worker/runtimes/validation.py +93 -0
- control_plane_api/worker/services/__init__.py +1 -0
- control_plane_api/worker/services/agent_communication_tools.py +908 -0
- control_plane_api/worker/services/agent_executor.py +485 -0
- control_plane_api/worker/services/agent_executor_v2.py +793 -0
- control_plane_api/worker/services/analytics_collector.py +457 -0
- control_plane_api/worker/services/analytics_service.py +464 -0
- control_plane_api/worker/services/approval_tools.py +310 -0
- control_plane_api/worker/services/approval_tools_agno.py +207 -0
- control_plane_api/worker/services/cancellation_manager.py +177 -0
- control_plane_api/worker/services/code_ingestion_tools.py +465 -0
- control_plane_api/worker/services/contextual_awareness_tools.py +405 -0
- control_plane_api/worker/services/data_visualization.py +834 -0
- control_plane_api/worker/services/event_publisher.py +531 -0
- control_plane_api/worker/services/jira_tools.py +257 -0
- control_plane_api/worker/services/remote_filesystem_tools.py +498 -0
- control_plane_api/worker/services/runtime_analytics.py +328 -0
- control_plane_api/worker/services/session_service.py +365 -0
- control_plane_api/worker/services/skill_context_enhancement.py +181 -0
- control_plane_api/worker/services/skill_factory.py +471 -0
- control_plane_api/worker/services/system_prompt_enhancement.py +410 -0
- control_plane_api/worker/services/team_executor.py +715 -0
- control_plane_api/worker/services/team_executor_v2.py +1866 -0
- control_plane_api/worker/services/tool_enforcement.py +254 -0
- control_plane_api/worker/services/workflow_executor/__init__.py +52 -0
- control_plane_api/worker/services/workflow_executor/event_processor.py +287 -0
- control_plane_api/worker/services/workflow_executor/event_publisher.py +210 -0
- control_plane_api/worker/services/workflow_executor/executors/__init__.py +15 -0
- control_plane_api/worker/services/workflow_executor/executors/base.py +270 -0
- control_plane_api/worker/services/workflow_executor/executors/json_executor.py +50 -0
- control_plane_api/worker/services/workflow_executor/executors/python_executor.py +50 -0
- control_plane_api/worker/services/workflow_executor/models.py +142 -0
- control_plane_api/worker/services/workflow_executor_tools.py +1748 -0
- control_plane_api/worker/skills/__init__.py +12 -0
- control_plane_api/worker/skills/builtin/context_graph_search/README.md +213 -0
- control_plane_api/worker/skills/builtin/context_graph_search/__init__.py +5 -0
- control_plane_api/worker/skills/builtin/context_graph_search/agno_impl.py +808 -0
- control_plane_api/worker/skills/builtin/context_graph_search/skill.yaml +67 -0
- control_plane_api/worker/skills/builtin/contextual_awareness/__init__.py +4 -0
- control_plane_api/worker/skills/builtin/contextual_awareness/agno_impl.py +62 -0
- control_plane_api/worker/skills/builtin/data_visualization/agno_impl.py +18 -0
- control_plane_api/worker/skills/builtin/data_visualization/skill.yaml +84 -0
- control_plane_api/worker/skills/builtin/docker/agno_impl.py +65 -0
- control_plane_api/worker/skills/builtin/docker/skill.yaml +60 -0
- control_plane_api/worker/skills/builtin/file_generation/agno_impl.py +47 -0
- control_plane_api/worker/skills/builtin/file_generation/skill.yaml +64 -0
- control_plane_api/worker/skills/builtin/file_system/agno_impl.py +32 -0
- control_plane_api/worker/skills/builtin/file_system/skill.yaml +54 -0
- control_plane_api/worker/skills/builtin/knowledge_api/__init__.py +4 -0
- control_plane_api/worker/skills/builtin/knowledge_api/agno_impl.py +50 -0
- control_plane_api/worker/skills/builtin/knowledge_api/skill.yaml +66 -0
- control_plane_api/worker/skills/builtin/python/agno_impl.py +25 -0
- control_plane_api/worker/skills/builtin/python/skill.yaml +60 -0
- control_plane_api/worker/skills/builtin/schema_fix_mixin.py +260 -0
- control_plane_api/worker/skills/builtin/shell/agno_impl.py +31 -0
- control_plane_api/worker/skills/builtin/shell/skill.yaml +60 -0
- control_plane_api/worker/skills/builtin/slack/__init__.py +3 -0
- control_plane_api/worker/skills/builtin/slack/agno_impl.py +1282 -0
- control_plane_api/worker/skills/builtin/slack/skill.yaml +276 -0
- control_plane_api/worker/skills/builtin/workflow_executor/agno_impl.py +62 -0
- control_plane_api/worker/skills/builtin/workflow_executor/skill.yaml +79 -0
- control_plane_api/worker/skills/loaders/__init__.py +5 -0
- control_plane_api/worker/skills/loaders/base.py +23 -0
- control_plane_api/worker/skills/loaders/filesystem_loader.py +357 -0
- control_plane_api/worker/skills/registry.py +208 -0
- control_plane_api/worker/tests/__init__.py +1 -0
- control_plane_api/worker/tests/conftest.py +12 -0
- control_plane_api/worker/tests/e2e/__init__.py +0 -0
- control_plane_api/worker/tests/e2e/test_context_graph_real_api.py +338 -0
- control_plane_api/worker/tests/e2e/test_context_graph_templates_e2e.py +523 -0
- control_plane_api/worker/tests/e2e/test_enforcement_e2e.py +344 -0
- control_plane_api/worker/tests/e2e/test_execution_flow.py +571 -0
- control_plane_api/worker/tests/e2e/test_single_execution_mode.py +656 -0
- control_plane_api/worker/tests/integration/__init__.py +0 -0
- control_plane_api/worker/tests/integration/test_builtin_skills_fixes.py +245 -0
- control_plane_api/worker/tests/integration/test_context_graph_search_integration.py +365 -0
- control_plane_api/worker/tests/integration/test_control_plane_integration.py +308 -0
- control_plane_api/worker/tests/integration/test_hook_enforcement_integration.py +579 -0
- control_plane_api/worker/tests/integration/test_scheduled_job_workflow.py +237 -0
- control_plane_api/worker/tests/integration/test_system_prompt_enhancement_integration.py +343 -0
- control_plane_api/worker/tests/unit/__init__.py +0 -0
- control_plane_api/worker/tests/unit/test_builtin_skill_autoload.py +396 -0
- control_plane_api/worker/tests/unit/test_context_graph_search.py +450 -0
- control_plane_api/worker/tests/unit/test_context_graph_templates.py +403 -0
- control_plane_api/worker/tests/unit/test_control_plane_client.py +401 -0
- control_plane_api/worker/tests/unit/test_control_plane_client_jobs.py +345 -0
- control_plane_api/worker/tests/unit/test_job_activities.py +353 -0
- control_plane_api/worker/tests/unit/test_skill_context_enhancement.py +321 -0
- control_plane_api/worker/tests/unit/test_system_prompt_enhancement.py +415 -0
- control_plane_api/worker/tests/unit/test_tool_enforcement.py +324 -0
- control_plane_api/worker/utils/__init__.py +1 -0
- control_plane_api/worker/utils/chunk_batcher.py +330 -0
- control_plane_api/worker/utils/environment.py +65 -0
- control_plane_api/worker/utils/error_publisher.py +260 -0
- control_plane_api/worker/utils/event_batcher.py +256 -0
- control_plane_api/worker/utils/logging_config.py +335 -0
- control_plane_api/worker/utils/logging_helper.py +326 -0
- control_plane_api/worker/utils/parameter_validator.py +120 -0
- control_plane_api/worker/utils/retry_utils.py +60 -0
- control_plane_api/worker/utils/streaming_utils.py +665 -0
- control_plane_api/worker/utils/tool_validation.py +332 -0
- control_plane_api/worker/utils/workspace_manager.py +163 -0
- control_plane_api/worker/websocket_client.py +393 -0
- control_plane_api/worker/worker.py +1297 -0
- control_plane_api/worker/workflows/__init__.py +0 -0
- control_plane_api/worker/workflows/agent_execution.py +909 -0
- control_plane_api/worker/workflows/scheduled_job_wrapper.py +332 -0
- control_plane_api/worker/workflows/team_execution.py +611 -0
- kubiya_control_plane_api-0.9.15.dist-info/METADATA +354 -0
- kubiya_control_plane_api-0.9.15.dist-info/RECORD +479 -0
- kubiya_control_plane_api-0.9.15.dist-info/WHEEL +5 -0
- kubiya_control_plane_api-0.9.15.dist-info/entry_points.txt +5 -0
- kubiya_control_plane_api-0.9.15.dist-info/licenses/LICENSE +676 -0
- kubiya_control_plane_api-0.9.15.dist-info/top_level.txt +3 -0
- scripts/__init__.py +1 -0
- scripts/migrations.py +39 -0
- scripts/seed_worker_queues.py +128 -0
- scripts/setup_agent_runtime.py +142 -0
- worker_internal/__init__.py +1 -0
- worker_internal/planner/__init__.py +1 -0
- worker_internal/planner/activities.py +1499 -0
- worker_internal/planner/agent_tools.py +197 -0
- worker_internal/planner/event_models.py +148 -0
- worker_internal/planner/event_publisher.py +67 -0
- worker_internal/planner/models.py +199 -0
- worker_internal/planner/retry_logic.py +134 -0
- worker_internal/planner/worker.py +300 -0
- worker_internal/planner/workflows.py +970 -0
|
@@ -0,0 +1,276 @@
|
|
|
1
|
+
apiVersion: kubiya.ai/v1
|
|
2
|
+
kind: Skill
|
|
3
|
+
metadata:
|
|
4
|
+
name: slack
|
|
5
|
+
version: 1.0.0
|
|
6
|
+
description: Slack integration with OAuth support and configurable capabilities for messaging, reactions, and channel management
|
|
7
|
+
author: Kubiya
|
|
8
|
+
icon: Slack
|
|
9
|
+
icon_type: react-icon
|
|
10
|
+
|
|
11
|
+
spec:
|
|
12
|
+
type: slack
|
|
13
|
+
|
|
14
|
+
implementations:
|
|
15
|
+
agno:
|
|
16
|
+
module: agno_impl
|
|
17
|
+
class: SlackTools
|
|
18
|
+
claude_code:
|
|
19
|
+
module: agno_impl
|
|
20
|
+
class: SlackTools
|
|
21
|
+
default:
|
|
22
|
+
module: agno_impl
|
|
23
|
+
class: SlackTools
|
|
24
|
+
|
|
25
|
+
pythonDependencies:
|
|
26
|
+
- slack-sdk>=3.27.0
|
|
27
|
+
- httpx>=0.27.0
|
|
28
|
+
|
|
29
|
+
systemRequirements:
|
|
30
|
+
supportedOS: [linux, darwin, win32]
|
|
31
|
+
minPythonVersion: "3.10"
|
|
32
|
+
|
|
33
|
+
environmentVariables:
|
|
34
|
+
- name: SLACK_BOT_TOKEN
|
|
35
|
+
required: false
|
|
36
|
+
description: Slack Bot User OAuth Token (xoxb-...). Used as fallback if Integration API is not configured.
|
|
37
|
+
|
|
38
|
+
- name: INTEGRATION_API_BASE
|
|
39
|
+
required: false
|
|
40
|
+
description: Integration API base URL for OAuth token fetching. Defaults to https://api.kubiya.ai
|
|
41
|
+
|
|
42
|
+
- name: KUBIYA_API_KEY
|
|
43
|
+
required: false
|
|
44
|
+
description: Kubiya API key for Integration API authentication. Required when using integration_id.
|
|
45
|
+
|
|
46
|
+
configSchema:
|
|
47
|
+
type: object
|
|
48
|
+
properties:
|
|
49
|
+
# Authentication Configuration
|
|
50
|
+
integration_id:
|
|
51
|
+
type: string
|
|
52
|
+
description: Integration API ID for OAuth-based authentication. Recommended for multi-workspace support.
|
|
53
|
+
|
|
54
|
+
secret_name:
|
|
55
|
+
type: string
|
|
56
|
+
description: |
|
|
57
|
+
(Deprecated: use 'secrets' instead) Name of a single secret containing the Slack Bot Token.
|
|
58
|
+
The secret value will be injected as an environment variable at runtime.
|
|
59
|
+
|
|
60
|
+
secrets:
|
|
61
|
+
type: array
|
|
62
|
+
description: |
|
|
63
|
+
List of secret names from the Kubiya secrets vault. Secret values will be injected as
|
|
64
|
+
environment variables at runtime (e.g., secret named "SLACK_TOKEN" becomes env var SLACK_TOKEN).
|
|
65
|
+
The first available secret will be used for authentication.
|
|
66
|
+
items:
|
|
67
|
+
type: string
|
|
68
|
+
example: ["SLACK_BOT_TOKEN", "SLACK_TOKEN_PROD"]
|
|
69
|
+
|
|
70
|
+
use_env_fallback:
|
|
71
|
+
type: boolean
|
|
72
|
+
description: Allow SLACK_BOT_TOKEN environment variable as fallback authentication method
|
|
73
|
+
default: true
|
|
74
|
+
|
|
75
|
+
# Capability Group Toggles
|
|
76
|
+
enable_read_messages:
|
|
77
|
+
type: boolean
|
|
78
|
+
description: Enable read message capabilities (channels, history, search, threads, users)
|
|
79
|
+
default: true
|
|
80
|
+
|
|
81
|
+
enable_write_messages:
|
|
82
|
+
type: boolean
|
|
83
|
+
description: Enable write message capabilities (send, post, reply, update, delete)
|
|
84
|
+
default: false
|
|
85
|
+
|
|
86
|
+
enable_reactions:
|
|
87
|
+
type: boolean
|
|
88
|
+
description: Enable reaction capabilities (add, remove, list reactions)
|
|
89
|
+
default: false
|
|
90
|
+
|
|
91
|
+
enable_channel_management:
|
|
92
|
+
type: boolean
|
|
93
|
+
description: Enable channel management capabilities (create, invite, archive, topics)
|
|
94
|
+
default: false
|
|
95
|
+
|
|
96
|
+
# Fine-grained Read Permissions
|
|
97
|
+
read_channels:
|
|
98
|
+
type: boolean
|
|
99
|
+
description: Allow listing channels in the workspace
|
|
100
|
+
default: true
|
|
101
|
+
|
|
102
|
+
read_messages:
|
|
103
|
+
type: boolean
|
|
104
|
+
description: Allow reading message history from channels
|
|
105
|
+
default: true
|
|
106
|
+
|
|
107
|
+
search_messages:
|
|
108
|
+
type: boolean
|
|
109
|
+
description: Allow searching messages across workspace
|
|
110
|
+
default: true
|
|
111
|
+
|
|
112
|
+
read_threads:
|
|
113
|
+
type: boolean
|
|
114
|
+
description: Allow reading thread replies
|
|
115
|
+
default: true
|
|
116
|
+
|
|
117
|
+
read_users:
|
|
118
|
+
type: boolean
|
|
119
|
+
description: Allow reading user information and listing users
|
|
120
|
+
default: true
|
|
121
|
+
|
|
122
|
+
# Fine-grained Write Permissions
|
|
123
|
+
send_messages:
|
|
124
|
+
type: boolean
|
|
125
|
+
description: Allow sending messages to channels
|
|
126
|
+
default: true
|
|
127
|
+
|
|
128
|
+
post_messages:
|
|
129
|
+
type: boolean
|
|
130
|
+
description: Allow posting messages (alias for send_messages)
|
|
131
|
+
default: true
|
|
132
|
+
|
|
133
|
+
reply_to_threads:
|
|
134
|
+
type: boolean
|
|
135
|
+
description: Allow replying to message threads
|
|
136
|
+
default: true
|
|
137
|
+
|
|
138
|
+
update_messages:
|
|
139
|
+
type: boolean
|
|
140
|
+
description: Allow updating and deleting messages
|
|
141
|
+
default: true
|
|
142
|
+
|
|
143
|
+
# Fine-grained Reaction Permissions
|
|
144
|
+
add_reactions:
|
|
145
|
+
type: boolean
|
|
146
|
+
description: Allow adding reactions to messages
|
|
147
|
+
default: true
|
|
148
|
+
|
|
149
|
+
remove_reactions:
|
|
150
|
+
type: boolean
|
|
151
|
+
description: Allow removing reactions from messages
|
|
152
|
+
default: true
|
|
153
|
+
|
|
154
|
+
# Fine-grained Channel Management Permissions
|
|
155
|
+
create_channels:
|
|
156
|
+
type: boolean
|
|
157
|
+
description: Allow creating new channels
|
|
158
|
+
default: false
|
|
159
|
+
|
|
160
|
+
invite_to_channels:
|
|
161
|
+
type: boolean
|
|
162
|
+
description: Allow inviting users to channels
|
|
163
|
+
default: false
|
|
164
|
+
|
|
165
|
+
archive_channels:
|
|
166
|
+
type: boolean
|
|
167
|
+
description: Allow archiving channels
|
|
168
|
+
default: false
|
|
169
|
+
|
|
170
|
+
set_channel_topic:
|
|
171
|
+
type: boolean
|
|
172
|
+
description: Allow setting channel topics and descriptions
|
|
173
|
+
default: false
|
|
174
|
+
|
|
175
|
+
# General Settings
|
|
176
|
+
timeout:
|
|
177
|
+
type: integer
|
|
178
|
+
description: API request timeout in seconds
|
|
179
|
+
default: 30
|
|
180
|
+
minimum: 5
|
|
181
|
+
maximum: 300
|
|
182
|
+
|
|
183
|
+
default_channel:
|
|
184
|
+
type: string
|
|
185
|
+
description: Default channel for operations (optional)
|
|
186
|
+
|
|
187
|
+
# Channel Scoping
|
|
188
|
+
allowed_channels:
|
|
189
|
+
type: array
|
|
190
|
+
description: |
|
|
191
|
+
Whitelist of channels this skill can access. Can be channel IDs (C01234567)
|
|
192
|
+
or channel names (general, team-chat). If specified, only these channels are accessible.
|
|
193
|
+
items:
|
|
194
|
+
type: string
|
|
195
|
+
example: ["general", "bot-testing", "C01234567"]
|
|
196
|
+
|
|
197
|
+
blocked_channels:
|
|
198
|
+
type: array
|
|
199
|
+
description: |
|
|
200
|
+
Blacklist of channels this skill cannot access. Can be channel IDs or channel names.
|
|
201
|
+
Takes precedence over allowed_channels if both are specified.
|
|
202
|
+
items:
|
|
203
|
+
type: string
|
|
204
|
+
example: ["admin-only", "sensitive-data"]
|
|
205
|
+
|
|
206
|
+
usage_guidelines:
|
|
207
|
+
type: string
|
|
208
|
+
description: Custom usage guidelines for AI agent system prompt injection
|
|
209
|
+
|
|
210
|
+
requiredScopes:
|
|
211
|
+
slack:
|
|
212
|
+
- channels:read
|
|
213
|
+
- channels:history
|
|
214
|
+
- chat:write
|
|
215
|
+
- users:read
|
|
216
|
+
- reactions:read
|
|
217
|
+
- reactions:write
|
|
218
|
+
- channels:manage
|
|
219
|
+
- groups:read
|
|
220
|
+
- groups:history
|
|
221
|
+
- groups:write
|
|
222
|
+
- im:read
|
|
223
|
+
- im:history
|
|
224
|
+
- im:write
|
|
225
|
+
- mpim:read
|
|
226
|
+
- mpim:history
|
|
227
|
+
- mpim:write
|
|
228
|
+
- search:read
|
|
229
|
+
|
|
230
|
+
examples:
|
|
231
|
+
- name: Get available channels
|
|
232
|
+
description: Get list of channels available to this skill based on scoping configuration
|
|
233
|
+
code: |
|
|
234
|
+
get_available_channels(types="public_channel,private_channel", limit=100)
|
|
235
|
+
|
|
236
|
+
- name: List all channels
|
|
237
|
+
description: Get a list of all public and private channels
|
|
238
|
+
code: |
|
|
239
|
+
list_channels(types="public_channel,private_channel", limit=100)
|
|
240
|
+
|
|
241
|
+
- name: Send a message
|
|
242
|
+
description: Send a simple message to a channel
|
|
243
|
+
code: |
|
|
244
|
+
send_message(channel="#general", text="Hello from Kubiya!")
|
|
245
|
+
|
|
246
|
+
- name: Reply to a thread
|
|
247
|
+
description: Reply to a specific message thread
|
|
248
|
+
code: |
|
|
249
|
+
reply_to_thread(
|
|
250
|
+
channel="C01234567",
|
|
251
|
+
thread_ts="1234567890.123456",
|
|
252
|
+
text="This is a threaded reply"
|
|
253
|
+
)
|
|
254
|
+
|
|
255
|
+
- name: Search messages
|
|
256
|
+
description: Search for messages across the workspace
|
|
257
|
+
code: |
|
|
258
|
+
search_messages(query="error logs", count=20, sort="timestamp")
|
|
259
|
+
|
|
260
|
+
- name: Add a reaction
|
|
261
|
+
description: Add a reaction emoji to a message
|
|
262
|
+
code: |
|
|
263
|
+
add_reaction(
|
|
264
|
+
channel="C01234567",
|
|
265
|
+
ts="1234567890.123456",
|
|
266
|
+
reaction="thumbsup"
|
|
267
|
+
)
|
|
268
|
+
|
|
269
|
+
- name: Create a channel
|
|
270
|
+
description: Create a new public or private channel
|
|
271
|
+
code: |
|
|
272
|
+
create_channel(
|
|
273
|
+
name="project-alpha",
|
|
274
|
+
is_private=False,
|
|
275
|
+
description="Discussion channel for Project Alpha"
|
|
276
|
+
)
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
"""Workflow Executor skill implementation for agno runtime."""
|
|
2
|
+
from control_plane_api.worker.services.workflow_executor_tools import WorkflowExecutorTools as BaseWorkflowExecutorTools
|
|
3
|
+
from control_plane_api.worker.skills.builtin.schema_fix_mixin import SchemaFixMixin
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
class WorkflowExecutorTools(BaseWorkflowExecutorTools):
|
|
8
|
+
"""
|
|
9
|
+
Workflow executor tools using existing WorkflowExecutorTools.
|
|
10
|
+
|
|
11
|
+
Wraps the existing workflow executor implementation.
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
def __init__(
|
|
15
|
+
self,
|
|
16
|
+
workflows: list = None,
|
|
17
|
+
workflow_type: str = None,
|
|
18
|
+
workflow_definition: str = None,
|
|
19
|
+
python_dsl_code: str = None,
|
|
20
|
+
validation_enabled: bool = True,
|
|
21
|
+
default_runner: str = None,
|
|
22
|
+
timeout: int = 3600,
|
|
23
|
+
default_parameters: dict = None,
|
|
24
|
+
kubiya_api_key: str = None,
|
|
25
|
+
kubiya_api_base: str = None,
|
|
26
|
+
execution_id: str = None,
|
|
27
|
+
**kwargs
|
|
28
|
+
):
|
|
29
|
+
"""
|
|
30
|
+
Initialize workflow executor tools.
|
|
31
|
+
|
|
32
|
+
Args:
|
|
33
|
+
workflows: Collection of workflow definitions
|
|
34
|
+
workflow_type: LEGACY - Workflow type (json or python_dsl)
|
|
35
|
+
workflow_definition: LEGACY - JSON workflow definition
|
|
36
|
+
python_dsl_code: LEGACY - Python DSL code
|
|
37
|
+
validation_enabled: Enable validation
|
|
38
|
+
default_runner: Default runner/environment
|
|
39
|
+
timeout: Execution timeout
|
|
40
|
+
default_parameters: Default parameters
|
|
41
|
+
kubiya_api_key: Kubiya API key
|
|
42
|
+
kubiya_api_base: Kubiya API base URL
|
|
43
|
+
execution_id: Execution ID for streaming
|
|
44
|
+
**kwargs: Additional configuration
|
|
45
|
+
"""
|
|
46
|
+
# Import os here to get env vars if not provided
|
|
47
|
+
import os
|
|
48
|
+
|
|
49
|
+
super().__init__(
|
|
50
|
+
name=kwargs.get("name", "workflow-executor"),
|
|
51
|
+
workflows=workflows or [],
|
|
52
|
+
workflow_type=workflow_type,
|
|
53
|
+
workflow_definition=workflow_definition,
|
|
54
|
+
python_dsl_code=python_dsl_code,
|
|
55
|
+
validation_enabled=validation_enabled,
|
|
56
|
+
default_runner=default_runner,
|
|
57
|
+
timeout=timeout,
|
|
58
|
+
default_parameters=default_parameters,
|
|
59
|
+
kubiya_api_key=kubiya_api_key or os.environ.get("KUBIYA_API_KEY"),
|
|
60
|
+
kubiya_api_base=kubiya_api_base or os.environ.get("KUBIYA_API_BASE", "https://api.kubiya.ai"),
|
|
61
|
+
execution_id=execution_id,
|
|
62
|
+
)
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
apiVersion: kubiya.ai/v1
|
|
2
|
+
kind: Skill
|
|
3
|
+
metadata:
|
|
4
|
+
name: workflow-executor
|
|
5
|
+
version: 1.0.0
|
|
6
|
+
description: Execute workflows defined via JSON or Python DSL
|
|
7
|
+
author: Kubiya
|
|
8
|
+
icon: Workflow
|
|
9
|
+
icon_type: lucide
|
|
10
|
+
|
|
11
|
+
spec:
|
|
12
|
+
type: workflow_executor
|
|
13
|
+
|
|
14
|
+
implementations:
|
|
15
|
+
agno:
|
|
16
|
+
module: agno_impl
|
|
17
|
+
class: WorkflowExecutorTools
|
|
18
|
+
claude_code:
|
|
19
|
+
# Claude Code executes workflow steps via its tool system
|
|
20
|
+
# No separate implementation needed - uses existing WorkflowExecutorTools
|
|
21
|
+
module: agno_impl
|
|
22
|
+
class: WorkflowExecutorTools
|
|
23
|
+
default:
|
|
24
|
+
module: agno_impl
|
|
25
|
+
class: WorkflowExecutorTools
|
|
26
|
+
|
|
27
|
+
pythonDependencies:
|
|
28
|
+
- kubiya-sdk
|
|
29
|
+
|
|
30
|
+
systemRequirements:
|
|
31
|
+
supportedOS: [linux, darwin, win32]
|
|
32
|
+
minPythonVersion: "3.10"
|
|
33
|
+
|
|
34
|
+
environmentVariables:
|
|
35
|
+
- name: KUBIYA_API_KEY
|
|
36
|
+
required: false
|
|
37
|
+
description: Kubiya API key for remote workflow execution
|
|
38
|
+
- name: KUBIYA_API_BASE
|
|
39
|
+
required: false
|
|
40
|
+
description: Kubiya API base URL
|
|
41
|
+
|
|
42
|
+
configSchema:
|
|
43
|
+
type: object
|
|
44
|
+
properties:
|
|
45
|
+
workflows:
|
|
46
|
+
type: array
|
|
47
|
+
description: Collection of workflow definitions
|
|
48
|
+
default: []
|
|
49
|
+
validation_enabled:
|
|
50
|
+
type: boolean
|
|
51
|
+
description: Enable workflow validation
|
|
52
|
+
default: true
|
|
53
|
+
default_runner:
|
|
54
|
+
type: string
|
|
55
|
+
description: Default runner/environment
|
|
56
|
+
default: ""
|
|
57
|
+
timeout:
|
|
58
|
+
type: integer
|
|
59
|
+
description: Workflow execution timeout in seconds
|
|
60
|
+
default: 3600
|
|
61
|
+
minimum: 30
|
|
62
|
+
maximum: 7200
|
|
63
|
+
default_parameters:
|
|
64
|
+
type: object
|
|
65
|
+
description: Default parameter values
|
|
66
|
+
default: {}
|
|
67
|
+
# Legacy fields
|
|
68
|
+
workflow_type:
|
|
69
|
+
type: string
|
|
70
|
+
description: "LEGACY: Workflow type (json or python_dsl)"
|
|
71
|
+
default: ""
|
|
72
|
+
workflow_definition:
|
|
73
|
+
type: string
|
|
74
|
+
description: "LEGACY: JSON workflow definition"
|
|
75
|
+
default: ""
|
|
76
|
+
python_dsl_code:
|
|
77
|
+
type: string
|
|
78
|
+
description: "LEGACY: Python DSL code"
|
|
79
|
+
default: ""
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
"""Base loader interface for skill discovery."""
|
|
2
|
+
from abc import ABC, abstractmethod
|
|
3
|
+
from typing import List
|
|
4
|
+
from control_plane_api.worker.skills.registry import LoadedSkill, SkillSource
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
class BaseSkillLoader(ABC):
|
|
8
|
+
"""Base class for skill loaders."""
|
|
9
|
+
|
|
10
|
+
@abstractmethod
|
|
11
|
+
def discover(self) -> List[LoadedSkill]:
|
|
12
|
+
"""Discover available skills from this source."""
|
|
13
|
+
pass
|
|
14
|
+
|
|
15
|
+
@abstractmethod
|
|
16
|
+
def get_source_type(self) -> SkillSource:
|
|
17
|
+
"""Get the source type for this loader."""
|
|
18
|
+
pass
|
|
19
|
+
|
|
20
|
+
@abstractmethod
|
|
21
|
+
def load_skill(self, skill_id: str) -> LoadedSkill:
|
|
22
|
+
"""Load a specific skill by ID."""
|
|
23
|
+
pass
|