Spaces:
Running
Running
Asish Karthikeya Gogineni commited on
Commit ·
8e40df0
1
Parent(s): 937fe82
fix: Add models/ prefix to Gemini model names
Browse files- 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=
|
| 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
|