94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"myaibot/internal/bot"
|
|
)
|
|
|
|
type Handler struct {
|
|
bot *bot.Bot
|
|
}
|
|
|
|
func New(b *bot.Bot) *Handler {
|
|
return &Handler{bot: b}
|
|
}
|
|
|
|
func (h *Handler) Handle(input string) bool {
|
|
fields := strings.Fields(input)
|
|
cmd, args := fields[0], fields[1:]
|
|
switch cmd {
|
|
case "/exit", "/quit":
|
|
fmt.Println("再见！")
|
|
return false
|
|
case "/help":
|
|
fmt.Println("命令列表:")
|
|
fmt.Println(" /models 列出所有供应商和模型")
|
|
fmt.Println(" /use <模型> 切换模型，如 /use deepseek-chat 或 /use deepseek/deepseek-chat")
|
|
fmt.Println(" /think <on|off> 开启或关闭当前供应商的思考模式")
|
|
fmt.Println(" /effort <low|high|max> 设置思考强度")
|
|
fmt.Println(" /context 打印当前聊天上下文")
|
|
fmt.Println(" /tools 列出可用工具")
|
|
fmt.Println(" /info 显示当前供应商、模型和思考配置")
|
|
fmt.Println(" /exit 退出")
|
|
case "/models":
|
|
for _, m := range h.bot.Models() {
|
|
fmt.Println(" " + m)
|
|
}
|
|
case "/use":
|
|
if len(args) == 0 {
|
|
fmt.Println("用法: /use <模型>，如 /use deepseek-chat")
|
|
return true
|
|
}
|
|
if err := h.bot.SwitchModel(args[0]); err != nil {
|
|
fmt.Printf("⚠️ %v\n", err)
|
|
return true
|
|
}
|
|
provider, model := h.bot.Current()
|
|
fmt.Printf("已切换到 %s/%s (对话历史已保留)\n", provider, model)
|
|
case "/think":
|
|
if len(args) == 0 {
|
|
fmt.Println("用法: /think <on|off>")
|
|
return true
|
|
}
|
|
v := map[string]string{"on": "enabled", "off": "disabled"}[args[0]]
|
|
if err := h.bot.SetThinking(v); err != nil {
|
|
fmt.Printf("⚠️ %v\n", err)
|
|
return true
|
|
}
|
|
fmt.Printf("思考模式已%s\n", map[string]string{"enabled": "开启", "disabled": "关闭"}[v])
|
|
case "/effort":
|
|
if len(args) == 0 {
|
|
fmt.Println("用法: /effort <low|high|max>")
|
|
return true
|
|
}
|
|
if err := h.bot.SetEffort(args[0]); err != nil {
|
|
fmt.Printf("⚠️ %v\n", err)
|
|
return true
|
|
}
|
|
fmt.Printf("思考强度已设置为 %s\n", args[0])
|
|
case "/context":
|
|
fmt.Print(h.bot.ContextDump())
|
|
case "/tools":
|
|
for _, t := range h.bot.Tools() {
|
|
fmt.Println(" " + t)
|
|
}
|
|
case "/info":
|
|
provider, model := h.bot.Current()
|
|
thinking, effort := h.bot.ThinkingConfig()
|
|
tool, vision := h.bot.CurrentRoles()
|
|
if thinking == "" {
|
|
thinking = "enabled(默认)"
|
|
}
|
|
if effort == "" {
|
|
effort = "high(默认)"
|
|
}
|
|
fmt.Printf("供应商: %s, 模型: %s, 思考模式: %s, 思考强度: %s\n", provider, model, thinking, effort)
|
|
fmt.Printf("工具调用AI: %s\n图片识别AI: %s\n", tool, vision)
|
|
default:
|
|
fmt.Printf("未知命令: %s，输入 /help 查看命令列表\n", cmd)
|
|
}
|
|
return true
|
|
}
|