Class: Payloop::Wrappers::GeminiAI

Inherits:
Object
  • Object
show all
Defined in:
lib/payloop/wrappers/geminiai.rb

Overview

Wrapper for Google GenerativeAI Ruby client (gemini-ai gem version 4.3.0)

Instance Method Summary collapse

Constructor Details

#initialize(config, collector, sentinel = nil) ⇒ GeminiAI

Returns a new instance of GeminiAI.



9
10
11
12
13
 
# File 'lib/payloop/wrappers/geminiai.rb', line 9

def initialize(config, collector, sentinel = nil)
  @config = config
  @collector = collector
  @sentinel = sentinel
end

Instance Method Details

#register(client) ⇒ Object 



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/payloop/wrappers/geminiai.rb', line 15

def register(client)
  validate_client!(client)

  # Prevent double registration
  return client if client.instance_variable_defined?(:@payloop_registered)

  # Store references in client instance
  client.instance_variable_set(:@payloop_config, @config)
  client.instance_variable_set(:@payloop_collector, @collector)
  client.instance_variable_set(:@payloop_sentinel, @sentinel)
  client.instance_variable_set(:@payloop_registered, true)

  # Wrap the generate_content and stream_generate_content methods
  wrap_generate_content_method(client)
  wrap_stream_generate_content_method(client)

  client
end