inst_llm 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ca52be39233578c28b7d4b2e7a91b46c0b972d3806050ebb64cf7874cb11ca6c
-  data.tar.gz: ba51f3024ad3982515a265ed427c99e01c814069804e1e6a917f516a02999362
+  metadata.gz: c3ef21142135987ed0d03bd322ca86bd478c299a2c1430d5afea62c714b4552e
+  data.tar.gz: 188e6a291efc1de2485d01496b01a20fae10d72d79a3a90c5153df90dc057c8d
 SHA512:
-  metadata.gz: 2e713687d83aedbd2b6ab655f849dc29fb544fb5ea74b76f52547022646edba3e6ed4a1c9eaf45d44dc313d3cb39cfe7e1dafbd16483011a7fd7cba767899148
-  data.tar.gz: 2a8591d2ecd8019ad46e0d5adda72ddc652a2de3719f239ecc78dcac3f79482feccb97bbc86812cb673dd48c61b45bc23d0a296f60bfa38e188e83a9850c63ef
+  metadata.gz: 904b2a084d3db540f620a030178a8f6141d3c02d87f942703850a234f7e2e0dccf7ee209abadfd57bc3f9c85c662889601a24e21ae9df59ae53f5aecde7fa7e8
+  data.tar.gz: 0bc2d34e8f9f19a836d5c96732990cba1cf824cd16722de7811ae73f2410c2ec0de738d0e96c86faf088f4ec8111bcb6564ac2a377ed539aac632bf7b785221d

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.3"
 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.3
 platform: ruby
 authors:
 - Zach Pendleton
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-04-25 00:00:00.000000000 Z
+date: 2024-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-bedrockruntime