Spaces:
Sleeping
Sleeping
Commit ·
c7412fb
1
Parent(s): 0bd372a
fix gemini model prefix issue
Browse files- codes/llm_provider.py +2 -2
codes/llm_provider.py
CHANGED
|
@@ -103,9 +103,9 @@ class GeminiProvider(LLMProvider):
|
|
| 103 |
# Convert OpenAI message format to Gemini format
|
| 104 |
gemini_messages = self._convert_messages(messages)
|
| 105 |
|
| 106 |
-
# Fix model name - Gemini expects models/model-name format
|
| 107 |
if not model.startswith('models/'):
|
| 108 |
-
model = f'models/{model}'
|
| 109 |
|
| 110 |
# Create model
|
| 111 |
gemini_model = self.genai.GenerativeModel(model)
|
|
|
|
| 103 |
# Convert OpenAI message format to Gemini format
|
| 104 |
gemini_messages = self._convert_messages(messages)
|
| 105 |
|
| 106 |
+
"""# Fix model name - Gemini expects models/model-name format
|
| 107 |
if not model.startswith('models/'):
|
| 108 |
+
model = f'models/{model}'"""
|
| 109 |
|
| 110 |
# Create model
|
| 111 |
gemini_model = self.genai.GenerativeModel(model)
|