105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
|
package ai
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"strings"
|
|||
|
|
"unicode"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type langCtxKey struct{}
|
|||
|
|
|
|||
|
|
// WithResponseLanguage attaches UI/locale language to ctx for system prompt injection.
|
|||
|
|
func WithResponseLanguage(ctx context.Context, lang string) context.Context {
|
|||
|
|
if ctx == nil {
|
|||
|
|
ctx = context.Background()
|
|||
|
|
}
|
|||
|
|
return context.WithValue(ctx, langCtxKey{}, NormalizeResponseLanguage(lang))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ResponseLanguageFrom returns zh-TW | en (default zh-TW).
|
|||
|
|
func ResponseLanguageFrom(ctx context.Context) string {
|
|||
|
|
if ctx == nil {
|
|||
|
|
return "zh-TW"
|
|||
|
|
}
|
|||
|
|
if v, ok := ctx.Value(langCtxKey{}).(string); ok && v != "" {
|
|||
|
|
return NormalizeResponseLanguage(v)
|
|||
|
|
}
|
|||
|
|
return "zh-TW"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NormalizeResponseLanguage maps preferred_language / Accept-Language → zh-TW | en.
|
|||
|
|
func NormalizeResponseLanguage(raw string) string {
|
|||
|
|
s := strings.ToLower(strings.TrimSpace(raw))
|
|||
|
|
if s == "" {
|
|||
|
|
return "zh-TW"
|
|||
|
|
}
|
|||
|
|
if i := strings.IndexAny(s, ",;"); i >= 0 {
|
|||
|
|
s = strings.TrimSpace(s[:i])
|
|||
|
|
}
|
|||
|
|
s = strings.ReplaceAll(s, "_", "-")
|
|||
|
|
if s == "en" || strings.HasPrefix(s, "en-") {
|
|||
|
|
return "en"
|
|||
|
|
}
|
|||
|
|
// zh / zh-TW / zh-Hant / zh-CN → 產品主語系繁中
|
|||
|
|
return "zh-TW"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SystemPromptForLanguage is always injected as the first system message before user content.
|
|||
|
|
// 只約束語言，不要綁「產品助手」人設——否則仿寫／產文會像客服而不是本人在發 Threads。
|
|||
|
|
func SystemPromptForLanguage(lang string) string {
|
|||
|
|
switch NormalizeResponseLanguage(lang) {
|
|||
|
|
case "en":
|
|||
|
|
return strings.TrimSpace(`
|
|||
|
|
Language rule (mandatory):
|
|||
|
|
- Write in natural English unless the user explicitly asks for another language.
|
|||
|
|
- For JSON/structured output, keep human-readable string values in English.
|
|||
|
|
- Do not add meta commentary (e.g. "As an AI…") unless asked.
|
|||
|
|
`)
|
|||
|
|
default:
|
|||
|
|
return strings.TrimSpace(`
|
|||
|
|
語言規則（強制）：
|
|||
|
|
- 使用自然的「繁體中文（台灣用語）」，除非使用者明確要求其他語言。
|
|||
|
|
- 若要求 JSON／結構化輸出，給人看的字串也要用繁中。
|
|||
|
|
- 不要加「作為 AI…」「以下是分析…」這類 meta 前綴；直接完成任務。
|
|||
|
|
- 產文／仿寫時像真人滑手機在打字，不要像客服或文案機器人。
|
|||
|
|
`)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PickLanguage prefers member preferred_language, then Accept-Language header.
|
|||
|
|
func PickLanguage(preferredLanguage, acceptLanguage string) string {
|
|||
|
|
if strings.TrimSpace(preferredLanguage) != "" {
|
|||
|
|
return NormalizeResponseLanguage(preferredLanguage)
|
|||
|
|
}
|
|||
|
|
return NormalizeResponseLanguage(acceptLanguage)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InferScriptLanguage 從文本統計漢字 vs 拉丁字母，推 zh-TW | en；信號不足回 ""。
|
|||
|
|
// 用於人設指紋／範例：英文人物應回英文，不該被 UI 語系或「繁中」硬規則蓋掉。
|
|||
|
|
func InferScriptLanguage(texts ...string) string {
|
|||
|
|
var han, latin int
|
|||
|
|
for _, t := range texts {
|
|||
|
|
for _, r := range t {
|
|||
|
|
switch {
|
|||
|
|
case unicode.Is(unicode.Han, r):
|
|||
|
|
han++
|
|||
|
|
case (r >= 'A' && r <= 'Z') || (r >= 'a' && r <= 'z'):
|
|||
|
|
latin++
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
total := han + latin
|
|||
|
|
if total < 16 {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
// 拉丁明顯佔優 → 英文
|
|||
|
|
if latin >= han*2 && latin >= 16 {
|
|||
|
|
return "en"
|
|||
|
|
}
|
|||
|
|
// 漢字夠多 → 繁中
|
|||
|
|
if han >= 12 && han >= latin {
|
|||
|
|
return "zh-TW"
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|