Spaces:
Running
Running
Update modules/api.py
Browse files- 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(
|