igniter 0.5.0 → 0.5.1

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/executor.rb

@@ -3,7 +3,7 @@
 require "json"
 
 module Igniter
-  module LLM
+  module AI
     # Raised when the tool-use loop exceeds max_tool_iterations.
     class ToolLoopError < Error; end
 
@@ -14,7 +14,7 @@ module Igniter
     #
     # == Simple usage (no tools)
     #
-    #   class DocumentSummarizer < Igniter::LLM::Executor
+    #   class DocumentSummarizer < Igniter::AI::Executor
     #     provider :ollama
     #     model "llama3.2"
     #     system_prompt "You are a concise technical writer."
@@ -26,7 +26,7 @@ module Igniter
     #
     # == With Igniter::Tool classes (auto tool-use loop)
     #
-    #   class ResearchAgent < Igniter::LLM::Executor
+    #   class ResearchAgent < Igniter::AI::Executor
     #     provider :anthropic
     #     model "claude-sonnet-4-6"
     #     system_prompt "You are a research assistant. Use tools when needed."
@@ -43,7 +43,7 @@ module Igniter
     #
     # == Provider failover
     #
-    #   class MyAgent < Igniter::LLM::Executor
+    #   class MyAgent < Igniter::AI::Executor
     #     provider :anthropic, fallback: [:openai, :ollama]
     #     model "claude-sonnet-4-6", fallback: ["gpt-4o", "llama3.2"]
     #     # On ProviderError, retries with OpenAI/GPT-4o, then Ollama/llama3.2
@@ -56,7 +56,7 @@ module Igniter
         # @param fallback [Array<Symbol>] providers to try on ProviderError, in order
         def provider(name = nil, fallback: nil)
           if name.nil?
-            @provider_chain&.first || Igniter::LLM.config.default_provider
+            @provider_chain&.first || Igniter::AI.config.default_provider
           else
             chain = [name] + Array(fallback)
             @provider_chain = chain.map(&:to_sym)
@@ -76,7 +76,7 @@ module Igniter
         def model(name = nil, fallback: nil)
           if name.nil?
             @model_chain&.first || default_model_for(
-              @provider_chain&.first || Igniter::LLM.config.default_provider
+              @provider_chain&.first || Igniter::AI.config.default_provider
             )
           else
             @model_chain = [name] + Array(fallback)
@@ -132,15 +132,15 @@ module Igniter
         private
 
         def default_model_for(prov)
-          Igniter::LLM.config.provider_config(prov).default_model
+          Igniter::AI.config.provider_config(prov).default_model
         rescue StandardError
           "llama3.2"
         end
 
         def provider_config
-          Igniter::LLM.provider_instance(provider).instance_of?(Class) ? nil : Igniter::LLM.config.provider_config(provider)
+          Igniter::AI.provider_instance(provider).instance_of?(Class) ? nil : Igniter::AI.config.provider_config(provider)
         rescue StandardError
-          Igniter::LLM.config.ollama
+          Igniter::AI.config.ollama
         end
       end
 
@@ -165,9 +165,9 @@ module Igniter
       # @return [String] final LLM text response
       def complete(prompt, context: nil) # rubocop:disable Metrics/MethodLength,Metrics/AbcSize
         with_provider_fallback do
-          # Accept both Igniter::Tool and Igniter::Skill subclasses (duck-type check).
+          # Accept both Igniter::Tool and Igniter::AI::Skill subclasses (duck-type check).
           # Using respond_to? avoids a circular require: skill.rb requires this file,
-          # so we can't safely reference Igniter::Skill or Igniter::Tool::Discoverable here.
+          # so we can't safely reference Igniter::AI::Skill or Igniter::Tool::Discoverable here.
           tool_classes = self.class.tools.select do |t|
             t.is_a?(Class) && t.respond_to?(:tool_name) && t.respond_to?(:to_schema)
           end
@@ -228,7 +228,7 @@ module Igniter
       # Iterate through the provider/model chain, retrying on ProviderError.
       # ConfigurationError propagates immediately (missing API key = config bug).
       #
