rach 0.2.8 → 0.2.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aae006344d427b9f6b585ec28419beca527ee1659faeb6188e24b73a0c0d4eaf
-  data.tar.gz: a7348260087b8829c8465d1a3a6a6129be0a7c467e2f93fdb4dcb753ee87ae24
+  metadata.gz: b9479cf41fe01b10e9c117939304120042337fa3a3ce8bc27fc1649e69894bcc
+  data.tar.gz: ea441f9a3978f58bc40284cfa5cefe49533f9a53275c8c85aca3761247dcf91b
 SHA512:
-  metadata.gz: ac363b216035937755af0e712aaa2b40d871321a791f5bb2d8be7ca3f407a3fee7e36c5746cc982f3b361c52c9b044ea89624994809547155a0c5b62241a75d7
-  data.tar.gz: 61238f0299ff7f149e49cfe2b4ab660aae899337e59783e0d54f91216e0a3cc990bec0aa0b1b81abf008e2c9c2468a463395250372d5f098b1bb29bfeb70882e
+  metadata.gz: 8e185fa5b250d9433a71af3e31089b046ed4087ace488e608e54184f444bc152f9e02f0a5515a2a878ff7d07c56537a53906b111c39c8ee9cd42edc39b2014b8
+  data.tar.gz: ee0e72f793aee837570acfddc55da33aee2b9a13161e813121de6abe6955877828ee7c20638a0731a40367a9c40ce1aacdda2edaa548f3c4fd0ac0a59b7ce31c

data/lib/rach/client.rb

@@ -13,7 +13,7 @@ module Rach
         setup_providers(providers)
       elsif access_token && model
         provider = Provider.for(model)
-        setup_providers({ provider.key => { access_token: access_token } })
+        setup_providers({ provider.key => { access_token: access_token, model: model } })
       else
         raise ArgumentError, "Either (providers) or (access_token AND model) must be provided"
       end

data/lib/rach/provider/google.rb

@@ -0,0 +1,105 @@
+require 'gemini-ai'
+
+module Rach
+  module Provider
+    class Google < Base
+
+      def initialize(access_token: nil, logger: nil, **kwargs)
+        @client = create_client(access_token, **kwargs)
+        @logger = logger
+      end
+
+      def chat(**parameters)
+        messages = parameters.dig(:parameters, :messages) || []
+        system_message = messages.find { |msg| msg[:role] == "system" }
+        messages = messages.reject { |msg| msg[:role] == "system" } if system_message
+
+        # Convert messages to Gemini format
+        contents = messages.map do |msg|
+          {
+            role: msg[:role] == "assistant" ? "model" : "user",
+            parts: { text: msg[:content] }
+          }
+        end
+
+        # If there's a system message, prepend it to user's first message
+        if system_message
+          first_user_message = contents.find { |msg| msg[:role] == "user" }
+          if first_user_message
+            first_user_message[:parts][:text] = "#{system_message[:content]}\n\n#{first_user_message[:parts][:text]}"
+          end
+        end
+
+        request_params = { contents: contents }
+
+        # Handle response format if provided
+        if response_format = parameters.dig(:parameters, :response_format)
+          request_params[:generation_config] = {
+            response_mime_type: 'application/json',
+            response_schema: convert_response_format(response_format)
+          }
+        end
+
+        if request_params.dig(:generation_config, :response_schema)
+          request_params[:generation_config][:response_schema].delete(:additionalProperties)
+          request_params[:generation_config][:response_schema].delete(:required)
+        end
+
+        if @logger
+          @logger.info("Making API call to Google Gemini")
+          @logger.info("Request parameters: #{request_params.inspect}")
+        end
+
+        raw_response = @client.generate_content(request_params)
+
+        if @logger
+          @logger.info("Response: #{raw_response.inspect}")
+        end
+
+        Response.new(
+          id: raw_response["candidates"][0]["content"]["parts"][0]["text"],
+          model: raw_response["modelVersion"],
+          content: raw_response["candidates"][0]["content"]["parts"][0]["text"],
+          usage: {
+            "prompt_tokens" => raw_response["usageMetadata"]["promptTokenCount"],
+            "completion_tokens" => raw_response["usageMetadata"]["candidatesTokenCount"],
+            "total_tokens" => raw_response["usageMetadata"]["totalTokenCount"]
+          },
+          raw_response: raw_response
+        )
+      end
+
+      def self.supports?(model)
+        model.start_with?("gemini")
+      end
+
+      private
+
+      def create_client(access_token, **config)
+        client_config = {
+          credentials: {
+            service: 'generative-language-api',
+            api_key: access_token
+          },
+          options: { model: config[:model] }
+        }
+        
+        # Only merge additional options that aren't already set
+        client_config[:options].merge!(config.except(:model))
+
+        # Function calling and structured output are broken in this gem
+        gemini = ::Gemini.new(**client_config)
+        base_address = gemini.instance_variable_get(:@base_address)
+        base_address.gsub!('v1', 'v1beta') if base_address
+        gemini
+      end
+
+      def convert_response_format(format)
+        return unless format[:type] == "json_schema"
+
+        schema = format.dig(:json_schema, :schema)
+        schema.deep_symbolize_keys
+      end
+    end
+  end
+end 

data/lib/rach/provider.rb

@@ -3,7 +3,8 @@ module Rach
 
     AVAILABLE_PROVIDERS = [
       Provider::OpenAI,
-      Provider::Anthropic
+      Provider::Anthropic,
+      Provider::Google
     ].to_h { |p| [p.key, p] }.freeze
 
     def self.for(model)

data/lib/rach/version.rb

@@ -1,4 +1,4 @@
 module Rach
-  VERSION = "0.2.8"
+  VERSION = "0.2.10"
 end
 

data/lib/rach.rb

@@ -14,6 +14,7 @@ require_relative "rach/function"
 require_relative "rach/provider/base"
 require_relative "rach/provider/openai"
 require_relative "rach/provider/anthropic"
+require_relative "rach/provider/google"
 require_relative "rach/provider"
 require_relative "rach/prompt"
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rach
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.2.10
 platform: ruby
 authors:
 - Roger Garcia
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-01-30 00:00:00.000000000 Z
+date: 2025-02-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -100,6 +100,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '7.3'
+- !ruby/object:Gem::Dependency
+  name: gemini-ai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 4.2.0
 description: Rach is a lightweight framework for orchestrating AI agents
 email:
 - rach@rogergarcia.me
@@ -119,6 +133,7 @@ files:
 - lib/rach/provider.rb
 - lib/rach/provider/anthropic.rb
 - lib/rach/provider/base.rb
+- lib/rach/provider/google.rb
 - lib/rach/provider/openai.rb
 - lib/rach/response.rb
 - lib/rach/response_format.rb