127 lines
4.8 KiB
Go
127 lines
4.8 KiB
Go
package ai
|
||
|
||
import (
|
||
"testing"
|
||
)
|
||
|
||
func TestIsCustomerHandoffRequest(t *testing.T) {
|
||
tests := []struct {
|
||
input string
|
||
expected bool
|
||
}{
|
||
// Russian varied phrasing (not just "переведи на оператора")
|
||
{"переведи на оператора", true},
|
||
{"переведите на оператора пожалуйста", true},
|
||
{"соедините со специалистом", true},
|
||
{"позови сотрудника", true},
|
||
{"позовите сотрудника поддержки", true},
|
||
{"мне нужен живой человек", true},
|
||
{"хочу поговорить с человеком", true},
|
||
{"хочу поговорить со специалистом", true},
|
||
{"хватит бота", true},
|
||
{"хватит робота, дайте человека", true},
|
||
{"не хочу общаться с ботом", true},
|
||
{"можно человека?", true},
|
||
{"есть кто живой?", true},
|
||
{"свяжите с оператором", true},
|
||
{"переключи на оператора", true},
|
||
{"дай оператора", true},
|
||
{"дай человека", true},
|
||
{"перенаправь оператору", true},
|
||
{"перенаправьте на специалиста", true},
|
||
{"тупой бот, дай человека", true},
|
||
{"нужен оператор", true},
|
||
{"нужен специалист", true},
|
||
{"оператор", true},
|
||
{"человек", true},
|
||
{"к оператору", true},
|
||
|
||
// English varied phrasing
|
||
{"can I speak to someone?", true},
|
||
{"connect me to an agent", true},
|
||
{"I want to talk to a real person", true},
|
||
{"let me speak with support", true},
|
||
{"transfer me to a representative", true},
|
||
{"representative please", true},
|
||
{"human please", true},
|
||
{"stop bot and get human", true},
|
||
{"live agent", true},
|
||
{"operator", true},
|
||
{"human", true},
|
||
|
||
// Other languages
|
||
{"necesito hablar con una persona", true},
|
||
{"quiero hablar con un humano", true},
|
||
{"mit einem Mitarbeiter sprechen", true},
|
||
{"转人工", true},
|
||
{"请帮我转接人工客服", true},
|
||
|
||
// Non-handoff questions & greetings (including mentions of operators without transfer intent)
|
||
{"Привет хочу возрат денег", false},
|
||
{"Здравствуйте, как вернуть товар?", false},
|
||
{"Добрый день, сколько стоит доставка?", false},
|
||
{"Hello, how do I track my order?", false},
|
||
{"Can I get a refund for my purchase?", false},
|
||
{"Операторы не создаются", false},
|
||
{"Операторы не создаються", false},
|
||
{"Кто такой оператор?", false},
|
||
{"Оператор работает ночью?", false},
|
||
{"Спасибо, всё понятно!", false},
|
||
{"OK, thank you very much", false},
|
||
{"", false},
|
||
{" ", false},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.input, func(t *testing.T) {
|
||
got := IsCustomerHandoffRequest(tt.input)
|
||
if got != tt.expected {
|
||
t.Errorf("IsCustomerHandoffRequest(%q) = %v, want %v", tt.input, got, tt.expected)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestIsModelClaimingHandoff(t *testing.T) {
|
||
tests := []struct {
|
||
input string
|
||
expected bool
|
||
}{
|
||
// Russian model claims
|
||
{"Похоже я не могу решить ваш вопрос, перенаправляю оператору. Ожидайте!", true},
|
||
{"Перевожу вас на оператора, пожалуйста подождите.", true},
|
||
{"Переключаю вас на оператора службы поддержки.", true},
|
||
{"Перенаправляю вас на специалиста.", true},
|
||
{"Соединяю со специалистом, оставайтесь на связи.", true},
|
||
{"Передаю ваш запрос оператору.", true},
|
||
{"Передаю диалог коллегам из поддержки.", true},
|
||
{"Сейчас переведу на оператора.", true},
|
||
|
||
// English model claims
|
||
{"Transferring you to an agent, please wait a moment.", true},
|
||
{"Connecting you to an operator right away.", true},
|
||
{"Connecting you to a human representative.", true},
|
||
{"Routing you to an agent.", true},
|
||
{"I am passing you to our support team.", true},
|
||
|
||
// Chinese model claims
|
||
{"正在为您转接人工客服，请稍候。", true},
|
||
{"为您转接人工客服处理。", true},
|
||
|
||
// Normal replies
|
||
{"Здравствуйте! Чем я могу вам помочь?", false},
|
||
{"Чтобы оформить возврат, перейдите в личный кабинет.", false},
|
||
{"To return an item, please open your order history.", false},
|
||
{"", false},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.input, func(t *testing.T) {
|
||
got := IsModelClaimingHandoff(tt.input)
|
||
if got != tt.expected {
|
||
t.Errorf("IsModelClaimingHandoff(%q) = %v, want %v", tt.input, got, tt.expected)
|
||
}
|
||
})
|
||
}
|
||
}
|