Class: AgentHarness::Providers::Gemini

Inherits:
Base
  • Object
show all
Defined in:
lib/agent_harness/providers/gemini.rb

Overview

Google Gemini CLI provider

Provides integration with the Google Gemini CLI tool.

Constant Summary collapse

MODEL_PATTERN =

Model name pattern for Gemini models

/^gemini-[\d.]+-(?:pro|flash|ultra)(?:-\d+)?$/i

Constants inherited from Base

Base::COMMON_ERROR_PATTERNS

Instance Attribute Summary

Attributes inherited from Base

#config, #executor, #logger

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Base

#configure, #initialize, #sandboxed_environment?, #send_message

Methods included from Adapter

#build_mcp_flags, #dangerous_mode_flags, #fetch_mcp_servers, included, #parse_rate_limit_reset, #send_message, #session_flags, #supported_mcp_transports, #supports_dangerous_mode?, #supports_mcp?, #supports_sessions?, #validate_mcp_servers!

Constructor Details

This class inherits a constructor from AgentHarness::Providers::Base

Class Method Details

.available? ⇒ Boolean

Returns:

  • (Boolean) 


24
25
26
27
 
# File 'lib/agent_harness/providers/gemini.rb', line 24

def available?
  executor = AgentHarness.configuration.command_executor
  !!executor.which(binary_name)
end

.binary_name ⇒ Object 



20
21
22
 
# File 'lib/agent_harness/providers/gemini.rb', line 20

def binary_name
  "gemini"
end

.discover_models ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/agent_harness/providers/gemini.rb', line 51

def discover_models
  return [] unless available?

  # Gemini CLI doesn't have a standard model listing command
  # Return common models
  [
    {name: "gemini-2.0-flash", family: "gemini-2-0-flash", tier: "standard", provider: "gemini"},
    {name: "gemini-2.5-pro", family: "gemini-2-5-pro", tier: "advanced", provider: "gemini"},
    {name: "gemini-1.5-pro", family: "gemini-1-5-pro", tier: "standard", provider: "gemini"},
    {name: "gemini-1.5-flash", family: "gemini-1-5-flash", tier: "mini", provider: "gemini"}
  ]
end

.firewall_requirements ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/agent_harness/providers/gemini.rb', line 29

def firewall_requirements
  {
    domains: [
      "generativelanguage.googleapis.com",
      "oauth2.googleapis.com",
      "accounts.google.com",
      "www.googleapis.com"
    ],
    ip_ranges: []
  }
end

.instruction_file_paths ⇒ Object 



41
42
43
44
45
46
47
48
49
 
# File 'lib/agent_harness/providers/gemini.rb', line 41

def instruction_file_paths
  [
    {
      path: "GEMINI.md",
      description: "Google Gemini agent instructions",
      symlink: true
    }
  ]
end

.model_family(provider_model_name) ⇒ Object 



64
65
66
67
 
# File 'lib/agent_harness/providers/gemini.rb', line 64

def model_family(provider_model_name)
  # Strip version suffix: "gemini-1.5-pro-001" -> "gemini-1.5-pro"
  provider_model_name.sub(/-\d+$/, "")
end

.provider_model_name(family_name) ⇒ Object 



69
70
71
 
# File 'lib/agent_harness/providers/gemini.rb', line 69

def provider_model_name(family_name)
  family_name
end

.provider_name ⇒ Object 



16
17
18
 
# File 'lib/agent_harness/providers/gemini.rb', line 16

def provider_name
  :gemini
end

.supports_model_family?(family_name) ⇒ Boolean

Returns:

  • (Boolean) 


73
74
75
 
# File 'lib/agent_harness/providers/gemini.rb', line 73

def supports_model_family?(family_name)
  MODEL_PATTERN.match?(family_name) || family_name.start_with?("gemini-")
end

Instance Method Details

#auth_status ⇒ Object 



139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
 
# File 'lib/agent_harness/providers/gemini.rb', line 139

