srishtichugh commited on
Commit
c7412fb
·
1 Parent(s): 0bd372a

fix gemini model prefix issue

Browse files
Files changed (1) hide show
  1. 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)