rainbow_llm 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b7f668ffcf5f6b67b910f729dd10670908264762798119e3fb0d21b4e27995f
-  data.tar.gz: d0f42404d0d702e8c720b1e5febc8b17fcf02e626d867acd600cbc4ad1fedff0
+  metadata.gz: 691cc8d3f8b1be4eee9736aa9f6ff83a33ea34a21690b7f3b6000584f516c954
+  data.tar.gz: 8c14c1d0b6fd037ca4bd92badad32afaa55543715a440794519e096d09446b86
 SHA512:
-  metadata.gz: fcffac6222cdefa8255eb457913e802621fa262cac2d72ecac56d8e4031b18d71fb345d260911ddebe44b326d5f348ba00f6c33f4cae9df20b3dad662ff686d8
-  data.tar.gz: dc7598c47b9cf858a4e41be22c5b268f8206206f8451f657fc0ef20db25b137e8d51a9fc3019aac44fc184a87ce05bfd40d55954c326cb1d4bb8604372be199b
+  metadata.gz: 4a35bae48e9d8ff273dbbb758206eb423c6897feae868b2e58384a65801aeb56d865977cfe4c1621c03953e317caf28ba8ceeb52a7e64f0a874fa444b869e111
+  data.tar.gz: 943fb9815ff94a90a06b635dd7f2763e73bb1806fcee4f97ce5b0111824ce7c73928233609e5ee54d090bac9fd2cbb368f9f34194215a54ec9f2c787e9c55c56

data/lib/rainbow_llm/chat_builder.rb

@@ -57,17 +57,15 @@ module RainbowLLM
         provider_type = RainbowLLM.configuration.providers.dig(provider.to_sym, :provider) || provider
         assume_model_exists = RainbowLLM.configuration.providers.dig(provider.to_sym, :assume_model_exists)
 
-        base = {model:, provider: provider_type, assume_model_exists:, context:}.compact
-
-        opts = {}
-        opts[:temperature] = temperature if temperature
-        opts[:schema] = schema if schema
+        opts = {model:, provider: provider_type, assume_model_exists:, context:}.compact
 
         result =
           begin
-            chat = RubyLLM.chat(**base.merge(opts))
+            chat = RubyLLM.chat(**opts)
+            chat = chat.with_temperature(temperature) if temperature
+            chat = chat.with_schema(schema) if schema
             response = chat.ask(prompt)
-            content = RainbowLLM.check_content(RainbowLLM.extract_content(response))
+            content = check_content(extract_content(response))
 
             if content
               details[model_string] = {status: :success}
@@ -95,16 +93,18 @@ module RainbowLLM
       # If we get here, all providers failed
       Response.new(content: nil, model: nil, details:)
     end
-  end
 
-  def self.extract_content(response)
-    return nil if response.nil?
+    private
 
-    response.content.strip
-  end
+    def extract_content(response)
+      return nil if response.nil?
 
-  def self.check_content(content)
-    return nil if content.nil? || content.to_s.strip.empty?
-    content
+      response.content.strip
+    end
+
+    def check_content(content)
+      return nil if content.nil? || content.to_s.strip.empty?
+      content
+    end
   end
 end

data/lib/rainbow_llm/response.rb

@@ -1 +1,3 @@
-Response = Data.define(:content, :model, :details)
+module RainbowLLM
+  Response = Data.define(:content, :model, :details)
+end

data/lib/rainbow_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RainbowLLM
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rainbow_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - a-chris