Class: Rasti::AI::Gemini::Client

Inherits:
Client
  • Object
show all
Defined in:
lib/rasti/ai/gemini/client.rb

Constant Summary

Constants inherited from Client

Client::RETRYABLE_STATUS_CODES

Instance Method Summary collapse

Methods inherited from Client

#initialize

Constructor Details

This class inherits a constructor from Rasti::AI::Client

Instance Method Details

#generate_content(contents:, model: nil, tools: [], system_instruction: nil, generation_config: nil) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/rasti/ai/gemini/client.rb', line 6

def generate_content(contents:, model:nil, tools:[], system_instruction:nil, generation_config:nil)
  model_name = model || Rasti::AI.gemini_default_model

  body = {contents: contents}

  body[:tools] = tools unless tools.empty?
  body[:system_instruction] = system_instruction unless system_instruction.nil?
  body[:generation_config] = generation_config unless generation_config.nil?

  post "/models/#{model_name}:generateContent", body
end