refactor: 将客服后端重构为宿主可嵌入模块
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
This commit is contained in:
t
+126
-27
@@ -1,13 +1,138 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
const SettingsPrefix = "ai_agent_"
|
||||
|
||||
func FromSettings(values map[string]string) (*Config, error) {
|
||||
cfg := defaultConfig()
|
||||
setString(values, SettingsPrefix+"language", &cfg.Language)
|
||||
setString(values, SettingsPrefix+"storage_default", (*string)(&cfg.Storage.Default))
|
||||
if err := setInt64(values, SettingsPrefix+"storage_max_upload_size_mb", &cfg.Storage.MaxUploadSizeMB); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
setString(values, SettingsPrefix+"storage_local_root", &cfg.Storage.Local.Root)
|
||||
setString(values, SettingsPrefix+"storage_local_base_url", &cfg.Storage.Local.BaseURL)
|
||||
setString(values, SettingsPrefix+"storage_oss_endpoint", &cfg.Storage.OSS.Endpoint)
|
||||
setString(values, SettingsPrefix+"storage_oss_bucket", &cfg.Storage.OSS.Bucket)
|
||||
setString(values, SettingsPrefix+"storage_oss_access_key_id", &cfg.Storage.OSS.AccessKeyID)
|
||||
setString(values, SettingsPrefix+"storage_oss_access_key_secret", &cfg.Storage.OSS.AccessKeySecret)
|
||||
setString(values, SettingsPrefix+"storage_oss_base_url", &cfg.Storage.OSS.BaseURL)
|
||||
if err := setBool(values, SettingsPrefix+"storage_oss_private", &cfg.Storage.OSS.Private); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := setInt(values, SettingsPrefix+"storage_oss_signed_url_expire", &cfg.Storage.OSS.SignedURLExpire); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
setString(values, "ai_vector_path", &cfg.VectorDB.Path)
|
||||
if err := setBool(values, SettingsPrefix+"wxwork_enabled", &cfg.WxWork.Enabled); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
setString(values, SettingsPrefix+"wxwork_corp_id", &cfg.WxWork.CorpID)
|
||||
setString(values, SettingsPrefix+"wxwork_corp_secret", &cfg.WxWork.CorpSecret)
|
||||
setString(values, SettingsPrefix+"wxwork_agent_id", &cfg.WxWork.AgentID)
|
||||
setString(values, SettingsPrefix+"wxwork_rsa_private_key", &cfg.WxWork.RSAPrivateKey)
|
||||
setString(values, SettingsPrefix+"wxwork_token", &cfg.WxWork.Token)
|
||||
setString(values, SettingsPrefix+"wxwork_encoding_aes_key", &cfg.WxWork.EncodingAESKey)
|
||||
if err := setBool(values, SettingsPrefix+"wxwork_notify_enabled", &cfg.WxWork.Notify.Enabled); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := setJSON(values, SettingsPrefix+"wxwork_notify_to_users", &cfg.WxWork.Notify.ToUsers); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := setBool(values, SettingsPrefix+"wxwork_notify_safe", &cfg.WxWork.Notify.Safe); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := setBool(values, SettingsPrefix+"wxwork_notify_dedupe", &cfg.WxWork.Notify.EnableDuplicateCheck); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := setInt(values, SettingsPrefix+"wxwork_notify_dedupe_interval", &cfg.WxWork.Notify.DuplicateCheckInterval); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
func defaultConfig() *Config {
|
||||
return &Config{
|
||||
Language: "zh-CN",
|
||||
Server: ServerConfig{Port: 8080},
|
||||
Storage: StorageConfig{
|
||||
Default: enums.AssetProviderLocal, MaxUploadSizeMB: 20,
|
||||
Local: LocalStorageConfig{Root: "data/storage", BaseURL: "/storage"},
|
||||
OSS: OSSStorageConfig{SignedURLExpire: 600},
|
||||
},
|
||||
VectorDB: VectorDBConfig{Path: "data/agent/vectors.db"},
|
||||
WxWork: WxWorkConfig{Notify: WxWorkNotifyConfig{
|
||||
EnableDuplicateCheck: true, DuplicateCheckInterval: 1800,
|
||||
}},
|
||||
}
|
||||
}
|
||||
|
||||
func setString(values map[string]string, key string, target *string) {
|
||||
if value, ok := values[key]; ok {
|
||||
*target = strings.TrimSpace(value)
|
||||
}
|
||||
}
|
||||
|
||||
func setInt(values map[string]string, key string, target *int) error {
|
||||
value, ok := values[key]
|
||||
if !ok || strings.TrimSpace(value) == "" {
|
||||
return nil
|
||||
}
|
||||
parsed, err := strconv.Atoi(strings.TrimSpace(value))
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid setting %s: %w", key, err)
|
||||
}
|
||||
*target = parsed
|
||||
return nil
|
||||
}
|
||||
|
||||
func setInt64(values map[string]string, key string, target *int64) error {
|
||||
value, ok := values[key]
|
||||
if !ok || strings.TrimSpace(value) == "" {
|
||||
return nil
|
||||
}
|
||||
parsed, err := strconv.ParseInt(strings.TrimSpace(value), 10, 64)
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid setting %s: %w", key, err)
|
||||
}
|
||||
*target = parsed
|
||||
return nil
|
||||
}
|
||||
|
||||
func setBool(values map[string]string, key string, target *bool) error {
|
||||
value, ok := values[key]
|
||||
if !ok || strings.TrimSpace(value) == "" {
|
||||
return nil
|
||||
}
|
||||
parsed, err := strconv.ParseBool(strings.TrimSpace(value))
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid setting %s: %w", key, err)
|
||||
}
|
||||
*target = parsed
|
||||
return nil
|
||||
}
|
||||
|
||||
func setJSON(values map[string]string, key string, target any) error {
|
||||
value, ok := values[key]
|
||||
if !ok || strings.TrimSpace(value) == "" {
|
||||
return nil
|
||||
}
|
||||
if err := json.Unmarshal([]byte(value), target); err != nil {
|
||||
return fmt.Errorf("invalid setting %s: %w", key, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
Language string `yaml:"language"`
|
||||
Server ServerConfig `yaml:"server"`
|
||||
@@ -15,7 +140,6 @@ type Config struct {
|
||||
Logger LoggerConfig `yaml:"logger"`
|
||||
Storage StorageConfig `yaml:"storage"`
|
||||
VectorDB VectorDBConfig `yaml:"vectorDB"`
|
||||
MCP MCPConfig `yaml:"mcp"`
|
||||
WxWork WxWorkConfig `yaml:"wxWork"`
|
||||
}
|
||||
|
||||
@@ -114,34 +238,9 @@ type OSSStorageConfig struct {
|
||||
}
|
||||
|
||||
type VectorDBConfig struct {
|
||||
Type string `yaml:"type"`
|
||||
Qdrant QdrantVectorDBConfig `yaml:"qdrant"`
|
||||
LanceDB LanceDBVectorDBConfig `yaml:"lancedb"`
|
||||
}
|
||||
|
||||
type QdrantVectorDBConfig struct {
|
||||
Host string `yaml:"host"`
|
||||
GrpcPort int `yaml:"grpcPort"`
|
||||
APIKey string `yaml:"apiKey"`
|
||||
UseTLS bool `yaml:"useTls"`
|
||||
}
|
||||
|
||||
type LanceDBVectorDBConfig struct {
|
||||
Path string `yaml:"path"`
|
||||
}
|
||||
|
||||
type MCPConfig struct {
|
||||
Enabled bool `yaml:"enabled"`
|
||||
Servers map[string]MCPServerConfig `yaml:"servers"`
|
||||
}
|
||||
|
||||
type MCPServerConfig struct {
|
||||
Enabled bool `yaml:"enabled"`
|
||||
Endpoint string `yaml:"endpoint"`
|
||||
TimeoutMS int `yaml:"timeoutMs"`
|
||||
Headers map[string]string `yaml:"headers"`
|
||||
}
|
||||
|
||||
// WxWorkConfig defines the WeCom application used for customer-service
|
||||
// callbacks and notifications. Dashboard login is owned by be-system.
|
||||
type WxWorkConfig struct {
|
||||
|
||||
Reference in New Issue
Block a user