Handle None prompts

PhilippThoelke · Jul 12, 2024 · 15b9d22 · 15b9d22
1 parent 05ea84b
commit 15b9d22
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/goofi/nodes/inputs/textgeneration.py b/src/goofi/nodes/inputs/textgeneration.py
@@ -120,7 +120,7 @@ def generate_gemini_response(self, text, temp, keep_conversation):
         return response.text
 
     def process(self, prompt: Data):
-        if prompt.data is None:
+        if prompt is None or prompt.data is None:
             return None
 
         self.import_libs(self.params["text_generation"]["model"].value)