rach 0.2.8 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aae006344d427b9f6b585ec28419beca527ee1659faeb6188e24b73a0c0d4eaf
-  data.tar.gz: a7348260087b8829c8465d1a3a6a6129be0a7c467e2f93fdb4dcb753ee87ae24
+  metadata.gz: d38fee5588b8410bd344b1848429f4cd5534dbfc46bdb3c7c55193156b3f78b5
+  data.tar.gz: 8a8b353426c367c4b0a7663d66a0a82a91c594a23384f12ed17f3484513aa105
 SHA512:
-  metadata.gz: ac363b216035937755af0e712aaa2b40d871321a791f5bb2d8be7ca3f407a3fee7e36c5746cc982f3b361c52c9b044ea89624994809547155a0c5b62241a75d7
-  data.tar.gz: 61238f0299ff7f149e49cfe2b4ab660aae899337e59783e0d54f91216e0a3cc990bec0aa0b1b81abf008e2c9c2468a463395250372d5f098b1bb29bfeb70882e
+  metadata.gz: 0c963bd5389c85b39ac2f2fb8ef94f3e35167bd265e2dfa77c6a63642ae42d156954e9d151fbea994b3c12fe5d5a7db15f2671501449a74df17e4e26a1e3383d
+  data.tar.gz: 8f3e9700244b3fc29414a512b1cbf84818b72d73cc44d60a6dc581835a77d680f0616008477bbaf54f68e2fcf2dbfa540fc104aa346cd3882c47826c578a90d9

data/lib/rach/provider/google.rb

@@ -0,0 +1,77 @@
+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
+
+        if @logger
+          @logger.info("Making API call to Google Gemini")
+          @logger.info("Request parameters: #{contents.inspect}")
+        end
+
+        raw_response = @client.generate_content({ contents: contents })
+
+        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: 'gemini-pro' }
+        }
+        client_config[:options].merge!(config)
+
+        ::Gemini.new(**client_config)
+      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.9"
 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.9
 platform: ruby
 authors:
 - Roger Garcia
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-01-30 00:00:00.000000000 Z
+date: 2025-02-05 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