lammy 0.8.3 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d44c524678b02c6e7bb07135102e42750ff70786f2941d7768f2d1598b7a8f98
-  data.tar.gz: 8d94092fc72aecf3d1b2bad51f864785a99761448ea423f68bc82437fe8ed270
+  metadata.gz: 2ff43d19861d0ae98debcecac11133a2abb26a9a7cd44636b63ba5195bf6984f
+  data.tar.gz: 0dbe0ed8665a34ce50cc6c0862d77f926b298f353035018ebdb27dbe1e7d3674
 SHA512:
-  metadata.gz: 0ef97bb570e1879206760211b0d7f2c76382281f73b6f3d2bd38801cf94348e22837dc79465d9b55c352ae7f917e3bdb4e33db5eb8057237f20ed206f25a9dc7
-  data.tar.gz: 6db191e6da9b0b7f762d65e194fee4484b86965ebb74d6e6e67957cdd06d9a90d6162df86d8727bdf67754f3b1d3fedce1ff186956fe7fc35c8ef98245a7ae42
+  metadata.gz: 07e4aaae22128eb3e16447c136bab320c5887649611cb8acc5f29709c53841c52442040fd8a21609d2dd6df19e5565950f618334ca6fe1cb2515f71d25d09202
+  data.tar.gz: 3b5352026f96575f874187b02f02ec57795748409853fb59e52df490223e05563ea5f53530d2c83e1930b4666601da9382c39f83bd2e43cd071c74aae52f93d0

data/lib/lammy/chat.rb

@@ -23,14 +23,14 @@ module Lammy
         # Call the original method to get the user message
         user_message = original_method.bind(self).call(*args, &block)
 
-        client = case settings[:model]
-                 when *OpenAI::MODELS
-                   OpenAI.new(settings)
-                 when *Claude::MODELS
-                   Claude.new(settings)
-                 else
-                   raise "Unsupported model: #{settings[:model]}"
-                 end
+        model = settings[:model] || Lammy.configuration.model
+        client = if OpenAI::MODELS.any? { |regex| model =~ regex }
+                  OpenAI.new(settings)
+                elsif Claude::MODELS.any? { |regex| model =~ regex }
+                  Claude.new(settings)
+                else
+                  raise "Unsupported model: #{model}"
+                end
 
         client.chat(user_message, @system_message, @stream)
       end

data/lib/lammy/claude.rb

@@ -6,11 +6,18 @@ require 'hashie'
 module Lammy
   # Use the Claude API's Ruby library
   class Claude
-    MODELS = %w[
-      claude-3-5-sonnet-latest claude-3-5-haiku-latest claude-3-opus-latest
-      claude-3-5-sonnet-20240620 claude-3-5-haiku-20241022
-      claude-3-opus-20240229 claude-3-sonnet-20240229 claude-3-haiku-20240307
-      claude-2.1 claude-2.0 claude-instant-1.2
+    MODELS = [
+      /\Aclaude-3-5-sonnet-latest\z/,
+      /\Aclaude-3-5-haiku-latest\z/,
+      /\Aclaude-3-opus-latest\z/,
+      /\Aclaude-3-5-sonnet-20240620\z/,
+      /\Aclaude-3-5-haiku-20241022\z/,
+      /\Aclaude-3-opus-20240229\z/,
+      /\Aclaude-3-sonnet-20240229\z/,
+      /\Aclaude-3-haiku-20240307\z/,
+      /\Aclaude-2\.1\z/,
+      /\Aclaude-2\.0\z/,
+      /\Aclaude-instant-1\.2\z/
     ].freeze
 
     attr_reader :settings
@@ -24,7 +31,7 @@ module Lammy
       response = client.messages(
         parameters: {
           system: system_message,
-          model: settings[:model],
+          model: settings[:model] || Lammy.configuration.model,
           max_tokens: settings[:max_tokens] || 4096,
           stream: stream ? ->(chunk) { stream.call(stream_content(chunk)) } : nil,
           messages: user_message.is_a?(Array) ? user_message : [vision(L.user(user_message))]
@@ -57,6 +64,7 @@ module Lammy
 
     def client
       return settings[:client] if settings[:client]
+      return Lammy.configuration.client if Lammy.configuration.client
 
       @client ||= ::Anthropic::Client.new(
         access_token: ENV.fetch('ANTHROPIC_API_KEY')

data/lib/lammy/openai.rb

@@ -7,20 +7,20 @@ require 'base64'
 module Lammy
   # Use the OpenAI API's Ruby library
   class OpenAI
-    MODELS = %w[
-      gpt-4o gpt-4o-2024-08-06 gpt-4o-2024-05-13 gpt-4o chatgpt-4o-latest
+    MODELS = [
+      /^gpt-4o(?:-\d{4}-\d{2}-\d{2})?$/, /^chatgpt-4o-latest$/,
 
-      gpt-4o-mini gpt-4o-mini-2024-07-18
+      /^gpt-4o-mini(?:-\d{4}-\d{2}-\d{2})?$/,
 
-      o1-preview o1-preview-2024-09-12
+      /^o1(?:-preview(?:-\d{4}-\d{2}-\d{2})?)?$/,
 
-      o1-mini o1-mini-2024-09-12
+      /^o1-mini(?:-\d{4}-\d{2}-\d{2})?$/,
 
-      gpt-3.5-turbo
+      /^gpt-3\.5-turbo$/,
 
-      gpt-4-turbo gpt-4-turbo-2024-04-09 gpt-4 gpt-4-32k gpt-4-0125-preview gpt-4-1106-preview gpt-4-vision-preview
-      gpt-3.5-turbo-0125 gpt-3.5-turbo-instruct gpt-3.5-turbo-1106 gpt-3.5-turbo-0613 gpt-3.5-turbo-16k-0613
-      gpt-3.5-turbo-0301 davinci-002 babbage-002
+      /^gpt-4(?:-turbo(?:-\d{4}-\d{2}-\d{2})?|-32k|-\d{4}-preview|-vision-preview)?$/,
+      /^gpt-3\.5-turbo-(?:\d{4}|\d{2}k-\d{4}|-instruct)$/,
+      /^(?:davinci|babbage)-002$/
     ].freeze
 
     EMBEDDINGS = %w[
@@ -40,7 +40,7 @@ module Lammy
 
       request = client.chat(
         parameters: {
-          model: settings[:model],
+          model: settings[:model] || Lammy.configuration.model,
           response_format: schema,
           messages: messages,
           stream: stream ? ->(chunk) { stream.call(stream_content(chunk)) } : nil
@@ -118,6 +118,7 @@ module Lammy
 
     def client
       return settings[:client] if settings[:client]
+      return Lammy.configuration.client if Lammy.configuration.client
 
       @client ||= ::OpenAI::Client.new(
         access_token: ENV.fetch('OPENAI_ACCESS_TOKEN')

data/lib/lammy.rb

@@ -1,5 +1,24 @@
 # frozen_string_literal: true
 
+module Lammy
+  class Configuration
+    attr_accessor :model, :client
+
+    def initialize
+      @model = nil
+      @client = nil
+    end
+  end
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  def self.configure
+    yield(configuration)
+  end
+end
+
 require 'lammy/embeddings'
 require 'lammy/claude'
 require 'lammy/openai'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lammy
 version: !ruby/object:Gem::Version
-  version: 0.8.3
+  version: 0.9.0
 platform: ruby
 authors:
 - Kamil Nicieja
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-29 00:00:00.000000000 Z
+date: 2024-12-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: anthropic