akra35567 commited on
Commit
666a186
·
1 Parent(s): 10f586c

Update modules/api.py

Browse files
Files changed (1) hide show
  1. modules/api.py +1 -1
modules/api.py CHANGED
@@ -151,7 +151,7 @@ class LLMManager:
151
  gemini_hist = []
152
  for msg in messages[1:]:
153
  role = "user" if msg["role"] == "user" else "model"
154
- gemini_hist.append({"role": role, "parts": [{"text": msg["content"}]})
155
  resp = self.gemini_model.generate_content(
156
  gemini_hist,
157
  generation_config=genai.GenerationConfig(
 
151
  gemini_hist = []
152
  for msg in messages[1:]:
153
  role = "user" if msg["role"] == "user" else "model"
154
+ gemini_hist.append({"role": role, "parts": [{"text": msg["content"]}]})
155
  resp = self.gemini_model.generate_content(
156
  gemini_hist,
157
  generation_config=genai.GenerationConfig(