Sivan Ratson commited on
Commit
0fa707e
·
1 Parent(s): c1f1b14

upgrade to gemini-flash-2.0-exp

Browse files
Files changed (1) hide show
  1. 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-1.5-flash:generateContent"
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