@@ -55,7 +55,7 @@ public static ChatCompletionRequest.ChatCompletionRequestBuilder getCompletionRe
55
55
return completionRequestBuilder ;
56
56
}
57
57
58
- public static List <ChatMessage > getPrompt (String sessionId , String newPrompt ) {
58
+ public static List <ChatMessage > buildPrompt (String sessionId , String newPrompt ) {
59
59
if (!PROMPT_MAP .containsKey (sessionId )) {
60
60
if (null != botConfig .getBasicPrompt ()){
61
61
List <ChatMessage > promptList = new ArrayList <>();
@@ -72,23 +72,30 @@ public static void updatePrompt(String sessionId, List<ChatMessage> promptList)
72
72
PROMPT_MAP .put (sessionId , promptList );
73
73
}
74
74
75
- public static void deleteFirstPrompt (String sessionId ) {
76
- if (PROMPT_MAP .containsKey (sessionId )){
75
+ public static boolean isPromptEmpty (String sessionId ){
76
+ if (!PROMPT_MAP .containsKey (sessionId )){
77
+ return true ;
78
+ }
79
+ List <ChatMessage > promptList = PROMPT_MAP .get (sessionId );
80
+ if (null != botConfig .getBasicPrompt ()){
81
+ return promptList .size () == 1 ;
82
+ }else {
83
+ return promptList .size () == 0 ;
84
+ }
85
+ }
86
+
87
+ public static boolean deleteFirstPrompt (String sessionId ) {
88
+ if (!isPromptEmpty (sessionId )){
89
+ int index = null != botConfig .getBasicPrompt () ? 1 : 0 ;
77
90
List <ChatMessage > promptList = PROMPT_MAP .get (sessionId );
78
- //有basicPrompt删除第二个
79
- if (null != botConfig .getBasicPrompt ()){
80
- if (promptList .size () > 1 ){
81
- promptList .remove (1 );
82
- updatePrompt (sessionId , promptList );
83
- }
84
- }else {
85
- //没有basicPrompt删除第一个
86
- if (promptList .size () > 0 ){
87
- promptList .remove (0 );
88
- updatePrompt (sessionId , promptList );
89
- }
90
- }
91
+ //问
92
+ promptList .remove (index );
93
+ //答
94
+ promptList .remove (index );
95
+ updatePrompt (sessionId , promptList );
96
+ return true ;
91
97
}
98
+ return false ;
92
99
}
93
100
94
101
public static void resetPrompt (String sessionId ) {
0 commit comments