lammy 0.3.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9225fc482f95ed862d6e12bc4541731c8854127900e0bfe369075ef1092fae8e
-  data.tar.gz: 720ce8849858edb0832c34b0b10ef736c6353b12daa1e5cecf68c137de94a91d
+  metadata.gz: dddd7ebf694f48f5d75ff062990bdec4f4e2c1188af7f958fe12a6856928c103
+  data.tar.gz: ea361dc698c42d8c8c37318e0e98300d7fb416e31de53d2b4f3ab05882bf480d
 SHA512:
-  metadata.gz: 83be8412698ad307dba650aaa54e230cf9b47ee9e140e9dac1d26c2927f28b43649ac9b2e6196154a09b8b608a04b8fb72e7c0792e2a6bd799a2ff1edd339494
-  data.tar.gz: 9add86016f31a13416848ff64a7b9eeea03808b9220b239b029049c1ffc04796a9253f71dfac623928b8eddacbd8ffa182a243fd1a7c50c5a5ade6bdc8a68b50
+  metadata.gz: c925d41bfa4a2b66867e43e081187f76f0143775d1e2ac0044b6a243fc3aadcaec13499b9678cbc891b076ae50c9b8612999d183c987b77ae99838658f4f222d
+  data.tar.gz: dcfb1071b3eb78ad66e1151b2d04d4b58581d576ff211f48d2126d9d52af991abbb387bb5cdc36c925efca9ca990a8be1bf1d19767ac5b4dd5aa9d904c997077

data/lib/lammy/chat.rb

@@ -17,22 +17,31 @@ module L
       define_method(method_name) do |*args, &block|
         # Initialize context
         @system_message = nil
+        @prefilled_message = nil
 
-        # Make `context` method available within the instance
+        # `context` sets the system message and is available within the instance
         define_singleton_method(:context) do |message|
           @system_message = message
         end
 
+        # `prefill` sets the prefilled message and is available within the instance
+        define_singleton_method(:prefill) do |message|
+          @prefilled_message = message
+        end
+
         # 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(settings)
-          client.chat(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, @prefilled_message)
       end
     end
   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
@@ -22,7 +22,7 @@ module L
     end
 
     # Generate a response with support for structured output
-    def chat(user_message, system_message = nil)
+    def chat(user_message, system_message = nil, _ = nil)
       schema = schema(settings)
       response = client.chat(
         parameters: {

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.3.0
+  version: 0.5.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:
   - - ">="