def auth_status
  api_key = [ENV["GEMINI_API_KEY"], ENV["GOOGLE_API_KEY"]].find { |key| key && !key.strip.empty? }
  if api_key
    return {valid: true, expires_at: nil, error: nil, auth_method: :api_key}
  end

  credentials = read_gemini_credentials
  return {valid: false, expires_at: nil, error: "No Gemini credentials found. Run 'gemini auth login' or set GEMINI_API_KEY or GOOGLE_API_KEY"} unless credentials

  token = credentials["access_token"] || credentials["oauth_token"]
  unless token.is_a?(String) && !token.strip.empty?
    return {valid: false, expires_at: nil, error: "No authentication token in Gemini credentials"}
  end

  expires_at = parse_gemini_expiry(credentials)
  if expires_at && expires_at < Time.now
    {valid: false, expires_at: expires_at, error: "Gemini session expired. Run 'gemini auth login' to re-authenticate"}
  else
    {valid: true, expires_at: expires_at, error: nil, auth_method: :oauth}
  end
rescue IOError, JSON::ParserError => e
  {valid: false, expires_at: nil, error: e.message}
end

#auth_type ⇒ Object 



98
99
100
 
# File 'lib/agent_harness/providers/gemini.rb', line 98

def auth_type
  :oauth
end

#capabilities ⇒ Object 



86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/agent_harness/providers/gemini.rb', line 86

def capabilities
  {
    streaming: true,
    file_upload: true,
    vision: true,
    tool_use: true,
    json_mode: true,
    mcp: false,
    dangerous_mode: false
  }
end

#display_name ⇒ Object 



82
83
84
 
# File 'lib/agent_harness/providers/gemini.rb', line 82

def display_name
  "Google Gemini"
end

#error_patterns ⇒ Object 



115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/agent_harness/providers/gemini.rb', line 115

def error_patterns
  {
    rate_limited: [
      /rate.?limit/i,
      /quota.?exceeded/i,
      /429/
    ],
    auth_expired: [
      /authentication/i,
      /unauthorized/i,
      /invalid.?credentials/i,
      /login.*required/i,
      /not.*logged.*in/i,
      /credentials.*expired/i,
      /account.*not.*verified/i
    ],
    transient: [
      /timeout/i,
      /temporary/i,
      /503/
    ]
  }
end

#execution_semantics ⇒ Object 



102
103
104
105
106
107
108
109
110
111
112
113
 
# File 'lib/agent_harness/providers/gemini.rb', line 102

def execution_semantics
  {
    prompt_delivery: :flag,
    output_format: :text,
    sandbox_aware: false,
    uses_subcommand: false,
    non_interactive_flag: nil,
    legitimate_exit_codes: [0],
    stderr_is_diagnostic: true,
    parses_rate_limit_reset: false
  }
end

#health_status ⇒ Object 



163
164
165
166
167
168
169
170
171
172
173
174
 
# File 'lib/agent_harness/providers/gemini.rb', line 163

def health_status
  unless self.class.available?
    return {healthy: false, message: "Gemini CLI not found in PATH. Install from https://github.com/google-gemini/gemini-cli"}
  end

  auth = auth_status
  unless auth[:valid]
    return {healthy: false, message: auth[:error]}
  end

  {healthy: true, message: "Gemini CLI available and authenticated"}
end

#name ⇒ Object 



78
79
80
 
# File 'lib/agent_harness/providers/gemini.rb', line 78

def name
  "gemini"
end

#validate_config ⇒ Object 



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
 
# File 'lib/agent_harness/providers/gemini.rb', line 176

def validate_config
  errors = []

  model = @config.model
  if !model.nil? && !model.is_a?(String)
    errors << "model must be a string"
  elsif model.is_a?(String) && !model.empty?
    unless self.class.supports_model_family?(model)
      errors << "Unrecognized model '#{model}'. Expected a Gemini model (e.g., gemini-2.0-flash, gemini-2.5-pro)"
    end
  end

  flags = @config.default_flags
  unless flags.nil?
    if flags.is_a?(Array)
      invalid = flags.reject { |f| f.is_a?(String) }
      errors << "default_flags contains non-string values" if invalid.any?
    else
      errors << "default_flags must be an array of strings"
    end
  end

  {valid: errors.empty?, errors: errors}
end