raix 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 318057e8ece37b63c06884a61c37dc1ef15f38cee2c05d5305bcaf6d3697420e
-  data.tar.gz: 250229d71a808203689b87cea7c1ce8dd31b085c92b54dce392787299cd420d6
+  metadata.gz: 00be6de8258fbe226ea622d4d0e591fa6035ddd8d5ad627dd316291561317355
+  data.tar.gz: 2749f23745e8385a770d1214c131c3babd2b48e2c983c10eaa1775ae85151879
 SHA512:
-  metadata.gz: dc51d8fab907f8ffa5e95df2ef308ee3cd5fc443f46803e8a6f184be80e12719be2d25aeb51349a7912b35df55866db66cdec532c0258bc66a97aeabbc4017d0
-  data.tar.gz: 7b393143a5da05ba75ac11a8a77e31b0e61c1e7e9cb564fc6c986a4f63e2f98a24d43056f613e8e4685baab85251c9e819bcf99fcdc22ee7428b50eb9895d4e7
+  metadata.gz: 561520feed1ab27d400e1482ecc3a3740fdb3d21b0ac09603f59b1b798facf9d6bb83f910c4f77b66ce7b5a60d6278bbe174429853e74aa5e5ccda771603566e
+  data.tar.gz: 537a7d27ad0d9a717f52bdb7a0aaa0d6f8479f0261f9730f18d0e93255c734e027d577d9284425a213d151fb08dd95c2f988a21015cd7746706e989dedb91cd0

data/CHANGELOG.md

@@ -14,3 +14,4 @@
 
 ## [0.4.0] - 2024-10-18
 - adds support for Anthropic-style prompt caching
+- defaults to `max_completion_tokens` when using OpenAI directly

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    raix (0.3.2)
+    raix (0.4.1)
       activesupport (>= 6.0)
       open_router (~> 0.2)
 

data/lib/raix/chat_completion.rb

@@ -2,10 +2,11 @@
 
 require "active_support/concern"
 require "active_support/core_ext/object/blank"
-require "raix/message_adapters/base"
 require "open_router"
 require "openai"
 
+require_relative "message_adapters/base"
+
 module Raix
   # The `ChatCompletion`` module is a Rails concern that provides a way to interact
   # with the OpenRouter Chat Completion API via its client. The module includes a few

data/lib/raix/message_adapters/base.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "active_support/core_ext/module/delegation"
+
+module Raix
+  module MessageAdapters
+    # Transforms messages into the format expected by the OpenAI API
+    class Base
+      attr_accessor :context
+
+      delegate :cache_at, :model, to: :context
+
+      def initialize(context)
+        @context = context
+      end
+
+      def transform(message)
+        return message if message[:role].present?
+
+        if message[:function].present?
+          { role: "assistant", name: message.dig(:function, :name), content: message.dig(:function, :arguments).to_json }
+        elsif message[:result].present?
+          { role: "function", name: message[:name], content: message[:result] }
+        else
+          content(message)
+        end
+      end
+
+      protected
+
+      def content(message)
+        case message
+        in { system: content }
+          { role: "system", content: }
+        in { user: content }
+          { role: "user", content: }
+        in { assistant: content }
+          { role: "assistant", content: }
+        else
+          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.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
+      end
+    end
+  end
+end

data/lib/raix/response_format.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+require "active_support/core_ext/object/deep_dup"
+
+module Raix
+  # Handles the formatting of responses for AI interactions.
+  #
+  # This class is responsible for converting input data into a JSON schema
+  # that can be used to structure and validate AI responses. It supports
+  # nested structures and arrays, ensuring that the output conforms to
+  # the expected format for AI model interactions.
+  #
+  # @example
+  #   input = { name: { type: "string" }, age: { type: "integer" } }
+  #   format = ResponseFormat.new("PersonInfo", input)
+  #   schema = format.to_schema
+  #
+  # @attr_reader [String] name The name of the response format
+  # @attr_reader [Hash] input The input data to be formatted
+  class ResponseFormat
+    def initialize(name, input)
+      @name = name
+      @input = input
+    end
+
+    def to_json(*)
+      JSON.pretty_generate(to_schema)
+    end
+
+    def to_schema
+      {
+        type: "json_schema",
+        json_schema: {
+          name: @name,
+          schema: {
+            type: "object",
+            properties: decode(@input.deep_dup),
+            required: @input.keys,
+            additionalProperties: false
+          },
+          strict: true
+        }
+      }
+    end
+
+    private
+
+    def decode(input)
+      {}.tap do |response|
+        case input
+        when Array
+          properties = {}
+          input.each { |item| properties.merge!(decode(item)) }
+
+          response[:type] = "array"
+          response[:items] = {
+            type: "object",
+            properties:,
+            required: properties.keys.select { |key| properties[key].delete(:required) },
+            additionalProperties: false
+          }
+        when Hash
+          input.each do |key, value|
+            response[key] = if value.is_a?(Hash) && value.key?(:type)
+                              value
+                            else
+                              decode(value)
+                            end
+          end
+        else
+          raise "Invalid input"
+        end
+      end
+    end
+  end
+end

data/lib/raix/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Raix
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

data/lib/raix.rb

@@ -4,6 +4,7 @@ require_relative "raix/version"
 require_relative "raix/chat_completion"
 require_relative "raix/function_dispatch"
 require_relative "raix/prompt_declarations"
+require_relative "raix/response_format"
 
 # The Raix module provides configuration options for the Raix gem.
 module Raix

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Obie Fernandez
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-10-19 00:00:00.000000000 Z
+date: 2024-10-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -58,7 +58,9 @@ files:
 - lib/raix.rb
 - lib/raix/chat_completion.rb
 - lib/raix/function_dispatch.rb
+- lib/raix/message_adapters/base.rb
 - lib/raix/prompt_declarations.rb
+- lib/raix/response_format.rb
 - lib/raix/version.rb
 - raix.gemspec
 - sig/raix.rbs