-      # Does NOT call Igniter::LLM.provider_instance directly; instead it resets
+      # Does NOT call Igniter::AI.provider_instance directly; instead it resets
       # @provider_instance = nil before each attempt so that the #provider_instance
       # accessor re-evaluates (or a test's define_method override takes effect).
       def with_provider_fallback # rubocop:disable Metrics/MethodLength
@@ -243,7 +243,7 @@ module Igniter
 
           begin
             return yield
-          rescue Igniter::LLM::ProviderError => e
+          rescue Igniter::AI::ProviderError => e
             last_error = e
           end
         end
@@ -317,7 +317,7 @@ module Igniter
       end
 
       def provider_instance
-        @provider_instance ||= Igniter::LLM.provider_instance(@last_provider || self.class.provider)
+        @provider_instance ||= Igniter::AI.provider_instance(@last_provider || self.class.provider)
       end
 
       def build_messages(prompt:, context: nil)

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/providers/anthropic.rb

@@ -5,7 +5,7 @@ require "json"
 require "uri"
 
 module Igniter
-  module LLM
+  module AI
     module Providers
       # Anthropic Claude provider.
       # Requires ANTHROPIC_API_KEY environment variable or explicit api_key:.
@@ -142,7 +142,7 @@ module Igniter
           response = http.request(request)
           handle_response(response)
         rescue Errno::ECONNREFUSED, Errno::EADDRNOTAVAIL, SocketError, Net::OpenTimeout => e
-          raise Igniter::LLM::ProviderError, "Cannot connect to Anthropic API: #{e.message}"
+          raise Igniter::AI::ProviderError, "Cannot connect to Anthropic API: #{e.message}"
         end
 
         def headers
@@ -161,18 +161,18 @@ module Igniter
               {}
             end
             error_msg = body.dig("error", "message") || response.body.to_s.slice(0, 200)
-            raise Igniter::LLM::ProviderError, "Anthropic API error #{response.code}: #{error_msg}"
+            raise Igniter::AI::ProviderError, "Anthropic API error #{response.code}: #{error_msg}"
           end
 
           JSON.parse(response.body)
         rescue JSON::ParserError => e
-          raise Igniter::LLM::ProviderError, "Anthropic returned invalid JSON: #{e.message}"
+          raise Igniter::AI::ProviderError, "Anthropic returned invalid JSON: #{e.message}"
         end
 
         def validate_api_key!
           return if @api_key && !@api_key.empty?
 
-          raise Igniter::LLM::ConfigurationError,
+          raise Igniter::AI::ConfigurationError,
                 "Anthropic API key not configured. Set ANTHROPIC_API_KEY or pass api_key: to the provider."
         end
       end

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/providers/base.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Igniter
-  module LLM
+  module AI
     module Providers
       class Base
         attr_reader :last_usage

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/providers/ollama.rb

@@ -5,7 +5,7 @@ require "json"
 require "uri"
 
 module Igniter
-  module LLM
+  module AI
     module Providers
       # Ollama provider — calls the local Ollama REST API.
       # Requires Ollama to be running: https://ollama.com
@@ -63,7 +63,7 @@ module Igniter
           response = http.request(request)
           handle_response(response)
         rescue Errno::ECONNREFUSED, Errno::EADDRNOTAVAIL, SocketError, Net::OpenTimeout => e
-          raise Igniter::LLM::ProviderError, "Cannot connect to Ollama at #{@base_url}: #{e.message}"
+          raise Igniter::AI::ProviderError, "Cannot connect to Ollama at #{@base_url}: #{e.message}"
         end
 
         def get(path)
@@ -76,13 +76,13 @@ module Igniter
 
         def handle_response(response)
           unless response.is_a?(Net::HTTPSuccess)
-            raise Igniter::LLM::ProviderError,
+            raise Igniter::AI::ProviderError,
                   "Ollama API error #{response.code}: #{response.body.to_s.slice(0, 200)}"
           end
 
           JSON.parse(response.body)
         rescue JSON::ParserError => e
-          raise Igniter::LLM::ProviderError, "Ollama returned invalid JSON: #{e.message}"
+          raise Igniter::AI::ProviderError, "Ollama returned invalid JSON: #{e.message}"
         end
 
         def normalize_messages(messages)

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/providers/openai.rb

