lanvent 3 лет назад
Родитель
Сommit
720de9d73f
2 измененных файлов с 5 добавлено и 2 удалено
  1. 3 0
      bot/session_manager.py
  2. 2 2
      channel/chat_channel.py

+ 3 - 0
bot/session_manager.py

@@ -50,6 +50,9 @@ class SessionManager(object):
             如果session_id不在sessions中,创建一个新的session并添加到sessions中
             如果system_prompt不会空,会更新session的system_prompt并重置session
         '''
+        if session_id is None:
+            return self.sessioncls(session_id, system_prompt, **self.session_args)
+        
         if session_id not in self.sessions:
             self.sessions[session_id] = self.sessioncls(session_id, system_prompt, **self.session_args)
         elif system_prompt is not None:  # 如果有新的system_prompt,更新并重置session

+ 2 - 2
channel/chat_channel.py

@@ -111,11 +111,11 @@ class ChatChannel(Channel):
                                                   
             img_match_prefix = check_prefix(content, conf().get('image_create_prefix'))
             if img_match_prefix:
-                content = content.replace(img_match_prefix, '', 1).strip()
+                content = content.replace(img_match_prefix, '', 1)
                 context.type = ContextType.IMAGE_CREATE
             else:
                 context.type = ContextType.TEXT
-            context.content = content
+            context.content = content.strip()
             if 'desire_rtype' not in context and conf().get('always_reply_voice') and ReplyType.VOICE not in self.NOT_SUPPORT_REPLYTYPE:
                 context['desire_rtype'] = ReplyType.VOICE
         elif context.type == ContextType.VOICE: