Asish Karthikeya Gogineni commited on
Commit
8e40df0
·
1 Parent(s): 937fe82

fix: Add models/ prefix to Gemini model names

Browse files
Files changed (1) hide show
  1. code_chatbot/rag.py +6 -1
code_chatbot/rag.py CHANGED
@@ -118,8 +118,13 @@ class ChatEngine:
118
  if not os.getenv("GOOGLE_API_KEY"):
119
  raise ValueError("Google API Key is required for Gemini")
120
 
 
 
 
 
 
121
  return ChatGoogleGenerativeAI(
122
- model=self.model_name or "gemini-2.5-flash",
123
  google_api_key=api_key,
124
  temperature=0.2, # Low temp for agents
125
  convert_system_message_to_human=True
 
118
  if not os.getenv("GOOGLE_API_KEY"):
119
  raise ValueError("Google API Key is required for Gemini")
120
 
121
+ # Use models/ prefix for Gemini models
122
+ model_name = self.model_name or "gemini-1.5-flash"
123
+ if not model_name.startswith("models/"):
124
+ model_name = f"models/{model_name}"
125
+
126
  return ChatGoogleGenerativeAI(
127
+ model=model_name,
128
  google_api_key=api_key,
129
  temperature=0.2, # Low temp for agents
130
  convert_system_message_to_human=True