@@ -5,7 +5,7 @@ require "json"
 require "uri"
 
 module Igniter
-  module LLM
+  module AI
     module Providers
       # OpenAI provider (also compatible with Azure OpenAI and any OpenAI-compatible API).
       # Requires OPENAI_API_KEY environment variable or explicit api_key:.
@@ -144,7 +144,7 @@ module Igniter
           response = http.request(request)
           handle_response(response)
         rescue Errno::ECONNREFUSED, Errno::EADDRNOTAVAIL, SocketError, Net::OpenTimeout => e
-          raise Igniter::LLM::ProviderError, "Cannot connect to OpenAI API at #{@base_url}: #{e.message}"
+          raise Igniter::AI::ProviderError, "Cannot connect to OpenAI API at #{@base_url}: #{e.message}"
         end
 
         def headers
@@ -162,18 +162,18 @@ module Igniter
               {}
             end
             error_msg = body.dig("error", "message") || response.body.to_s.slice(0, 200)
-            raise Igniter::LLM::ProviderError, "OpenAI API error #{response.code}: #{error_msg}"
+            raise Igniter::AI::ProviderError, "OpenAI API error #{response.code}: #{error_msg}"
           end
 
           JSON.parse(response.body)
         rescue JSON::ParserError => e
-          raise Igniter::LLM::ProviderError, "OpenAI returned invalid JSON: #{e.message}"
+          raise Igniter::AI::ProviderError, "OpenAI returned invalid JSON: #{e.message}"
         end
 
         def validate_api_key!
           return if @api_key && !@api_key.empty?
 
-          raise Igniter::LLM::ConfigurationError,
+          raise Igniter::AI::ConfigurationError,
                 "OpenAI API key not configured. Set OPENAI_API_KEY or pass api_key: to the provider."
         end
       end

data/{lib/igniter → packages/igniter-ai/lib/igniter/ai}/skill/feedback.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
 module Igniter
-  class Skill
+  module AI
     # Immutable record of one rated invocation of a Skill.
