language-operator 0.0.1 → 0.1.31

data/lib/language_operator/cli/commands/install.rb

@@ -0,0 +1,404 @@
+# frozen_string_literal: true
+
+require 'thor'
+require 'open3'
+require_relative '../formatters/progress_formatter'
+require_relative '../helpers/cluster_validator'
+require_relative '../helpers/user_prompts'
+
+module LanguageOperator
+  module CLI
+    module Commands
+      # Install, upgrade, and uninstall commands for the language-operator
+      class Install < Thor
+        HELM_REPO_NAME = 'language-operator'
+        HELM_REPO_URL = 'https://language-operator.github.io/charts'
+        CHART_NAME = 'language-operator/language-operator'
+        RELEASE_NAME = 'language-operator'
+        DEFAULT_NAMESPACE = 'language-operator-system'
+
+        # Long descriptions for commands
+        LONG_DESCRIPTIONS = {
+          install: <<-DESC,
+            Install the language-operator into your Kubernetes cluster using Helm.
+
+            This command will:
+            1. Add the language-operator Helm repository
+            2. Update Helm repositories
+            3. Install the language-operator chart
+            4. Verify the installation
+
+            Examples:
+              # Install with defaults
+              aictl install
+
+              # Install with custom values
+              aictl install --values my-values.yaml
+
+              # Install specific version
+              aictl install --version 0.1.0
+
+              # Dry run to see what would be installed
+              aictl install --dry-run
+          DESC
+          upgrade: <<-DESC,
+            Upgrade the language-operator to a newer version using Helm.
+
+            This command will:
+            1. Update Helm repositories
+            2. Upgrade the language-operator release
+            3. Wait for the rollout to complete
+            4. Verify the operator is running
+
+            Examples:
+              # Upgrade to latest version
+              aictl upgrade
+
+              # Upgrade with custom values
+              aictl upgrade --values my-values.yaml
+
+              # Upgrade to specific version
+              aictl upgrade --version 0.2.0
+          DESC
+          uninstall: <<-DESC
+            Uninstall the language-operator from your Kubernetes cluster.
+
+            WARNING: This will remove the operator but NOT the CRDs or custom resources.
+            Agents, tools, models, and personas will remain in the cluster.
+
+            Examples:
+              # Uninstall with confirmation
+              aictl uninstall
+
+              # Force uninstall without confirmation
+              aictl uninstall --force
+
+              # Uninstall from specific namespace
+              aictl uninstall --namespace my-namespace
+          DESC
+        }.freeze
+
+        # Helper method to get long descriptions for commands
+        def self.long_desc_for(command)
+          LONG_DESCRIPTIONS[command]
+        end
+
+        desc 'install', 'Install the language-operator using Helm'
+        long_desc <<-DESC
+          Install the language-operator into your Kubernetes cluster using Helm.
+
+          This command will:
+          1. Add the language-operator Helm repository
+          2. Update Helm repositories
+          3. Install the language-operator chart
+          4. Verify the installation
+
+          Examples:
+            # Install with defaults
+            aictl install
+
+            # Install with custom values
+            aictl install --values my-values.yaml
+
+            # Install specific version
+            aictl install --version 0.1.0
+
+            # Dry run to see what would be installed
+            aictl install --dry-run
+        DESC
+        option :values, type: :string, desc: 'Path to custom Helm values file'
+        option :namespace, type: :string, default: DEFAULT_NAMESPACE, desc: 'Kubernetes namespace'
+        option :version, type: :string, desc: 'Specific chart version to install'
+        option :dry_run, type: :boolean, default: false, desc: 'Preview installation without applying'
+        option :wait, type: :boolean, default: true, desc: 'Wait for deployment to complete'
+        option :create_namespace, type: :boolean, default: true, desc: 'Create namespace if it does not exist'
+        def install
+          # Check if helm is available
+          check_helm_installed!
+
+          # Check if operator is already installed
+          if operator_installed? && !options[:dry_run]
+            Formatters::ProgressFormatter.warn('Language operator is already installed')
+            puts
+            puts 'To upgrade, use:'
+            puts '  aictl upgrade'
+            return
+          end
+
+          namespace = options[:namespace]
+
+          puts 'Installing language-operator...'
+          puts "  Namespace: #{namespace}"
+          puts "  Chart: #{CHART_NAME}"
+          puts
+
+          # Add Helm repository
+          add_helm_repo unless options[:dry_run]
+
+          # Build helm install command
+          cmd = build_helm_command('install', namespace)
+
+          # Execute helm install
+          if options[:dry_run]
+            puts 'Dry run - would execute:'
+            puts "  #{cmd}"
+            puts
+            success, output = run_helm_command(cmd)
+            puts output
+          else
+            Formatters::ProgressFormatter.with_spinner('Installing language-operator') do
+              success, output = run_helm_command(cmd)
+              raise "Helm install failed: #{output}" unless success
+            end
+
+            Formatters::ProgressFormatter.success('Language operator installed successfully!')
+            puts
+            puts 'Next steps:'
+            puts '  1. Create a cluster: aictl cluster create my-cluster'
+            puts '  2. Create a model: aictl model create gpt4 --provider openai --model gpt-4-turbo'
+            puts '  3. Create an agent: aictl agent create "your agent description"'
+          end
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Installation failed: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'upgrade', 'Upgrade the language-operator using Helm'
+        long_desc <<-DESC
+          Upgrade the language-operator to a newer version using Helm.
+
+          This command will:
+          1. Update Helm repositories
+          2. Upgrade the language-operator release
+          3. Wait for the rollout to complete
+          4. Verify the operator is running
+
+          Examples:
+            # Upgrade to latest version
+            aictl upgrade
+
+            # Upgrade with custom values
+            aictl upgrade --values my-values.yaml
+
+            # Upgrade to specific version
+            aictl upgrade --version 0.2.0
+        DESC
+        option :values, type: :string, desc: 'Path to custom Helm values file'
+        option :namespace, type: :string, default: DEFAULT_NAMESPACE, desc: 'Kubernetes namespace'
+        option :version, type: :string, desc: 'Specific chart version to upgrade to'
+        option :dry_run, type: :boolean, default: false, desc: 'Preview upgrade without applying'
+        option :wait, type: :boolean, default: true, desc: 'Wait for deployment to complete'
+        def upgrade
+          # Check if helm is available
+          check_helm_installed!
+
+          # Check if operator is installed
+          unless operator_installed?
+            Formatters::ProgressFormatter.error('Language operator is not installed')
+            puts
+            puts 'To install, use:'
+            puts '  aictl install'
+            exit 1
+          end
+
+          namespace = options[:namespace]
+
+          puts 'Upgrading language-operator...'
+          puts "  Namespace: #{namespace}"
+          puts "  Chart: #{CHART_NAME}"
+          puts
+
+          # Update Helm repository
+          update_helm_repo unless options[:dry_run]
+
+          # Build helm upgrade command
+          cmd = build_helm_command('upgrade', namespace)
+
+          # Execute helm upgrade
+          if options[:dry_run]
+            puts 'Dry run - would execute:'
+            puts "  #{cmd}"
+            puts
+            success, output = run_helm_command(cmd)
+            puts output
+          else
+            Formatters::ProgressFormatter.with_spinner('Upgrading language-operator') do
+              success, output = run_helm_command(cmd)
+              raise "Helm upgrade failed: #{output}" unless success
+            end
+
+            Formatters::ProgressFormatter.success('Language operator upgraded successfully!')
+          end
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Upgrade failed: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'uninstall', 'Uninstall the language-operator using Helm'
+        long_desc <<-DESC
+          Uninstall the language-operator from your Kubernetes cluster.
+
+          WARNING: This will remove the operator but NOT the CRDs or custom resources.
+          Agents, tools, models, and personas will remain in the cluster.
+
+          Examples:
+            # Uninstall with confirmation
+            aictl uninstall
+
+            # Force uninstall without confirmation
+            aictl uninstall --force
+
+            # Uninstall from specific namespace
+            aictl uninstall --namespace my-namespace
+        DESC
+        option :namespace, type: :string, default: DEFAULT_NAMESPACE, desc: 'Kubernetes namespace'
+        option :force, type: :boolean, default: false, desc: 'Skip confirmation prompt'
+        def uninstall
+          # Check if helm is available
+          check_helm_installed!
+
+          # Check if operator is installed
+          unless operator_installed?
+            Formatters::ProgressFormatter.warn('Language operator is not installed')
+            return
+          end
+
+          namespace = options[:namespace]
+
+          # Confirm deletion unless --force
+          unless options[:force]
+            puts "This will uninstall the language-operator from namespace '#{namespace}'"
+            puts
+            puts 'WARNING: This will NOT delete:'
+            puts '  - CRDs (CustomResourceDefinitions)'
+            puts '  - LanguageAgent resources'
+            puts '  - LanguageTool resources'
+            puts '  - LanguageModel resources'
+            puts '  - LanguagePersona resources'
+            puts
+            return unless Helpers::UserPrompts.confirm('Continue with uninstall?')
+          end
+
+          # Build helm uninstall command
+          cmd = "helm uninstall #{RELEASE_NAME} --namespace #{namespace}"
+
+          # Execute helm uninstall
+          Formatters::ProgressFormatter.with_spinner('Uninstalling language-operator') do
+            success, output = run_helm_command(cmd)
+            raise "Helm uninstall failed: #{output}" unless success
+          end
+
+          Formatters::ProgressFormatter.success('Language operator uninstalled successfully!')
+          puts
+          puts 'Note: CRDs and custom resources remain in the cluster.'
+          puts 'To completely remove all resources, you must manually delete them.'
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Uninstall failed: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        private
+
+        def check_helm_installed!
+          _, _, status = Open3.capture3('which helm')
+          return if status.success?
+
+          Formatters::ProgressFormatter.error('Helm is not installed')
+          puts
+          puts 'Install Helm from: https://helm.sh/docs/intro/install/'
+          exit 1
+        end
+
+        def operator_installed?
+          namespace = options[:namespace] || DEFAULT_NAMESPACE
+          cmd = "helm list --namespace #{namespace} --filter #{RELEASE_NAME} --output json"
+          success, output = run_helm_command(cmd)
+          return false unless success
+
+          require 'json'
+          releases = JSON.parse(output)
+          releases.any? { |r| r['name'] == RELEASE_NAME }
+        rescue StandardError
+          false
+        end
+
+        def add_helm_repo
+          Formatters::ProgressFormatter.with_spinner('Adding Helm repository') do
+            # Check if repo already exists
+            cmd = 'helm repo list --output json'
+            success, output = run_helm_command(cmd)
+
+            if success
+              require 'json'
+              repos = JSON.parse(output)
+              repo_exists = repos.any? { |r| r['name'] == HELM_REPO_NAME }
+
+              unless repo_exists
+                cmd = "helm repo add #{HELM_REPO_NAME} #{HELM_REPO_URL}"
+                success, output = run_helm_command(cmd)
+                raise "Failed to add Helm repo: #{output}" unless success
+              end
+            else
+              # helm repo list failed, try adding anyway
+              cmd = "helm repo add #{HELM_REPO_NAME} #{HELM_REPO_URL}"
+              success, output = run_helm_command(cmd)
+              raise "Failed to add Helm repo: #{output}" unless success
+            end
+
+            # Update repo
+            cmd = "helm repo update #{HELM_REPO_NAME}"
+            success, output = run_helm_command(cmd)
+            raise "Failed to update Helm repo: #{output}" unless success
+          end
+        end
+
+        def update_helm_repo
+          Formatters::ProgressFormatter.with_spinner('Updating Helm repository') do
+            cmd = "helm repo update #{HELM_REPO_NAME}"
+            success, output = run_helm_command(cmd)
+            raise "Failed to update Helm repo: #{output}" unless success
+          end
+        end
+
+        def build_helm_command(action, namespace)
+          cmd = ['helm', action, RELEASE_NAME]
+
+          # Add chart name for install
+          cmd << CHART_NAME if action == 'install'
+
+          # Add namespace
+          cmd << '--namespace' << namespace
+
+          # Add create-namespace for install
+          cmd << '--create-namespace' if action == 'install' && options[:create_namespace]
+
+          # Add values file
+          cmd << '--values' << options[:values] if options[:values]
+
+          # Add version
+          cmd << '--version' << options[:version] if options[:version]
+
+          # Add wait
+          cmd << '--wait' if options[:wait]
+
+          # Add dry-run
+          cmd << '--dry-run' if options[:dry_run]
+
+          cmd.join(' ')
+        end
+
+        def run_helm_command(cmd)
+          stdout, stderr, status = Open3.capture3(cmd)
+          output = stdout + stderr
+          [status.success?, output.strip]
+        end
+      end
+    end
+  end
+end

