From 8d113bf0aa83abecc28e1f9d530c113171d51be6 Mon Sep 17 00:00:00 2001
From: somebody <onesome01@protonmail.com>
Date: Mon, 19 Dec 2022 20:22:33 -0600
Subject: [PATCH] Handle no action id

---
 aiserver.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index 28a8a480..f0c22d7f 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -9598,7 +9598,10 @@ def generate_story_image(
         log_image_generation(prompt, display_prompt, file_name, generation_type, log_data)
         #let's also add this data to the action so we know where the latest picture is at
         logger.info("setting picture filename")
-        koboldai_vars.actions.set_picture(int(log_data['actionId']), file_name, prompt)
+        try:
+            koboldai_vars.actions.set_picture(int(log_data['actionId']), file_name, prompt)
+        except KeyError:
+            pass
 
     logger.debug("Time to Generate Image {}".format(time.time()-start_time))