geminiext 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 0217cac9967438f0bbd4fe0dc38b6c0aba1d21475b937c6b4d1b4d41d51a12b6
+  data.tar.gz: df412fa82b43d42dcaca12b993cfa76f6c9e49dea76e55dfa9cbcddc79a9bee6
+SHA512:
+  metadata.gz: 884cdc51610f1d2beefd9884a4b779560adfff38c14deafc3bad84b3dc2ce84079cabf29eae88d96e6688d73dc25824d23eaea7fc2b379fac327de87974e29ea
+  data.tar.gz: e9f69ffe68c4434738612884e54f89d5e471016ee1425846d374f9efa80b13e3fb4210f3f21a3b088e292861717fe0fa3f45473414d28307ec5b63781fe84b3b

data/README.md

@@ -0,0 +1 @@
+# geminiext

data/lib/geminiext.rb

@@ -0,0 +1,23 @@
+require 'gemini-ai'
+
+# require 'geminiext/model'
+# require 'geminiext/messages'
+# require 'geminiext/response_extender'
+
+module GeminiExt
+  MAX_TOKENS = ENV.fetch('GEMINI_MAX_TOKENS', 8_192).to_i
+  
+  def self.new(model: 'gemini-1.5-flash-001')
+    Gemini.new(load_config(model: model))
+  end
+
+  def self.load_config(model: 'gemini-1.5-flash-001')
+    {
+      credentials: {
+        service: 'generative-language-api',
+        api_key: ENV.fetch('GEMINI_API_KEY')
+      },
+      options: { model:}
+    }    
+  end
+end

data/lib/openaiext/messages.rb.disabled

@@ -0,0 +1,34 @@
+module geminiext
+  class Messages < Array
+    def initialize messages = nil
+      super parse_messages(messages)
+    end
+
+    def add(message) = concat(parse_messages(message))
+
+    private
+    def parse_messages(messages)
+      return [] if messages.nil?
+
+      messages = [messages] unless messages.is_a?(Array)
+      
+      # if first element is ok, then do not parse the rest
+      return messages if messages.first in { role: String | Symbol, content: String | Array | Hash}
+
+      messages.flat_map do |msg|
+        if msg.is_a?(Hash)
+          if msg.keys.size == 1
+            role, content = msg.first
+            { role: role.to_s, content: content }
+          elsif msg.key?(:role) && msg.key?(:content)
+            { role: msg[:role].to_s, content: msg[:content] }
+          else
+            msg.map { |role, content| { role: role.to_s, content: content } }
+          end
+        else
+          raise ArgumentError, "Invalid message format: #{msg}"
+        end
+      end
+    end
+  end
+end

data/lib/openaiext/model.rb.disabled

@@ -0,0 +1,27 @@
+module geminiext
+  module Model
+    GPT_BASIC_MODEL = ENV.fetch('OPENAI_GPT_BASIC_MODEL', 'gpt-4o-mini')
+    GPT_ADVANCED_MODEL = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL', 'gpt-4o')
+    GPT_ADVANCED_MODEL_LATEST = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL_LATEST', 'chatgpt-4o-latest')
+
+    O1_BASIC_MODEL = ENV.fetch('OPENAI_O1_BASIC_MODEL', 'o1-mini')
+    O1_ADVANCED_MODEL = ENV.fetch('OPENAI_O1_ADVANCED_MODEL', 'o1-preview')
+
+    def self.select(model)
+      case model
+      when :gpt_basic
+        GPT_BASIC_MODEL
+      when :gpt_advanced
+        GPT_ADVANCED_MODEL
+      when :gpt_advanced_latest
+        GPT_ADVANCED_MODEL_LATEST
+      when :o1_basic
+        O1_BASIC_MODEL
+      when :o1_advanced
+        O1_ADVANCED_MODEL
+      else
+        model
+      end
+    end
+  end
+end

data/lib/openaiext/response_extender.rb.disabled

@@ -0,0 +1,44 @@
+module ResponseExtender
+  def chat_params = self[:chat_params]
+
+  def message = dig('choices', 0, 'message')
+
+  def content = dig('choices', 0, 'message', 'content')
+  def content? = !content.nil?
+
+  def tool_calls = dig('choices', 0, 'message', 'tool_calls')
+  def tool_calls? = !tool_calls.nil?
+
+  def functions
+    return if tool_calls.nil?
+    
+    functions = tool_calls.filter { |tool| tool['type'].eql? 'function' }
+    return if functions.empty?
+    
+    functions_list = []
+    functions.map.with_index do |function, function_index|
+      function_info = tool_calls.dig(function_index, 'function')
+      function_def = { id: function['id'], name: function_info['name'], arguments: Oj.load(function_info['arguments'], symbol_keys: true) }
+      
+      def function_def.run(context:)
+        {
+          tool_call_id: self[:id],
+          role: :tool,
+          name: self[:name],
+          content: context.send(self[:name], **self[:arguments])
+        }
+      end
+
+      functions_list << function_def
+    end
+
+    functions_list
+  end
+
+  def functions_run_all(context:)
+    raise 'No functions to run' if functions.nil?
+    functions.map { |function| function.run(context:) }
+  end
+
+  def functions? = !functions.nil?
+end

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: geminiext
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Gedean Dias
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-11-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: gemini-ai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+description: Based on gemini-ai, adds some extra features
+email: gedean.dias@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- lib/geminiext.rb
+- lib/openaiext/messages.rb.disabled
+- lib/openaiext/model.rb.disabled
+- lib/openaiext/response_extender.rb.disabled
+homepage: https://github.com/gedean/geminiext
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.23
+signing_key: 
+specification_version: 4
+summary: Ruby Gemini Extended
+test_files: []