Files
agent-desk-ng/internal/ai/runtime/graphs/hitl.go

102 lines
3.2 KiB
Go

package graphs
import (
"strings"
"agent-desk/internal/pkg/i18nx"
)
const (
InterruptTypeTicketCreationConfirmation = "ticket_creation_confirmation"
InterruptTypeHandoffConfirmation = "handoff_confirmation"
)
var (
ConfirmOrCancelPrompt = i18nx.Get("graph.confirmOrCancel")
NeedExplicitConfirmationPrompt = i18nx.Get("graph.needExplicitConfirmation")
ConfirmationExpiredReply = i18nx.Get("graph.confirmationExpired")
CancelCreateTicketReply = i18nx.Get("graph.cancelCreateTicket")
CancelHandoffReply = i18nx.Get("graph.cancelHandoff")
)
func ConfirmOrCancelPromptForLocale(locale string) string {
if strings.TrimSpace(locale) == "" {
locale = i18nx.DefaultLocale
}
return i18nx.Getf(locale, "graph.confirmOrCancel")
}
func NeedExplicitConfirmationPromptForLocale(locale string) string {
if strings.TrimSpace(locale) == "" {
locale = i18nx.DefaultLocale
}
return i18nx.Getf(locale, "graph.needExplicitConfirmation")
}
func ConfirmationExpiredReplyForLocale(locale string) string {
if strings.TrimSpace(locale) == "" {
locale = i18nx.DefaultLocale
}
return i18nx.Getf(locale, "graph.confirmationExpired")
}
func CancelCreateTicketReplyForLocale(locale string) string {
if strings.TrimSpace(locale) == "" {
locale = i18nx.DefaultLocale
}
return i18nx.Getf(locale, "graph.cancelCreateTicket")
}
func CancelHandoffReplyForLocale(locale string) string {
if strings.TrimSpace(locale) == "" {
locale = i18nx.DefaultLocale
}
return i18nx.Getf(locale, "graph.cancelHandoff")
}
type ConfirmationDecision string
const (
ConfirmationDecisionConfirm ConfirmationDecision = "confirm"
ConfirmationDecisionCancel ConfirmationDecision = "cancel"
)
func ParseConfirmationDecision(value string) ConfirmationDecision {
value = strings.ToLower(strings.TrimSpace(value))
if value == "" {
return ""
}
cancelWords := []string{
"取消", "不用", "不需要", "算了", "no", "cancel", "stop",
"нет", "отмена", "отменить", "не надо", "не нужно", "стоп", "хватит", "отбой",
}
for _, item := range cancelWords {
if strings.Contains(value, item) {
return ConfirmationDecisionCancel
}
}
confirmWords := []string{
"确认", "是", "好的", "可以", "ok", "yes", "continue", "confirm", "继续", "同意",
"да", "подтверждаю", "подтвердить", "переводи", "переведи", "давай",
"согласен", "согласна", "верно", "точно", "конечно",
}
for _, item := range confirmWords {
if strings.Contains(value, item) {
return ConfirmationDecisionConfirm
}
}
return ""
}
func IsCancellationReply(replyText string) bool {
replyText = strings.TrimSpace(replyText)
return strings.Contains(replyText, CancelCreateTicketReply) ||
strings.Contains(replyText, CancelHandoffReply) ||
strings.Contains(replyText, "已取消本次工单创建。") ||
strings.Contains(replyText, "已取消本次转人工。") ||
strings.Contains(replyText, "Создание обращения отменено.") ||
strings.Contains(replyText, "Передача человеку отменена.") ||
strings.Contains(replyText, "Ticket creation cancelled.") ||
strings.Contains(replyText, "Human handoff cancelled.")
}