module L::Chat

def handle_llm(method_name)
  settings = @next_llm_settings
  @next_llm_settings = nil
  # Unbind the original method
  original_method = instance_method(method_name)
  # Redefine the method
  define_method(method_name) do |*args, &block|
    # Initialize context
    @system_message = nil
    # Make `context` method available within the instance
    define_singleton_method(:context) do |message|
      @system_message = message
    end
    # Call the original method to get the user message
    user_message = original_method.bind(self).call(*args, &block)
    case settings[:model]
    when *OpenAI::MODELS
      client = OpenAI.new(settings)
      client.chat(user_message, @system_message)
    else
      raise "Unsupported model: #{settings[:model]}"
    end
  end
end

def llm(**kwargs)
  @next_llm_settings = kwargs
end