aia 0.9.24 → 0.10.2

data/lib/aia/chat_processor_service.rb

@@ -4,7 +4,7 @@ module AIA
   class ChatProcessorService
     def initialize(ui_presenter, directive_processor = nil)
       @ui_presenter = ui_presenter
-      @speaker = AIA.speak? ? AiClient.new(AIA.config.speech_model) : nil
+      @speaker = AIA.speak? ? AiClient.new(AIA.config.audio.speech_model) : nil
       @directive_processor = directive_processor
     end
 
@@ -12,10 +12,10 @@ module AIA
     def speak(text)
       return unless AIA.speak?
 
-      @speaker ||= AiClient.new(AIA.config.speech_model) if AIA.config.speech_model
+      @speaker ||= AiClient.new(AIA.config.audio.speech_model) if AIA.config.audio.speech_model
 
       if @speaker
-        `#{AIA.config.speak_command} #{@speaker.speak(text).path}`
+        `#{AIA.config.audio.speak_command} #{@speaker.speak(text).path}`
       else
         puts "Warning: Unable to speak. Speech model not configured properly."
       end
@@ -29,15 +29,15 @@ module AIA
       end
 
       # Debug output to understand what we're receiving
-      puts "[DEBUG ChatProcessor] Result class: #{result.class}" if AIA.config.debug
-      puts "[DEBUG ChatProcessor] Result inspect: #{result.inspect[0..500]}..." if AIA.config.debug
+      logger.debug("Result received", result_class: result.class.name)
+      logger.debug("Result details", inspect: result.inspect[0..500])
 
       # Preserve token information if available for metrics
       if result.is_a?(String)
-        puts "[DEBUG ChatProcessor] Processing as String" if AIA.config.debug
+        logger.debug("Processing result", type: "String")
         { content: result, metrics: nil }
       elsif result.respond_to?(:multi_model?) && result.multi_model?
-        puts "[DEBUG ChatProcessor] Processing as multi-model response" if AIA.config.debug
+        logger.debug("Processing result", type: "multi-model response")
         # Handle multi-model response with metrics
         {
           content: result.content,
@@ -45,7 +45,7 @@ module AIA
           multi_metrics: result.metrics_list
         }
       elsif result.respond_to?(:content)
-        puts "[DEBUG ChatProcessor] Processing as standard response with content method" if AIA.config.debug
+        logger.debug("Processing result", type: "standard response with content method")
         # Standard response object with content method
         {
           content: result.content,
@@ -56,7 +56,7 @@ module AIA
           }
         }
       else
-        puts "[DEBUG ChatProcessor] Processing as fallback (unexpected type)" if AIA.config.debug
+        logger.debug("Processing result", type: "fallback (unexpected type)")
         # Fallback for unexpected response types
         { content: result.to_s, metrics: nil }
       end
@@ -71,15 +71,15 @@ module AIA
       # Handle per-model conversations (Hash) or single conversation (Array) - ADR-002 revised
       if conversation_or_conversations.is_a?(Hash)
         # Multi-model with per-model contexts: pass Hash directly to adapter
-        puts "[DEBUG ChatProcessor] Sending per-model conversations to client" if AIA.config.debug
+        logger.debug("Sending per-model conversations to client")
         result = AIA.client.chat(conversation_or_conversations)
       else
         # Single conversation for single model
-        puts "[DEBUG ChatProcessor] Sending conversation to client: #{conversation_or_conversations.inspect[0..500]}..." if AIA.config.debug
+        logger.debug("Sending conversation to client", conversation: conversation_or_conversations.inspect[0..500])
         result = AIA.client.chat(conversation_or_conversations)
       end
 
-      puts "[DEBUG ChatProcessor] Client returned: #{result.class} - #{result.inspect[0..500]}..." if AIA.config.debug
+      logger.debug("Client returned", result_class: result.class.name, result: result.inspect[0..500])
       result
     end
 
@@ -88,11 +88,17 @@ module AIA
       # With multiple models, we don't need to change the model in the same way
       # The RubyLLMAdapter now handles multiple models internally
       # This method is kept for backward compatibility but may not be needed
-      return if AIA.config.model.is_a?(Array)
+      models = AIA.config.models
+      return if models.is_a?(Array) && models.size > 1
 
-      client_model = AIA.client.model.id  # RubyLLM::Model instance
+      return unless AIA.client.respond_to?(:model) && AIA.client.model.respond_to?(:id)
+      client_model = AIA.client.model.id
 
-      unless AIA.config.model.downcase.include?(client_model.downcase)
+      # Get the first model name for comparison
+      first_model = models.first
+      model_name = first_model.respond_to?(:name) ? first_model.name : first_model.to_s
+
+      unless model_name.downcase.include?(client_model.downcase)
         AIA.client = AIA.client.class.new
       end
     end
@@ -101,13 +107,15 @@ module AIA
     def output_response(response)
       speak(response)
 
+      out_file = AIA.config.output.file
+
       # Output to STDOUT or file based on out_file configuration
-      if AIA.config.out_file.nil? || 'STDOUT' == AIA.config.out_file.upcase
+      if out_file.nil? || 'STDOUT' == out_file.upcase
         print "\nAI:\n  "
         puts response
       else
         mode = AIA.append? ? 'a' : 'w'
-        File.open(AIA.config.out_file, mode) do |file|
+        File.open(out_file, mode) do |file|
           file.puts "\nAI: "
           # Handle multi-line responses by adding proper indentation
           response_lines = response.to_s.split("\n")
@@ -117,8 +125,9 @@ module AIA
         end
       end
 
-      if AIA.config.log_file
-        File.open(AIA.config.log_file, 'a') do |f|
+      history_file = AIA.config.output.history_file
+      if history_file
+        File.open(history_file, 'a') do |f|
           f.puts "=== #{Time.now} ==="
           f.puts "Prompt: #{AIA.config.prompt_id}"
           f.puts "Response: #{response}"
@@ -140,10 +149,11 @@ module AIA
     def determine_operation_type
       # With multiple models, determine operation type from the first model
       # or provide a generic description
-      if AIA.config.model.is_a?(Array) && AIA.config.model.size > 1
+      models = AIA.config.models
+      if models.is_a?(Array) && models.size > 1
         "MULTI-MODEL PROCESSING"
       else
-        mode = AIA.config.client.model.modalities
+        mode = AIA.client.model.modalities
         mode.input.join(',') + " TO " + mode.output.join(',')
       end
     end