gemini_cache 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7bc6c689b1bad80c2f7865816137cd8cbadd3a3c112938d7a0f8982820f54065
+  data.tar.gz: c18e33fe4733104d6945cd1e72255066e2e49e339cc18873cfad3276721e7c9a
+SHA512:
+  metadata.gz: 6f8659b09c15ab666ca8e858389eb39a4e2efabb9368a98685bce34bd45049359cf5a1c0c8e9080ea5bcf5d05f113df17d784f768154d720c9a87f135f0b4de5
+  data.tar.gz: eb06c97493baf8a5cf61d6020fbb47094e9b831ec257afae63148ba5127867a9f8c5ba8b3241fe548e6727766afe3e3a8f9137d0def687c1cfa13f9638837b43

data/lib/gemini_cache.rb

@@ -0,0 +1,125 @@
+require 'faraday'
+require 'json'
+
+module GeminiCache
+  def self.create(contents:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+    content = {
+      model: "models/#{model}",
+      display_name:,
+      contents:,
+      ttl: "#{ttl}s"
+    }.to_json
+  
+    conn = Faraday.new(
+      url: 'https://generativelanguage.googleapis.com',
+      headers: { 'Content-Type' => 'application/json' }
+    )
+  
+    response = conn.post('/v1beta/cachedContents') do |req|
+      req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+      req.body = content
+    end
+  
+    return JSON.parse(response.body) if response.status == 200
+  
+    raise "Erro ao criar cache: #{response.status} - #{response.body}"
+  rescue Faraday::Error => e
+    raise "Erro na requisição: #{e.message}"
+  end
+
+  def self.get(name: nil, display_name: nil)
+    raise 'Nome do cache ou display name é obrigatório' if name.nil? && display_name.nil?
+    raise 'Nome do cache e display name não podem ser informados juntos' if !name.nil? && !display_name.nil?
+    
+    return GeminiCache::Cache.list.find { |item| item['name'].eql? name } if !name.nil?
+    return GeminiCache::Cache.list.find { |item| item['displayName'].eql? display_name } if !display_name.nil?
+  end
+
+  def self.list
+    conn = Faraday.new(
+      url: 'https://generativelanguage.googleapis.com',
+      headers: { 'Content-Type' => 'application/json' }
+    )
+
+    response = conn.get("/v1beta/cachedContents") do |req|
+      req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+    end
+    
+    return [] if JSON.parse(response.body).empty?
+
+    JSON.parse(response.body)['cachedContents'].map do |item|
+      def item.delete = GeminiCache::Cache.delete(name: self['name'])
+      def item.set_ttl(ttl = 120) = GeminiCache::Cache.update(name: self['name'], content: { ttl: "#{ttl}s" })
+
+      def item.generate_content(contents:)
+        conn = Faraday.new(
+          url: 'https://generativelanguage.googleapis.com',
+          headers: { 'Content-Type' => 'application/json' }
+        ) do |f|
+          f.options.timeout = 300        # timeout em segundos para a requisição completa
+          f.options.open_timeout = 300   # timeout em segundos para abrir a conexão
+        end
+
+        response = conn.post("/v1beta/models/#{self['model'].split('/').last}:generateContent") do |req|
+          req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+          req.body = {
+            cached_content: self['name'],
+            contents:
+          }.to_json
+        end
+        
+        if response.status == 200
+          resp = JSON.parse(response.body)
+          def resp.content = dig('candidates', 0, 'content', 'parts', 0, 'text')
+          return resp
+        end
+
+        raise "Erro ao gerar conteúdo: #{response.body}"
+      rescue Faraday::Error => e
+        raise "Erro na requisição: #{e.message}"
+      end
+      
+      def item.single_prompt(prompt:) = generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }]).content
+
+      item
+    end
+
+  rescue Faraday::Error => e
+    raise "Erro na requisição: #{e.message}"
+  end
+  
+  def self.update(name:, content:)
+    conn = Faraday.new(
+      url: 'https://generativelanguage.googleapis.com',
+      headers: { 'Content-Type' => 'application/json' }
+    )
+
+    response = conn.patch("/v1beta/#{name}") do |req|
+      req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+      req.body = content.to_json
+    end
+
+    return JSON.parse(response.body) if response.status == 200
+    
+    raise "Erro ao atualizar cache: #{response.body}"
+  rescue Faraday::Error => e
+    raise "Erro na requisição: #{e.message}"
+  end
+
+  def self.delete(name:)
+    conn = Faraday.new(
+      url: 'https://generativelanguage.googleapis.com',
+      headers: { 'Content-Type' => 'application/json' }
+    )
+
+    response = conn.delete("/v1beta/#{name}") do |req|
+      req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+    end
+
+    return true if response.status == 200
+    
+    raise "Erro ao deletar cache: #{response.body}"
+  rescue Faraday::Error => e
+    raise "Erro na requisição: #{e.message}"
+  end
+end

