inst_llm 0.2.2 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ca52be39233578c28b7d4b2e7a91b46c0b972d3806050ebb64cf7874cb11ca6c
-  data.tar.gz: ba51f3024ad3982515a265ed427c99e01c814069804e1e6a917f516a02999362
+  metadata.gz: 32af75986a0d5552c9b4830e126c2082217f4927287df3cf1f0b99c7dadaf51e
+  data.tar.gz: 5e4b32fb74b2aae186fc2f0afdd10125b3d880100432bae9527866870535e0ee
 SHA512:
-  metadata.gz: 2e713687d83aedbd2b6ab655f849dc29fb544fb5ea74b76f52547022646edba3e6ed4a1c9eaf45d44dc313d3cb39cfe7e1dafbd16483011a7fd7cba767899148
-  data.tar.gz: 2a8591d2ecd8019ad46e0d5adda72ddc652a2de3719f239ecc78dcac3f79482feccb97bbc86812cb673dd48c61b45bc23d0a296f60bfa38e188e83a9850c63ef
+  metadata.gz: 5408ce4b2e94537186fc95c06a09c64983714b998032b886bcfa7ce871262c989fc012cd8276eb3fc9e4b8dd51015b1677e7233b855902144d06db9fe1e31d7e
+  data.tar.gz: 0447e73e81cdb5d12e02e0137ccc3233dafabef6e93a2fd3135f0caf9f28362ab1f37b8f207418d7f0e35c9f99d869e9a97bd1c006c82c3cbfc7a3b09c28f01d

data/lib/inst_llm/client.rb

@@ -11,6 +11,7 @@ module InstLLM
     MODELS = {
       "anthropic.claude-3-sonnet-20240229-v1:0": { format: :claude, provider: :bedrock, type: :chat },
       "anthropic.claude-3-haiku-20240307-v1:0": { format: :claude, provider: :bedrock, type: :chat },
+      "anthropic.claude-3-opus-20240229-v1:0": { format: :claude, provider: :bedrock, type: :chat },
 
       "meta.llama3-8b-instruct-v1:0": { format: :llama3, provider: :bedrock, type: :chat },
       "meta.llama3-70b-instruct-v1:0": { format: :llama3, provider: :bedrock, type: :chat },

data/lib/inst_llm/parameter/mistral_parameters.rb

@@ -23,9 +23,23 @@ module InstLLM
 
       private
 
+      def format_message(message)
+        return message if message[:content].is_a?(String)
+
+        content = Array(message[:content]).map do |m|
+          raise UnknownArgumentError, <<~ERR if m[:type] != :text
+            Model does not support multiple types. Expected :text, received #{m[:type]} 
+          ERR
+
+          m[:text]
+        end
+
+        { role: message[:role], content: content.join("\n\n") }
+      end
+
       def prompt
         system_message = nil
-        prompt = @messages.map do |message|
+        prompt = @messages.map { |message| format_message(message) }.map do |message|
           case message[:role].to_sym
           when :assistant
             "#{message[:content]}"

data/lib/inst_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module InstLLM
-  VERSION = "0.2.2"
+  VERSION = "0.2.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: inst_llm
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.4
 platform: ruby
 authors:
 - Zach Pendleton
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-04-25 00:00:00.000000000 Z
+date: 2024-04-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-bedrockruntime