Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
chris1nexus
commited on
Commit
·
d52dc83
1
Parent(s):
af259c7
Add gemini
Browse files- src/adapter.py +8 -1
- 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 =
|
| 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'
|