lammy 0.2.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0992b774797e59eeae152cacb65110b83cec70002e6afb7a40153ae05f942490'
-  data.tar.gz: d2680e0acfc73a28f89e5468a9d4292cd2a4bc9573a5570b9c40b86e3ffa8a14
+  metadata.gz: bab0a835c9ce12e44b4c5f0d4252eb3040e0d5cb3336aeefe24856f59a74a2a9
+  data.tar.gz: a627a3d18b19a86785a1c70d95304b5bc8ab820766faf15f7893a79d563aeb44
 SHA512:
-  metadata.gz: ee429f696a2dfcf9f79582db94bdf696c6b4900f903880daddc48d32d9ea13c60a76138b2e47dc47d9c9efb9ed94196af1221827fbcd3c0b9e5b2384d74b2611
-  data.tar.gz: d44d1b6161a805fce9f5da44ac481b909c1edfccd50eb74a61c3498e55642da5782637b1105c07408594de46b1162f5ac7004bdca6bc03caccac4ac48f190f28
+  metadata.gz: 0cd66977d719fb4bf752b96c9d5fcaeb1d776fda9d3f929366b4ac41a9913f2e16de6cb1454f0116214f1a7b404b815506f3e0375ac781f5ef2304aa01e54b6c
+  data.tar.gz: 8c39fea57e23ce2c39a926b039c01e132482237a5c9e6af4fd1bd9de20b884845f05930be747a202f5a3982fe22bbf2f3e308f9f605d934c9897b57b095c72c0

data/lib/lammy/chat.rb

@@ -26,13 +26,16 @@ module L
         # 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
-          client.chat(settings, user_message, @system_message)
-        else
-          raise "Unsupported model: #{settings[:model]}"
-        end
+        client = case settings[:model]
+                 when *OpenAI::MODELS
+                   OpenAI.new(settings)
+                 when *Claude::MODELS
+                   Claude.new(settings)
+                 else
+                   raise "Unsupported model: #{settings[:model]}"
+                 end
+
+        client.chat(user_message, @system_message)
       end
     end
   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

@@ -4,7 +4,7 @@ require 'openai'
 require 'hashie'
 
 module L
-  # Use the OpenAI API's official Ruby library
+  # Use the OpenAI API's Ruby library
   class OpenAI
     MODELS = %w[
       gpt-4o-mini gpt-4o gpt-4-turbo gpt-4 gpt-3.5-turbo gpt-4o-mini-2024-07-18 gpt-4o-2024-08-06
@@ -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')
       )

data/lib/lammy.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'lammy/embeddings'
+require 'lammy/claude'
 require 'lammy/openai'
 require 'lammy/schema'
 require 'lammy/chat'

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: lammy
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Kamil Nicieja
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-01 00:00:00.000000000 Z
+date: 2024-10-03 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
 - !ruby/object:Gem::Dependency
   name: hashie
   requirement: !ruby/object:Gem::Requirement
@@ -61,7 +75,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.6.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="