kevinandClaude
d57dff58f3
继续之前的重构：把根目录残留的非数据/资源类子目录（agents、agenttool、ai、 autoreply、completion、conversation、llm、message、mqtpp、stream、 toolmanager、toolrouter）一并搬到 internal/ 下，并把全工程引用它们的 import 路径从 "meshtastic_mqtt_server/<x>" 重写为 "meshtastic_mqtt_server/internal/<x>"。 变更 - git mv 12 个顶层目录进 internal/，无函数体改动。 - 全工程 sed 把 import path 加上 internal/ 前缀，包括 main.go、 internal/bot/bot_service.go、internal/store/bot_store.go 中对 mqtpp 的引用，以及 ai 子系统内部 agents/agenttool/autoreply/conversation/ llm/toolmanager/toolrouter 之间的相互引用。 验证 - go build ./... 通过；go test ./... 全部包通过。 - AI 自动回复链路（main → ai.NewService → autoreply.Service → botSender） 保持不变，仅 import 路径调整。 - 根目录最终只剩 main.go / main_test.go / internal/ 加 go.mod / go.sum 与数据资源目录（dist、doc、firmware、py、win、meshmap_frontend）。 Co-Authored-By: Claude <noreply@anthropic.com>
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package toolrouter
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"meshtastic_mqtt_server/internal/llm"
|
|
)
|
|
|
|
// Config holds the tool router configuration
|
|
type Config struct {
|
|
Enabled bool
|
|
OpenAIName string
|
|
Timeout int
|
|
MaxTokens int
|
|
SystemPrompt string
|
|
}
|
|
|
|
// State manages the tool router state
|
|
type State struct {
|
|
cfg *Config
|
|
ai *llm.State
|
|
}
|
|
|
|
// Option is a function that configures the State
|
|
type Option func(*State)
|
|
|
|
// NewState creates a new tool router state
|
|
func NewState(cfg *Config, ai *llm.State, options ...Option) (*State, error) {
|
|
if cfg == nil {
|
|
cfg = &Config{
|
|
Enabled: true,
|
|
Timeout: 30,
|
|
MaxTokens: 512,
|
|
SystemPrompt: "你可以按需直接调用可用工具来回答用户问题。\n每个工具的 description 描述了它的适用场景和调用条件。\n工具结果优先于模型内置知识；工具失败时必须如实说明，不要编造结果。\n只调用确实必要的工具。",
|
|
}
|
|
}
|
|
if ai == nil {
|
|
return nil, errors.New("tool router requires an LLM state")
|
|
}
|
|
if cfg.Enabled && strings.TrimSpace(cfg.OpenAIName) != "" {
|
|
if _, err := ai.GetProfile(cfg.OpenAIName); err != nil {
|
|
return nil, fmt.Errorf("invalid LLM provider name in tool router: %w", err)
|
|
}
|
|
}
|
|
state := &State{cfg: cfg, ai: ai}
|
|
for _, option := range options {
|
|
option(state)
|
|
}
|
|
return state, nil
|
|
}
|
|
|
|
// RouterProfile returns the LLM profile configured for the tool router
|
|
func (s *State) RouterProfile(fallback *llm.Profile) *llm.Profile {
|
|
if s == nil || s.cfg == nil || s.ai == nil {
|
|
return fallback
|
|
}
|
|
name := strings.TrimSpace(s.cfg.OpenAIName)
|
|
if name == "" {
|
|
return fallback
|
|
}
|
|
profile, err := s.ai.GetProfile(name)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return profile
|
|
}
|
|
|
|
// Config returns a copy of the current configuration
|
|
func (s *State) Config() Config {
|
|
if s == nil || s.cfg == nil {
|
|
return Config{}
|
|
}
|
|
return *s.cfg
|
|
}
|