make content check a little more resilient

OlympiaAI · Oct 21, 2024 · 8809ccb · 8809ccb
1 parent d5dac09
commit 8809ccb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/raix/message_adapters/base.rb b/lib/raix/message_adapters/base.rb
@@ -40,7 +40,7 @@ def content(message)
           raise ArgumentError, "Invalid message format: #{message.inspect}"
         end.tap do |msg|
           # convert to anthropic multipart format if model is claude-3 and cache_at is set
-          if model["anthropic/claude-3"] && cache_at && msg[:content].length > cache_at.to_i
+          if model.to_s.include?("anthropic/claude-3") && cache_at && msg[:content].to_s.length > cache_at.to_i
             msg[:content] = [{ type: "text", text: msg[:content], cache_control: { type: "ephemeral" } }]
           end
         end