Sivan Ratson
commited on
Commit
·
0fa707e
1
Parent(s):
c1f1b14
upgrade to gemini-flash-2.0-exp
Browse files- llm_providers.py +1 -1
llm_providers.py
CHANGED
@@ -22,7 +22,7 @@ class GeminiResponse:
|
|
22 |
class GeminiProvider:
|
23 |
def __init__(self, api_key: str):
|
24 |
self.api_key = api_key
|
25 |
-
self.base_url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-
|
26 |
|
27 |
def chat(self, messages: List[Dict[str, Any]]) -> GeminiResponse:
|
28 |
# Convert messages to Gemini format
|
|
|
22 |
class GeminiProvider:
|
23 |
def __init__(self, api_key: str):
|
24 |
self.api_key = api_key
|
25 |
+
self.base_url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent"
|
26 |
|
27 |
def chat(self, messages: List[Dict[str, Any]]) -> GeminiResponse:
|
28 |
# Convert messages to Gemini format
|