Skip to content

Commit acad28a

Browse files
Update model name handling and google default
1 parent b572e19 commit acad28a

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/git_bob/_endpoints.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,12 +163,13 @@ def prompt_deepseek(message: str, model="deepseek-chat", image=None, max_accumul
163163
return prompt_openai(message, model=model, image=image, max_accumulated_responses=max_accumulated_responses, max_response_tokens=max_response_tokens, base_url=base_url, api_key=api_key)
164164

165165

166-
def prompt_googleai(request, model="gemini-1.5-pro-002", image=None):
166+
def prompt_googleai(request, model="gemini-2.5-pro", image=None):
167167
"""Send a prompt to Google Gemini and return the response"""
168168
from google import generativeai as genai
169169
import os
170170

171171
model = model.replace("googleai:", "")
172+
model = model.replace("gemini:", "")
172173

173174
genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
174175
client = genai.GenerativeModel(model)

0 commit comments

Comments
 (0)