Merge pull request #36 from cisasteelersfan/patch-3

ruecat · Feb 24, 2024 · 40945d7 · 40945d7
2 parents 90c4dd8 + 21f2778
commit 40945d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bot/run.py b/bot/run.py
@@ -176,12 +176,12 @@ async def ollama_request(message: types.Message):
             if ACTIVE_CHATS.get(message.from_user.id) is None:
                 ACTIVE_CHATS[message.from_user.id] = {
                     "model": modelname,
-                    "messages": [{"role": "user", "content": prompt, "images": [image_base64]}],
+                    "messages": [{"role": "user", "content": prompt, "images": ([image_base64] if image_base64 else [])}],
                     "stream": True,
                 }
             else:
                 ACTIVE_CHATS[message.from_user.id]["messages"].append(
-                    {"role": "user", "content": prompt, "images": [image_base64]}
+                    {"role": "user", "content": prompt, "images": ([image_base64] if image_base64 else [])}
                 )
         logging.info(
             f"[Request]: Processing '{prompt}' for {message.from_user.first_name} {message.from_user.last_name}"