data/lib/geminiext/cache.rb

@@ -0,0 +1,127 @@
+require 'faraday'
+require 'json'
+
+module GeminiExt
+  module Cache
+    def self.create(contents:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+      content = {
+        model: "models/#{model}",
+        display_name:,
+        contents:,
+        ttl: "#{ttl}s"
+      }.to_json
+    
+      conn = Faraday.new(
+        url: 'https://generativelanguage.googleapis.com',
+        headers: { 'Content-Type' => 'application/json' }
+      )
+    
+      response = conn.post('/v1beta/cachedContents') do |req|
+        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+        req.body = content
+      end
+    
+      return JSON.parse(response.body) if response.status == 200
+    
+      raise "Erro ao criar cache: #{response.status} - #{response.body}"
+    rescue Faraday::Error => e
+      raise "Erro na requisição: #{e.message}"
+    end
+
+    def self.get(name: nil, display_name: nil)
+      raise 'Nome do cache ou display name é obrigatório' if name.nil? && display_name.nil?
+      raise 'Nome do cache e display name não podem ser informados juntos' if !name.nil? && !display_name.nil?
+      
+      return GeminiExt::Cache.list.find { |item| item['name'].eql? name } if !name.nil?
+      return GeminiExt::Cache.list.find { |item| item['displayName'].eql? display_name } if !display_name.nil?
+    end
+
+    def self.list
+      conn = Faraday.new(
+        url: 'https://generativelanguage.googleapis.com',
+        headers: { 'Content-Type' => 'application/json' }
+      )
+
+      response = conn.get("/v1beta/cachedContents") do |req|
+        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+      end
+      
+      return [] if JSON.parse(response.body).empty?
+
+      JSON.parse(response.body)['cachedContents'].map do |item|
+        def item.delete = GeminiExt::Cache.delete(name: self['name'])
+        def item.set_ttl(ttl = 120) = GeminiExt::Cache.update(name: self['name'], content: { ttl: "#{ttl}s" })
+
+        def item.generate_content(contents:)
+          conn = Faraday.new(
+            url: 'https://generativelanguage.googleapis.com',
+            headers: { 'Content-Type' => 'application/json' }
+          ) do |f|
+            f.options.timeout = 300        # timeout em segundos para a requisição completa
+            f.options.open_timeout = 300   # timeout em segundos para abrir a conexão
+          end
+
+          response = conn.post("/v1beta/models/#{self['model'].split('/').last}:generateContent") do |req|
+            req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+            req.body = {
+              cached_content: self['name'],
+              contents:
+            }.to_json
+          end
+          
+          if response.status == 200
+            resp = JSON.parse(response.body)
+            def resp.content = dig('candidates', 0, 'content', 'parts', 0, 'text')
+            return resp
+          end
+
+          raise "Erro ao gerar conteúdo: #{response.body}"
+        rescue Faraday::Error => e
+          raise "Erro na requisição: #{e.message}"
+        end
+        
+        def item.single_prompt(prompt: ) = generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }])
+
+        item
+      end
+
+    rescue Faraday::Error => e
+      raise "Erro na requisição: #{e.message}"
+    end
+    
+    def self.update(name:, content:)
+      conn = Faraday.new(
+        url: 'https://generativelanguage.googleapis.com',
+        headers: { 'Content-Type' => 'application/json' }
+      )
+
+      response = conn.patch("/v1beta/#{name}") do |req|
+        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+        req.body = content.to_json
+      end
+
+      return JSON.parse(response.body) if response.status == 200
+      
+      raise "Erro ao atualizar cache: #{response.body}"
+    rescue Faraday::Error => e
+      raise "Erro na requisição: #{e.message}"
+    end
+
+    def self.delete(name:)
+      conn = Faraday.new(
+        url: 'https://generativelanguage.googleapis.com',
+        headers: { 'Content-Type' => 'application/json' }
+      )
+
+      response = conn.delete("/v1beta/#{name}") do |req|
+        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+      end
+
+      return true if response.status == 200
+      
+      raise "Erro ao deletar cache: #{response.body}"
+    rescue Faraday::Error => e
+      raise "Erro na requisição: #{e.message}"
+    end
+  end
+end

data/lib/geminiext/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/geminiext/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/geminiext/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,61 @@
+--- !ruby/object:Gem::Specification
+name: gemini_cache
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Gedean Dias
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-11-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+description: Ruby's Gemini Context Caching wrapper
+email: gedean.dias@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/gemini_cache.rb
+- lib/geminiext/cache.rb
+- lib/geminiext/messages.rb.disabled
+- lib/geminiext/model.rb.disabled
+- lib/geminiext/response_extender.rb.disabled
+homepage: https://github.com/gedean/gemini_cache
+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 Context Caching
+test_files: []