gemini_api_ruby 1.0.3 → 1.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d836f7394ec2c0b798e2441386d427d53c29bee0d23e34e80a2f4303cce3525e
-  data.tar.gz: d2630739814cb78d6ef9ead4b64a2886c20da7183d7fb8129a5dc8d810b27175
+  metadata.gz: 547f60b4912b4a21c4f2faf3a204f437806a6250975994e3baad4d9df9734125
+  data.tar.gz: 8e239a07bc4c20a7aeca22afddd3ee2ec2f01bdc6145a437b16cbc7a48cd96e8
 SHA512:
-  metadata.gz: 93ca312338b03c83185837ff7787d02a8c85e57231f7014dae64cf73666004bd2f6673801916a65b6a4c1e85034410eb0a1bab8cbc3e5b998610594817b85975
-  data.tar.gz: '07609c4359bbf298c8254da60e8e9793c02aa83ade2b7ab43b17c6b8c87e6d906ccec3b4ad6a5fec8d66671934a9de0d63bd898184e46fadff487314605514bb'
+  metadata.gz: 1b3c3521cbb1b066478f4dcf64a9b023bdf190540f923ccae11cc1390a966f1aab60578251c1b3bb04925755cc65da7de90ee0af1b40e2618a6520e3f1553294
+  data.tar.gz: 3b5639015fb507862962f3e508606f75df26adb0ed3addd04088370d7d005ae9260a7cdff5840b52cdcc5b7f5ab7549bbd1c35a9ce7a19b044955722a9deaabe

data/lib/gemini_api_ruby/client.rb

@@ -1,25 +1,50 @@
+# lib/gemini_api_ruby/client.rb
+
 module GeminiApiRuby
   class Client
+    BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash".freeze
+
     def initialize(api_key:)
       @api_key = api_key
     end
 
-    def generate_text(prompt)
-      response = request(:post, "/v1/generate", { prompt: prompt })
-    end
-
-    def get_ticker(symbol)
-      request(:get, "/v1/pubticker/#{symbol}")
-    end
+    def generate_content(prompt)
+      endpoint = "#{BASE_URL}:generateContent?key=#{@api_key}"
+      payload = {
+        contents: [
+          {
+            parts: [
+              { text: prompt }
+            ]
+          }
+        ]
+      }
 
-    def get_order_book(symbol)
-      request(:get, "/v1/book/#{symbol}")
+      response = request(:post, endpoint, payload)
+      parse_response(response)
     end
 
     private
 
-    def request(method, endpoint, params = {})
-      Request.new(api_key: @api_key).send(method, endpoint, params)
+    def request(method, endpoint, payload)
+      connection = Faraday.new(url: endpoint) do |faraday|
+        faraday.request :json
+        faraday.response :json, content_type: /\bjson$/
+        faraday.adapter Faraday.default_adapter
+      end
+
+      connection.post do |req|
+        req.headers["Content-Type"] = "application/json"
+        req.body = payload.to_json
+      end
+    end
+
+    def parse_response(response)
+      if response.success?
+        response.body.dig("candidates", 0, "content", "parts", 0, "text") || "No text generated"
+      else
+        raise GeminiApiRuby::Error, "API request failed: #{response.body['error']['message']}"
+      end
     end
   end
-end
+end

data/lib/gemini_api_ruby/version.rb

@@ -1,3 +1,3 @@
 module GeminiApiRuby
-  VERSION = "1.0.3"
+  VERSION = "1.0.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gemini_api_ruby
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.5
 platform: ruby
 authors:
 - marcelodeus98