Class: GeminiClient

Inherits:
Object
  • Object
show all
Defined in:
lib/gemini_client.rb,
lib/gemini_client/version.rb

Defined Under Namespace

Modules: Method, Model Classes: Error

Constant Summary collapse

API_VERSION = 
"v1beta".freeze
VERSION = 
"0.1.0"

Instance Method Summary collapse

Constructor Details

#initialize(api_key:, version: API_VERSION, model: Model::GEMINI_1_5_FLASH) ⇒ GeminiClient

Returns a new instance of GeminiClient.



21
22
23
24
25
 
# File 'lib/gemini_client.rb', line 21

def initialize(api_key:, version: API_VERSION, model: Model::GEMINI_1_5_FLASH)
  @api_key = api_key
  @version = version
  @model = model
end

Instance Method Details

#count_tokens ⇒ Object 



36
37
38
 
# File 'lib/gemini_client.rb', line 36

def count_tokens
  # TODO
end

#embed_content ⇒ Object 



40
41
42
 
# File 'lib/gemini_client.rb', line 40

def embed_content
  # TODO
end

#generate_content(payload:, model: nil) ⇒ Object 



27
28
29
30
 
# File 'lib/gemini_client.rb', line 27

def generate_content(payload:, model: nil)
  url = api_url(api_key: @api_key, version: @version, model: model || @model, method_name: Method::GENERATE_CONTENT)
  request(:post, url, payload: JSON.dump(payload), headers: { "Content-Type" => "application/json" })
end

#model_info(model) ⇒ Object 



48
49
50
 
# File 'lib/gemini_client.rb', line 48

def model_info(model)
  # TODO
end

#models ⇒ Object 



44
45
46
 
# File 'lib/gemini_client.rb', line 44

def models
  # TODO
end

#stream_generate_content ⇒ Object 



32
33
34
 
# File 'lib/gemini_client.rb', line 32

def stream_generate_content
  # TODO
end