geminiext 0.0.2 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8584393659f6094698f8efc8f19d5bb2c1a1608641614cc2bbbf69734853328a
-  data.tar.gz: 586b5fb85e7b7074ead525b376379d79e01a471facf54c009b7575eb2dbabe10
+  metadata.gz: 06c950bcdc066e3a29b824226c9ce351a3555aece1cb6169526e7a350124ce98
+  data.tar.gz: a60ec80c78df74698c81bfbf5c009be396255ba51cc772d5541947e36c9c8db3
 SHA512:
-  metadata.gz: 20fc9f1d721df29f89f419e6ac9e9d76af4220b1e901e548562681f6d612f557f59e1a8415bb1a013575da14f055026a838b995096e2fce2bf566d4831b53731
-  data.tar.gz: 4d7b0de55589e664125455afc6c5a6a4fac115e610f4faa271a770e692b4ebbdd96e3a100c993fc2aeeadc9b93ce2dcd59152373ce0e6e8ee5eab425611091c3
+  metadata.gz: be97a24aa8df7fff3a54d53c0c236460db69910bdeb9b2760b9bb6099610f19870d7ed36c9626fb2ff6ef48d95b7f26207a7ceeb5d4df99c79e43440dbd9409e
+  data.tar.gz: 138f4a593288ae8593bf91900101bba8e37c0bbc2cadd48b82122c07d9ec1e67cb47896862d3fc4f77ad8c6cd64b3e2cb511a86017933f290bb52ebeedd890ac

data/lib/geminiext/cache.rb

@@ -28,6 +28,14 @@ module GeminiExt
       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',
@@ -37,8 +45,41 @@ module GeminiExt
       response = conn.get("/v1beta/cachedContents") do |req|
         req.params['key'] = ENV.fetch('GEMINI_API_KEY')
       end
+      
+      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' }
+          )
+
+          response = conn.post("/v1beta/models/#{self['model'].split('/').last}:generateContent") do |req|
+            req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+            req.body = {
+              contents:,
+              cached_content: self['name']
+            }.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
 
-      JSON.parse(response.body)
     rescue Faraday::Error => e
       raise "Erro na requisição: #{e.message}"
     end
@@ -51,7 +92,7 @@ module GeminiExt
 
       response = conn.patch("/v1beta/#{name}") do |req|
         req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-        req.body = content
+        req.body = content.to_json
       end
 
       return JSON.parse(response.body) if response.status == 200

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: geminiext
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.5
 platform: ruby
 authors:
 - Gedean Dias