57 lines
1.9 KiB
Go
57 lines
1.9 KiB
Go
package newagentprompt
|
||
|
||
import (
|
||
"strings"
|
||
|
||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||
"github.com/cloudwego/eino/schema"
|
||
)
|
||
|
||
const chatIntentSystemPrompt = `
|
||
你是 SmartFlow 的意图分类器。
|
||
你的唯一任务是判断用户本轮输入是"纯闲聊"还是"包含任务意图"。
|
||
|
||
判断规则：
|
||
1. chat：打招呼、感谢、简单问答、情感表达、闲聊，不涉及任何具体任务或操作请求。
|
||
2. task：包含任何需要规划/执行/操作的意图，包括但不限于查询信息、创建内容、修改数据、安排日程、继续已有任务等。
|
||
|
||
保守原则：当不确定时，倾向于判断为 task，宁可多走一次规划也不要丢失用户意图。
|
||
|
||
严格输出以下 JSON（不要输出 markdown，不要在 JSON 外补文字）：
|
||
{"intent":"chat或task","reply":"仅当intent=chat时填写你的闲聊回复，task时留空","reason":"简短判断依据"}
|
||
`
|
||
|
||
// BuildChatIntentSystemPrompt 返回意图分类系统提示词。
|
||
func BuildChatIntentSystemPrompt() string {
|
||
return strings.TrimSpace(chatIntentSystemPrompt)
|
||
}
|
||
|
||
// BuildChatIntentMessages 组装意图分类的 messages。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只取最近 6 条历史，保证分类高效；
|
||
// 2. 不注入 pinned blocks / tool schemas，分类不需要这些信息；
|
||
// 3. 不负责解析模型输出。
|
||
func BuildChatIntentMessages(conversationContext *newagentmodel.ConversationContext, userInput string) []*schema.Message {
|
||
messages := make([]*schema.Message, 0, 8)
|
||
|
||
messages = append(messages, schema.SystemMessage(BuildChatIntentSystemPrompt()))
|
||
|
||
if conversationContext != nil {
|
||
history := conversationContext.HistorySnapshot()
|
||
if len(history) > 6 {
|
||
history = history[len(history)-6:]
|
||
}
|
||
if len(history) > 0 {
|
||
messages = append(messages, history...)
|
||
}
|
||
}
|
||
|
||
trimmedInput := strings.TrimSpace(userInput)
|
||
if trimmedInput != "" {
|
||
messages = append(messages, schema.UserMessage(trimmedInput))
|
||
}
|
||
|
||
return messages
|
||
}
|