chris1nexus commited on
Commit
d52dc83
·
1 Parent(s): af259c7

Add gemini

Browse files
Files changed (2) hide show
  1. src/adapter.py +8 -1
  2. src/config.py +4 -1
src/adapter.py CHANGED
@@ -262,7 +262,7 @@ class GeminiAdapter(BaseAdapter):
262
  provider: str
263
  model: str
264
  api_key: Optional[str] = None
265
- timeout: float = 60.0
266
  base_url: Optional[str] = None
267
  extra_headers: Dict[str, str] = field(default_factory=dict)
268
 
@@ -287,6 +287,13 @@ class GeminiAdapter(BaseAdapter):
287
  if system:
288
  contents.insert(0, {"role": "system", "parts": [{"text": system}]})
289
  payload: Dict[str, Any] = {"contents": contents}
 
 
 
 
 
 
 
290
  r = requests.post(url, headers=headers, json=payload, timeout=self.timeout)
291
  self._raise_for_status(r)
292
  data = r.json()
 
262
  provider: str
263
  model: str
264
  api_key: Optional[str] = None
265
+ timeout: float = 300.0
266
  base_url: Optional[str] = None
267
  extra_headers: Dict[str, str] = field(default_factory=dict)
268
 
 
287
  if system:
288
  contents.insert(0, {"role": "system", "parts": [{"text": system}]})
289
  payload: Dict[str, Any] = {"contents": contents}
290
+ if 'gemini-3-pro' in self.model:
291
+ payload["generationConfig"] = {
292
+ "thinkingConfig": {
293
+ "thinkingLevel": "low"
294
+ }
295
+ }
296
+
297
  r = requests.post(url, headers=headers, json=payload, timeout=self.timeout)
298
  self._raise_for_status(r)
299
  data = r.json()
src/config.py CHANGED
@@ -31,6 +31,7 @@ GROK_MODELS = [
31
  ANTHROPIC_MODELS = [
32
  'claude-4-opus-20250514',
33
  'claude-opus-4-1-20250805',
 
34
  'claude-sonnet-4-5-20250929',
35
  'claude-haiku-4-5-20251001',
36
  'claude-4-sonnet-20250514']
@@ -43,7 +44,8 @@ GEMINI_MODELS = [
43
  'gemini-2.0-flash',
44
  'gemini-2.0-flash-lite',
45
  'gemini-2.5-flash',
46
- 'gemini-2.5-pro'
 
47
  ]
48
 
49
  OPENAI_MODELS = [
@@ -51,6 +53,7 @@ OPENAI_MODELS = [
51
  'gpt-4o-mini-2024-07-18',
52
  'gpt-4.5-preview-2025-02-27',
53
  'gpt-4.1-2025-04-14',
 
54
  'gpt-5-2025-08-07',
55
  'gpt-5-mini-2025-08-07',
56
  'gpt-5-nano-2025-08-07'
 
31
  ANTHROPIC_MODELS = [
32
  'claude-4-opus-20250514',
33
  'claude-opus-4-1-20250805',
34
+ 'claude-opus-4-5-20251101',
35
  'claude-sonnet-4-5-20250929',
36
  'claude-haiku-4-5-20251001',
37
  'claude-4-sonnet-20250514']
 
44
  'gemini-2.0-flash',
45
  'gemini-2.0-flash-lite',
46
  'gemini-2.5-flash',
47
+ 'gemini-2.5-pro',
48
+ 'gemini-3-pro-preview'
49
  ]
50
 
51
  OPENAI_MODELS = [
 
53
  'gpt-4o-mini-2024-07-18',
54
  'gpt-4.5-preview-2025-02-27',
55
  'gpt-4.1-2025-04-14',
56
+ 'gpt-5.1',
57
  'gpt-5-2025-08-07',
58
  'gpt-5-mini-2025-08-07',
59
  'gpt-5-nano-2025-08-07'