Class: GeminiApiRuby::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/gemini_api_ruby/client.rb

Constant Summary collapse

BASE_URL = 
"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash".freeze

Instance Method Summary collapse

Constructor Details

#initialize(api_key:) ⇒ Client

Returns a new instance of Client.



7
8
9
 
# File 'lib/gemini_api_ruby/client.rb', line 7

def initialize(api_key:)
  @api_key = api_key
end

Instance Method Details

#generate_content(prompt) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/gemini_api_ruby/client.rb', line 11

def generate_content(prompt)
  endpoint = "#{BASE_URL}:generateContent?key=#{@api_key}"
  payload = {
    contents: [
      {
        parts: [
          { text: prompt }
        ]
      }
    ]
  }

  response = request(:post, endpoint, payload)
  parse_response(response)
end