fix the error when first regen

Signed-off-by: pengzhile <[email protected]>
SYingjie · Apr 23, 2023 · 26663b8 · 26663b8
1 parent bcd1b7f
commit 26663b8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/pandora/__init__.py b/src/pandora/__init__.py
@@ -1,3 +1,3 @@
 # -*- coding: utf-8 -*-
 
-__version__ = '1.0.4'
+__version__ = '1.0.5'
diff --git a/src/pandora/bots/server.py b/src/pandora/bots/server.py
@@ -251,11 +251,15 @@ def goon(self):
 
     def regenerate(self):
         payload = request.json
+
+        conversation_id = payload.get('conversation_id')
+        if not conversation_id:
+            return self.talk()
+
         prompt = payload['prompt']
         model = payload['model']
         message_id = payload['message_id']
         parent_message_id = payload['parent_message_id']
-        conversation_id = payload['conversation_id']
         stream = payload.get('stream', True)
 
         return self.__process_stream(