akra35567 commited on
Commit
1d7c959
·
1 Parent(s): 750cb61

Update modules/api.py

Browse files
Files changed (1) hide show
  1. modules/api.py +1 -1
modules/api.py CHANGED
@@ -142,7 +142,7 @@ class LLMManager:
142
  gemini_hist = []
143
  for msg in messages[1:]:
144
  role = "user" if msg["role"] == "user" else "model"
145
- gemini_hist.append({"role": role, "parts": [{"text": msg["content"}]})
146
  resp = self.gemini_model.generate_content(
147
  gemini_hist,
148
  generation_config=genai.GenerationConfig(
 
142
  gemini_hist = []
143
  for msg in messages[1:]:
144
  role = "user" if msg["role"] == "user" else "model"
145
+ gemini_hist.append({"role": role, "parts": [{"text": msg["content"]}]})
146
  resp = self.gemini_model.generate_content(
147
  gemini_hist,
148
  generation_config=genai.GenerationConfig(