lammy 0.8.0 → 0.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ec35c0b162ca174c4baaa39d040632e10ce1e18a8b79e281161ff79f99a0758b
-  data.tar.gz: 88542e30eddbd8e1d26e2fbef989e7e2467f0503abc1544410f9f6f9490f6083
+  metadata.gz: 4708e6d34ee95ab93a6e737204ced2d8bac4dab9dde0102b1cfeaa9df7f31afd
+  data.tar.gz: 896b3e2211f6fd6ad2918f31657fd794d98e53a27027773c7b7d2f275f5ad390
 SHA512:
-  metadata.gz: 9f2230a56f061fb8595ad27a6539c42807e0507fbd86542866f79a9a40f8c98cd345d45a3d3611ebc7c738b5e0919b55f5567b27c60db83c393b64be25edcbde
-  data.tar.gz: d2c855118e9cfb30287ba8902c1dd74aed6bcd5161ba59b8752eeb755b75735ec1a634166b2493f718c5a6572d496f796e370c3232ec0e05dc01a0f30772441c
+  metadata.gz: 7b7c7d3fa8d35996d5287d9be7cbf16dee55ed828f39e38faf460e87f9d1d75f9f9073b1e96461ea7ddf3459a2d7d8b1617dadd0ccc2b6ad44e4a6db2fe2be89
+  data.tar.gz: de5c76c9ede8414de9fc9de0da4bbca907948be8c5ce43cae6477d1f0e868cedfcd8095ce06fbfaff1f960b58ed8bf2150c7051a8048fcacdbc2d5b3665eff8b

data/lib/lammy/chat.rb

@@ -1,20 +1,13 @@
 # frozen_string_literal: true
 
-module L
+module Lammy
   module Chat
-    def llm(**kwargs)
-      @next_llm_settings = kwargs
-    end
-
-    def handle_llm(method_name)
-      settings = @next_llm_settings
-      @next_llm_settings = nil
-
+    def self.handle(klass, method_name, settings)
       # Unbind the original method
-      original_method = instance_method(method_name)
+      original_method = klass.instance_method(method_name)
 
       # Redefine the method
-      define_method(method_name) do |*args, &block|
+      klass.define_method(method_name) do |*args, &block|
         # Initialize context
         @system_message = nil
 

data/lib/lammy/claude.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require 'anthropic'
+require 'hashie'
+
+module Lammy
+  # Use the Claude API's Ruby library
+  class Claude
+    MODELS = %w[
+      claude-3-5-sonnet-20240620
+      claude-3-opus-20240229 claude-3-sonnet-20240229 claude-3-haiku-20240307
+      claude-2.1 claude-2.0 claude-instant-1.2
+    ].freeze
+
+    attr_reader :settings
+
+    def initialize(settings)
+      @settings = settings
+    end
+
+    # Generate a response with support for structured output
+    def chat(user_message, system_message = nil, stream = nil)
+      response = client.messages(
+        parameters: {
+          system: system_message,
+          model: settings[: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))]
+        }.compact
+      )
+
+      stream || response.dig('content', 0, 'text')
+    end
+
+    private
+
+    def stream_content(chunk)
+      chunk.dig('delta', 'text')
+    end
+
+    def vision(message)
+      image = message[:_image]
+      base = message.except(:_image)
+
+      return base unless image
+
+      messages = [
+        { 'type' => 'image',
+          'source' => { 'type' => 'base64', 'media_type' => 'image/jpeg', 'data' => Base64.strict_encode64(image) } },
+        { 'type' => 'text', 'text' => message[:content] }
+      ]
+
+      base.merge(content: messages)
+    end
+
+    def client
+      return settings[:client] if settings[:client]
+
+      @client ||= ::Anthropic::Client.new(
+        access_token: ENV.fetch('ANTHROPIC_API_KEY')
+      )
+    end
+  end
+end

data/lib/lammy/embeddings.rb

@@ -1,20 +1,13 @@
 # frozen_string_literal: true
 
-module L
+module Lammy
   module Embeddings
-    def v(**kwargs)
-      @next_v_settings = kwargs
-    end
-
-    def handle_v(method_name)
-      settings = @next_v_settings
-      @next_v_settings = nil
-
+    def self.handle(klass, method_name, settings)
       # Unbind the original method
-      original_method = instance_method(method_name)
+      original_method = klass.instance_method(method_name)
 
       # Redefine the method
-      define_method(method_name) do |*args, &block|
+      klass.define_method(method_name) do |*args, &block|
         # # Initialize chunking settings
         # @chunk_by_size = nil
 

data/lib/lammy/openai.rb

@@ -4,7 +4,7 @@ require 'openai'
 require 'hashie'
 require 'base64'
 
-module L
+module Lammy
   # Use the OpenAI API's Ruby library
   class OpenAI
     MODELS = %w[

data/lib/lammy/schema.rb

@@ -15,7 +15,7 @@ module L
     def user(content, image: nil)
       return content if L::Schema.formatted?(content)
 
-      { role: :user, content: content, _image: image }
+      { role: :user, content: content, _image: image }.compact
     end
 
     def assistant(content)

data/lib/lammy.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+module Lammy; end
+
 require 'lammy/embeddings'
 require 'lammy/claude'
 require 'lammy/openai'
@@ -10,16 +12,34 @@ module L
   extend Schema
 
   def self.included(base)
-    base.extend Chat
-    base.extend Embeddings
     base.extend ClassMethods
   end
 
   # Wrap generative methods with handlers
   module ClassMethods
+    def llm(**kwargs)
+      @next_llm_settings = kwargs
+    end
+
+    def v(**kwargs)
+      @next_v_settings = kwargs
+    end
+
     def method_added(method_name)
-      handle_llm(method_name) if @next_llm_settings
-      handle_v(method_name) if @next_v_settings
+      if @next_llm_settings
+        next_llm_settings = @next_llm_settings
+        @next_llm_settings = nil
+
+        Lammy::Chat.handle(self, method_name, next_llm_settings)
+      end
+
+      if @next_v_settings
+        next_v_settings = @next_v_settings
+        @next_v_settings = nil
+
+        Lammy::Embeddings.handle(self, method_name, next_v_settings)
+      end
+
       super
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lammy
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.1
 platform: ruby
 authors:
 - Kamil Nicieja
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-08 00:00:00.000000000 Z
+date: 2024-10-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: anthropic
@@ -74,6 +74,7 @@ extra_rdoc_files: []
 files:
 - lib/lammy.rb
 - lib/lammy/chat.rb
+- lib/lammy/claude.rb
 - lib/lammy/embeddings.rb
 - lib/lammy/openai.rb
 - lib/lammy/schema.rb