data/lib/language_operator/cli/commands/model.rb

@@ -0,0 +1,266 @@
+# frozen_string_literal: true
+
+require 'thor'
+require 'yaml'
+require_relative '../formatters/progress_formatter'
+require_relative '../formatters/table_formatter'
+require_relative '../helpers/cluster_validator'
+require_relative '../helpers/cluster_context'
+require_relative '../helpers/user_prompts'
+require_relative '../helpers/resource_dependency_checker'
+require_relative '../helpers/editor_helper'
+require_relative '../../config/cluster_config'
+require_relative '../../kubernetes/client'
+require_relative '../../kubernetes/resource_builder'
+
+module LanguageOperator
+  module CLI
+    module Commands
+      # Model management commands
+      class Model < Thor
+        include Helpers::ClusterValidator
+
+        desc 'list', 'List all models in current cluster'
+        option :cluster, type: :string, desc: 'Override current cluster context'
+        def list
+          ctx = Helpers::ClusterContext.from_options(options)
+
+          models = ctx.client.list_resources('LanguageModel', namespace: ctx.namespace)
+
+          if models.empty?
+            Formatters::ProgressFormatter.info("No models found in cluster '#{ctx.name}'")
+            puts
+            puts 'Models define LLM configurations for agents.'
+            puts
+            puts 'Create a model with:'
+            puts '  aictl model create <name> --provider <provider> --model <model>'
+            return
+          end
+
+          table_data = models.map do |model|
+            name = model.dig('metadata', 'name')
+            provider = model.dig('spec', 'provider') || 'unknown'
+            model_name = model.dig('spec', 'modelName') || 'unknown'
+            status = model.dig('status', 'phase') || 'Unknown'
+
+            {
+              name: name,
+              provider: provider,
+              model: model_name,
+              status: status
+            }
+          end
+
+          Formatters::TableFormatter.models(table_data)
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Failed to list models: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'create NAME', 'Create a new model'
+        long_desc <<-DESC
+          Create a new LanguageModel resource in the cluster.
+
+          Examples:
+            aictl model create gpt4 --provider openai --model gpt-4-turbo
+            aictl model create claude --provider anthropic --model claude-3-opus-20240229
+            aictl model create local --provider openai_compatible --model llama-3 --endpoint http://localhost:8080
+        DESC
+        option :provider, type: :string, required: true, desc: 'LLM provider (e.g., openai, anthropic, openai_compatible)'
+        option :model, type: :string, required: true, desc: 'Model identifier (e.g., gpt-4, claude-3-opus)'
+        option :endpoint, type: :string, desc: 'Custom endpoint URL (for openai_compatible or self-hosted)'
+        option :cluster, type: :string, desc: 'Override current cluster context'
+        option :dry_run, type: :boolean, default: false, desc: 'Output the manifest without creating'
+        def create(name)
+          ctx = Helpers::ClusterContext.from_options(options)
+
+          # Build LanguageModel resource
+          resource = Kubernetes::ResourceBuilder.language_model(
+            name,
+            provider: options[:provider],
+            model: options[:model],
+            endpoint: options[:endpoint],
+            cluster: ctx.namespace
+          )
+
+          # Handle dry-run: output manifest and exit
+          if options[:dry_run]
+            puts resource.to_yaml
+            return
+          end
+
+          # Check if model already exists
+          begin
+            ctx.client.get_resource('LanguageModel', name, ctx.namespace)
+            Formatters::ProgressFormatter.error("Model '#{name}' already exists in cluster '#{ctx.name}'")
+            exit 1
+          rescue K8s::Error::NotFound
+            # Model doesn't exist, proceed with creation
+          end
+
+          # Create model
+          Formatters::ProgressFormatter.with_spinner("Creating model '#{name}'") do
+            ctx.client.apply_resource(resource)
+          end
+
+          Formatters::ProgressFormatter.success("Model '#{name}' created successfully")
+          puts
+          puts 'Model Details:'
+          puts "  Name:     #{name}"
+          puts "  Provider: #{options[:provider]}"
+          puts "  Model:    #{options[:model]}"
+          puts "  Endpoint: #{options[:endpoint]}" if options[:endpoint]
+          puts "  Cluster:  #{ctx.name}"
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Failed to create model: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'inspect NAME', 'Show detailed model information'
+        option :cluster, type: :string, desc: 'Override current cluster context'
+        def inspect(name)
+          ctx = Helpers::ClusterContext.from_options(options)
+
+          # Get model
+          begin
+            model = ctx.client.get_resource('LanguageModel', name, ctx.namespace)
+          rescue K8s::Error::NotFound
+            Formatters::ProgressFormatter.error("Model '#{name}' not found in cluster '#{ctx.name}'")
+            exit 1
+          end
+
+          puts "Model: #{name}"
+          puts "  Cluster:   #{ctx.name}"
+          puts "  Namespace: #{ctx.namespace}"
+          puts "  Provider:  #{model.dig('spec', 'provider')}"
+          puts "  Model:     #{model.dig('spec', 'modelName')}"
+          puts "  Endpoint:  #{model.dig('spec', 'endpoint')}" if model.dig('spec', 'endpoint')
+          puts "  Status:    #{model.dig('status', 'phase') || 'Unknown'}"
+          puts
+
+          # Get agents using this model
+          agents = ctx.client.list_resources('LanguageAgent', namespace: ctx.namespace)
+          agents_using = Helpers::ResourceDependencyChecker.agents_using_model(agents, name)
+
+          if agents_using.any?
+            puts "Agents using this model (#{agents_using.count}):"
+            agents_using.each do |agent|
+              puts "  - #{agent.dig('metadata', 'name')}"
+            end
+          else
+            puts 'No agents using this model'
+          end
+
+          puts
+          puts 'Labels:'
+          labels = model.dig('metadata', 'labels') || {}
+          if labels.empty?
+            puts '  (none)'
+          else
+            labels.each do |key, value|
+              puts "  #{key}: #{value}"
+            end
+          end
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Failed to inspect model: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'delete NAME', 'Delete a model'
+        option :cluster, type: :string, desc: 'Override current cluster context'
+        option :force, type: :boolean, default: false, desc: 'Skip confirmation'
+        def delete(name)
+          ctx = Helpers::ClusterContext.from_options(options)
+
+          # Get model
+          begin
+            model = ctx.client.get_resource('LanguageModel', name, ctx.namespace)
+          rescue K8s::Error::NotFound
+            Formatters::ProgressFormatter.error("Model '#{name}' not found in cluster '#{ctx.name}'")
+            exit 1
+          end
+
+          # Check for agents using this model
+          agents = ctx.client.list_resources('LanguageAgent', namespace: ctx.namespace)
+          agents_using = Helpers::ResourceDependencyChecker.agents_using_model(agents, name)
+
+          if agents_using.any? && !options[:force]
+            Formatters::ProgressFormatter.warn("Model '#{name}' is in use by #{agents_using.count} agent(s)")
+            puts
+            puts 'Agents using this model:'
+            agents_using.each do |agent|
+              puts "  - #{agent.dig('metadata', 'name')}"
+            end
+            puts
+            puts 'Delete these agents first, or use --force to delete anyway.'
+            puts
+            return unless Helpers::UserPrompts.confirm('Are you sure?')
+          end
+
+          # Confirm deletion unless --force
+          unless options[:force] || agents_using.any?
+            puts "This will delete model '#{name}' from cluster '#{ctx.name}':"
+            puts "  Provider: #{model.dig('spec', 'provider')}"
+            puts "  Model:    #{model.dig('spec', 'modelName')}"
+            puts "  Status:   #{model.dig('status', 'phase')}"
+            puts
+            return unless Helpers::UserPrompts.confirm('Are you sure?')
+          end
+
+          # Delete model
+          Formatters::ProgressFormatter.with_spinner("Deleting model '#{name}'") do
+            ctx.client.delete_resource('LanguageModel', name, ctx.namespace)
+          end
+
+          Formatters::ProgressFormatter.success("Model '#{name}' deleted successfully")
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Failed to delete model: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+
+        desc 'edit NAME', 'Edit model configuration'
+        option :cluster, type: :string, desc: 'Override current cluster context'
+        def edit(name)
+          ctx = Helpers::ClusterContext.from_options(options)
+
+          # Get current model
+          begin
+            model = ctx.client.get_resource('LanguageModel', name, ctx.namespace)
+          rescue K8s::Error::NotFound
+            Formatters::ProgressFormatter.error("Model '#{name}' not found in cluster '#{ctx.name}'")
+            exit 1
+          end
+
+          # Edit model YAML in user's editor
+          edited_yaml = Helpers::EditorHelper.edit_content(
+            model.to_yaml,
+            'model-',
+            '.yaml',
+            default_editor: 'vim'
+          )
+          edited_model = YAML.safe_load(edited_yaml)
+
+          # Apply changes
+          Formatters::ProgressFormatter.with_spinner("Updating model '#{name}'") do
+            ctx.client.apply_resource(edited_model)
+          end
+
+          Formatters::ProgressFormatter.success("Model '#{name}' updated successfully")
+        rescue StandardError => e
+          Formatters::ProgressFormatter.error("Failed to edit model: #{e.message}")
+          raise if ENV['DEBUG']
+
+          exit 1
+        end
+      end
+    end
+  end
+end