-    FeedbackEntry = Struct.new(
+    Skill::FeedbackEntry = Struct.new(
       :input,      # String — the prompt that was sent to the LLM
       :output,     # String — the response that was returned
       :rating,     # Symbol — :good, :bad, or :neutral
@@ -18,7 +18,7 @@ module Igniter
     end
 
     # Thread-safe in-memory store for feedback entries.
-    module FeedbackStore
+    module Skill::FeedbackStore
       class Memory
         MAX_SIZE = 500
 
@@ -69,7 +69,7 @@ module Igniter
     #   improved = skill.refine_system_prompt
     #   # Inspect it, then use it:
     #   MySkill.system_prompt improved
-    class FeedbackRefiner
+    class Skill::FeedbackRefiner
       TEMPLATE = <<~PROMPT
         You are improving a system prompt for an AI skill based on user feedback.
         Return ONLY the improved system prompt text, with no explanation or preamble.

data/{lib/igniter → packages/igniter-ai/lib/igniter/ai}/skill/output_schema.rb

@@ -3,7 +3,7 @@
 require "json"
 
 module Igniter
-  class Skill
+  module AI
     # DSL for declaring a typed JSON output contract on a Skill.
     #
     # When an +output_schema+ block is given, the Skill automatically:
@@ -13,7 +13,7 @@ module Igniter
     #
     # == Definition
     #
-    #   class AnalysisSkill < Igniter::Skill
+    #   class AnalysisSkill < Igniter::AI::Skill
     #     output_schema do
     #       field :summary,    String
     #       field :confidence, Float
@@ -27,7 +27,7 @@ module Igniter
     #   result.summary     # => "This document covers..."
     #   result.confidence  # => 0.91
     #   result.to_h        # => { summary: "...", confidence: 0.91, sources: [...] }
-    class OutputSchema
+    class Skill::OutputSchema
       # Raised when the LLM response cannot be parsed into the declared schema.
       class ParseError < Igniter::Error; end
 
@@ -78,7 +78,7 @@ module Igniter
         raise ParseError, "No JSON object found in LLM response" unless json_str
 
         data = JSON.parse(json_str)
-        StructuredResult.new(@fields, data)
+        Skill::StructuredResult.new(@fields, data)
       rescue JSON::ParserError => e
         raise ParseError, "Invalid JSON in LLM response: #{e.message}"
       end
@@ -86,7 +86,7 @@ module Igniter
 
     # A typed result object returned by skills with an +output_schema+.
     # Provides reader methods for each declared field plus +to_h+ / +to_json+.
-    class StructuredResult
+    class Skill::StructuredResult
       def initialize(fields, data)
         @fields = fields
         @data   = data.transform_keys(&:to_sym)
@@ -103,7 +103,7 @@ module Igniter
       end
 
       def inspect
-        "#<Igniter::Skill::StructuredResult #{to_h}>"
+        "#<Igniter::AI::Skill::StructuredResult #{to_h}>"
       end
     end
   end

data/packages/igniter-ai/lib/igniter/ai/skill/runtime_contract.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+module Igniter
+  module AI
+    class Skill < Executor
+      # Canonical runtime-facing contract for AI skills.
+      #
+      # This gathers the skill semantics that previously lived across separate
+      # class-level methods into one immutable object:
+      # - structured output contract
+      # - feedback/refinement behavior
+      # - capability requirements
+      # - nested tool/skill affordances
+      class RuntimeContract
+        attr_reader :output_schema, :feedback_enabled, :feedback_store,
+                    :required_capabilities, :tools
+
+        def initialize(output_schema:, feedback_enabled:, feedback_store:, required_capabilities:, tools:)
+          @output_schema = output_schema
+          @feedback_enabled = !!feedback_enabled
+          @feedback_store = feedback_store
+          @required_capabilities = Array(required_capabilities).map(&:to_sym).freeze
+          @tools = Array(tools).freeze
+          @tool_classes = @tools.select do |tool|
+            tool.is_a?(Class) && tool.respond_to?(:tool_name) && tool.respond_to?(:to_schema)
+          end.freeze
+          @tool_names = @tool_classes.map(&:tool_name).freeze
+          freeze
+        end
+
+        def structured_output?
+          output_schema.is_a?(Skill::OutputSchema)
+        end
+
+        def feedback?
+          feedback_enabled
+        end
+
+        def tool_classes
+          @tool_classes
+        end
+
+        def tool_names
+          @tool_names
+        end
+
+        def tool_schemas(provider = nil)
+          tool_classes.map { |tool| tool.to_schema(provider) }
+        end
+
+        def to_h
+          {
+            structured_output: structured_output?,
+            output_schema: serialize_output_schema,
+            feedback_enabled: feedback_enabled,
+            feedback_store: serialize_feedback_store,
+            required_capabilities: required_capabilities,
+            tool_names: tool_names,
+            tool_count: tool_classes.size,
+          }.freeze
+        end
+
+        private
+
+        def serialize_output_schema
+          if structured_output?
+            {
+              type: "structured",
+              fields: output_schema.fields.map { |field| { name: field.name, type: field.type.name } },
+            }
+          else
+            output_schema
+          end
+        end
+
+        def serialize_feedback_store
+          return nil unless feedback_store
+
+          {
+            class_name: feedback_store.class.name,
+            size: feedback_store.respond_to?(:size) ? feedback_store.size : nil,
+          }.freeze
+        end
+      end
+    end
+  end
+end

data/packages/igniter-ai/lib/igniter/ai/skill.rb

@@ -0,0 +1,107 @@
+# frozen_string_literal: true
+
+require "igniter/core/errors"
+require "igniter/core/tool"
+require_relative "executor"
+
+module Igniter
+  module AI
+    # Base class for AI-callable skills — composable units of agent capability.
+    class Skill < Executor
+      CapabilityError = Igniter::Tool::CapabilityError
+
+      include Igniter::Tool::Discoverable
+
+      class << self
+        def inherited(subclass)
+          super
+          subclass.instance_variable_set(:@output_schema, @output_schema)
+          subclass.instance_variable_set(:@feedback_enabled, @feedback_enabled)
+          copy_discoverable_state_to(subclass)
+        end
+
+        def output_schema(value = nil, &block)
+          if block
+            @output_schema = Skill::OutputSchema.new(&block)
+          elsif value
+            super(value)
+          else
+            @output_schema || super
+          end
+        end
+
+        def feedback_enabled(val = nil)
+          val.nil? ? (@feedback_enabled || false) : (@feedback_enabled = val)
+        end
+
+        def feedback_store(val = nil)
+          return @feedback_store if val.nil?
+
+          @feedback_store = val == :memory ? Skill::FeedbackStore::Memory.new : val
+        end
+
+        def runtime_contract
+          Skill::RuntimeContract.new(
+            output_schema: output_schema,
+            feedback_enabled: feedback_enabled,
+            feedback_store: feedback_store,
+            required_capabilities: required_capabilities,
+            tools: tools
+          )
+        end
+      end
+
+      protected
+
+      def complete(prompt, context: nil)
+        contract = self.class.runtime_contract
+        schema = contract.output_schema
+
+        adjusted = if schema.is_a?(Skill::OutputSchema)
+                     "#{prompt}\n\nRespond ONLY with valid JSON matching this schema: #{schema.to_json_description}"
+                   else
+                     prompt
+                   end
+
+        result = super(adjusted, context: context)
+        schema.is_a?(Skill::OutputSchema) ? schema.parse(result) : result
+      end
+
+      public
+
+      def feedback(output, rating:, notes: nil) # rubocop:disable Metrics/MethodLength
+        contract = self.class.runtime_contract
+        return self unless contract.feedback?
+
+        store = contract.feedback_store
+        return self unless store
+
+        output_str = output.to_s
+        matched    = (call_history || []).reverse.find { |h| h[:output] == output_str }
+
+        store.store(FeedbackEntry.new(
+                      input: matched&.dig(:input),
+                      output: output_str,
+                      rating: rating.to_sym,
+                      notes: notes,
+                      timestamp: Time.now
+                    ))
+        self
+      end
+
+      def refine_system_prompt
+        store = self.class.feedback_store
+        raise Igniter::Error, "No feedback_store configured on #{self.class.name}" unless store
+
+        FeedbackRefiner.new(provider_instance, current_model).refine(
+          self.class.system_prompt.to_s,
+          store.all
+        )
+      end
+    end
+  end
+end
+
+require_relative "skill/output_schema"
+require_relative "skill/feedback"
+require_relative "skill/runtime_contract"

data/packages/igniter-ai/lib/igniter/ai/tool_registry.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require "set"
+
+require "igniter/core/errors"
+require "igniter/core/tool"
+require_relative "skill"
+
+module Igniter
+  module AI
+    module ToolRegistry
+      SCOPES = %i[bundled managed workspace].freeze
+
+      @tools = {}
+
+      class << self
+        def register(*tool_classes, scope: :workspace) # rubocop:disable Metrics/MethodLength
+          unless SCOPES.include?(scope)
+            raise ArgumentError, "Invalid scope #{scope.inspect}. Must be one of #{SCOPES.inspect}"
+          end
+
+          tool_classes.flatten.each do |klass|
+            unless discoverable?(klass)
+              raise ArgumentError,
+                    "#{klass.inspect} must be an Igniter::Tool or Igniter::AI::Skill subclass"
+            end
+
+            @tools[klass.tool_name] = { klass: klass, scope: scope }
+          end
+          self
+        end
+
+        def find(name)
+          @tools[name.to_s]&.fetch(:klass)
+        end
+
+        def all(scope: nil)
+          entries_for(scope).map { |e| e[:klass] }
+        end
+
+        def tools_for(capabilities: [], scope: nil)
+          allowed  = capabilities.map(&:to_sym).to_set
+          matching = entries_for(scope).select do |e|
+            e[:klass].required_capabilities.all? { |c| allowed.include?(c) }
+          end
+          matching.map { |e| e[:klass] }
+        end
+
+        def schemas(provider = nil, capabilities: nil, scope: nil)
+          list = if capabilities
+                   tools_for(capabilities: capabilities, scope: scope)
+                 else
+                   all(scope: scope)
+                 end
+          list.map { |klass| klass.to_schema(provider) }
+        end
+
+        def size = @tools.size
+        def empty? = @tools.empty?
+
+        def clear!
+          @tools = {}
+          self
+        end
+
+        private
+
+        def entries_for(scope)
+          scope ? @tools.values.select { |e| e[:scope] == scope } : @tools.values
+        end
+
+        def discoverable?(klass)
+          klass.is_a?(Class) &&
+            (klass < Igniter::Tool || klass < Igniter::AI::Skill)
+        end
+      end
+    end
+  end
+end

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/transcription/providers/assemblyai.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Igniter
-  module LLM
+  module AI
     module Transcription
       module Providers
         # AssemblyAI transcription provider.
@@ -70,9 +70,9 @@ module Igniter
             request.body            = audio_data
 
             response = handle_response(http.request(request))
-            response["upload_url"] || raise(Igniter::LLM::ProviderError, "AssemblyAI: no upload_url in response")
+            response["upload_url"] || raise(Igniter::AI::ProviderError, "AssemblyAI: no upload_url in response")
           rescue Errno::ECONNREFUSED, SocketError, Net::OpenTimeout => e
-            raise Igniter::LLM::ProviderError, "Cannot connect to AssemblyAI API: #{e.message}"
+            raise Igniter::AI::ProviderError, "Cannot connect to AssemblyAI API: #{e.message}"
           end
 
           # ── Step 2: Submit job ─────────────────────────────────────────────
@@ -99,7 +99,7 @@ module Igniter
             request.body = JSON.generate(body)
 
             response = handle_response(http.request(request))
-            response["id"] || raise(Igniter::LLM::ProviderError, "AssemblyAI: no transcript id in response")
+            response["id"] || raise(Igniter::AI::ProviderError, "AssemblyAI: no transcript id in response")
           end
 
           # ── Step 3: Poll ───────────────────────────────────────────────────
@@ -114,12 +114,12 @@ module Igniter
               case result["status"]
               when "completed" then return result
               when "error"
-                raise Igniter::LLM::ProviderError,
+                raise Igniter::AI::ProviderError,
                       "AssemblyAI transcription failed: #{result["error"]}"
               end
 
               if Time.now > deadline
-                raise Igniter::LLM::ProviderError,
+                raise Igniter::AI::ProviderError,
                       "AssemblyAI transcription timed out after #{timeout}s (job_id: #{job_id})"
               end
 
@@ -134,7 +134,7 @@ module Igniter
             request  = Net::HTTP::Get.new(uri.path, auth_headers)
             handle_response(http.request(request))
           rescue Errno::ECONNREFUSED, SocketError, Net::OpenTimeout => e
-            raise Igniter::LLM::ProviderError, "Cannot connect to AssemblyAI API: #{e.message}"
+            raise Igniter::AI::ProviderError, "Cannot connect to AssemblyAI API: #{e.message}"
           end
 
           # ── Result building ────────────────────────────────────────────────
@@ -190,7 +190,7 @@ module Igniter
           def validate_api_key!
             return if @api_key && !@api_key.empty?
 
-            raise Igniter::LLM::ConfigurationError,
+            raise Igniter::AI::ConfigurationError,
                   "AssemblyAI API key not configured. Set ASSEMBLYAI_API_KEY or pass api_key: to the provider."
           end
         end

data/{lib/igniter/integrations/llm → packages/igniter-ai/lib/igniter/ai}/transcription/providers/base.rb

@@ -5,7 +5,7 @@ require "json"
 require "uri"
 
 module Igniter
-  module LLM
+  module AI
     module Transcription
       module Providers
         # Abstract base for all transcription providers.
@@ -105,11 +105,11 @@ module Igniter
               end
               msg = body["error"].is_a?(Hash) ? body.dig("error", "message") : body["error"]
               msg ||= response.body.to_s.slice(0, 200)
-              raise Igniter::LLM::ProviderError, "#{provider_name} error #{response.code}: #{msg}"
+              raise Igniter::AI::ProviderError, "#{provider_name} error #{response.code}: #{msg}"
             end
             JSON.parse(response.body)
           rescue JSON::ParserError => e
-            raise Igniter::LLM::ProviderError, "#{provider_name} returned invalid JSON: #{e.message}"
+            raise Igniter::AI::ProviderError, "#{provider_name} returned invalid JSON: #{e.message}"
           end
 
           def provider_name