소스 검색

feat: optimize conversation title generation output must be a valid JSON (#1077)

takatost 1 년 전
부모
커밋
4c8608dc61
2개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 0
      api/core/prompt/prompts.py
  2. 1 1
      api/events/event_handlers/generate_conversation_name_when_first_message_created.py

+ 1 - 0
api/core/prompt/prompts.py

@@ -3,6 +3,7 @@ CONVERSATION_TITLE_PROMPT = """You need to decompose the user's input into "subj
 Notice: the language type user using is abundant, can be English, Chinese, Español, Arabic, Japanese, and etc.
 Notice: the language type user using is abundant, can be English, Chinese, Español, Arabic, Japanese, and etc.
 MAKE SURE your output is the SAME language as the user's input!
 MAKE SURE your output is the SAME language as the user's input!
 Your output is restricted only to: (Input language) Intention + Subject(short as possible)
 Your output is restricted only to: (Input language) Intention + Subject(short as possible)
+Your output MUST be a valid JSON.
 
 
 Tip: When the user's question is directed at you (the language model), you can add an emoji to make it more fun.
 Tip: When the user's question is directed at you (the language model), you can add an emoji to make it more fun.
 
 

+ 1 - 1
api/events/event_handlers/generate_conversation_name_when_first_message_created.py

@@ -26,7 +26,7 @@ def handle(sender, **kwargs):
 
 
                 conversation.name = name
                 conversation.name = name
             except:
             except:
-                conversation.name = 'New Chat'
+                conversation.name = 'New conversation'
 
 
             db.session.add(conversation)
             db.session.add(conversation)
             db.session.commit()
             db.session.commit()