lammy 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0992b774797e59eeae152cacb65110b83cec70002e6afb7a40153ae05f942490'
-  data.tar.gz: d2680e0acfc73a28f89e5468a9d4292cd2a4bc9573a5570b9c40b86e3ffa8a14
+  metadata.gz: 9225fc482f95ed862d6e12bc4541731c8854127900e0bfe369075ef1092fae8e
+  data.tar.gz: 720ce8849858edb0832c34b0b10ef736c6353b12daa1e5cecf68c137de94a91d
 SHA512:
-  metadata.gz: ee429f696a2dfcf9f79582db94bdf696c6b4900f903880daddc48d32d9ea13c60a76138b2e47dc47d9c9efb9ed94196af1221827fbcd3c0b9e5b2384d74b2611
-  data.tar.gz: d44d1b6161a805fce9f5da44ac481b909c1edfccd50eb74a61c3498e55642da5782637b1105c07408594de46b1162f5ac7004bdca6bc03caccac4ac48f190f28
+  metadata.gz: 83be8412698ad307dba650aaa54e230cf9b47ee9e140e9dac1d26c2927f28b43649ac9b2e6196154a09b8b608a04b8fb72e7c0792e2a6bd799a2ff1edd339494
+  data.tar.gz: 9add86016f31a13416848ff64a7b9eeea03808b9220b239b029049c1ffc04796a9253f71dfac623928b8eddacbd8ffa182a243fd1a7c50c5a5ade6bdc8a68b50

data/lib/lammy/chat.rb

@@ -28,8 +28,8 @@ module L
 
         case settings[:model]
         when *OpenAI::MODELS
-          client = OpenAI.new
-          client.chat(settings, user_message, @system_message)
+          client = OpenAI.new(settings)
+          client.chat(user_message, @system_message)
         else
           raise "Unsupported model: #{settings[:model]}"
         end

data/lib/lammy/embeddings.rb

@@ -37,8 +37,8 @@ module L
 
         case settings[:model]
         when *OpenAI::EMBEDDINGS
-          client = OpenAI.new
-          client.embeddings(settings, input)
+          client = OpenAI.new(settings)
+          client.embeddings(input)
         else
           raise "Unsupported model: #{settings[:model]}"
         end

data/lib/lammy/openai.rb

@@ -15,8 +15,14 @@ module L
       text-embedding-3-small text-embedding-3-large text-embedding-ada-002
     ].freeze
 
+    attr_reader :settings
+
+    def initialize(settings)
+      @settings = settings
+    end
+
     # Generate a response with support for structured output
-    def chat(settings, user_message, system_message = nil)
+    def chat(user_message, system_message = nil)
       schema = schema(settings)
       response = client.chat(
         parameters: {
@@ -34,7 +40,7 @@ module L
     # OpenAI’s text embeddings measure the relatedness of text strings. An embedding is a vector of floating point
     # numbers. The distance between two vectors measures their relatedness. Small distances suggest high relatedness
     # and large distances suggest low relatedness.
-    def embeddings(settings, chunks)
+    def embeddings(chunks)
       responses = chunks.map do |chunk|
         response = client.embeddings(
           parameters: { model: settings[:model], dimensions: settings[:dimensions], input: chunk }
@@ -65,6 +71,8 @@ module L
     end
 
     def client
+      return settings[:client] if settings[:client]
+
       @client ||= ::OpenAI::Client.new(
         access_token: ENV.fetch('OPENAI_ACCESS_TOKEN')
       )

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lammy
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Kamil Nicieja