fix: Phase 6联调 — 插件管理器端口修正 + 多模型配置系统整合 + 历史消息刷新修复
## 调试日志
### 1. 插件管理器启动失败
- **症状**: DevTools 显示插件管理器一直"已停止",手动启动正常
- **排查**: 对比 process-manager.js 传入的环境变量 vs plugin-manager config.go 读取的变量
- **根因**: config.js 传入 PLUGIN_MANAGER_PORT=8094,但 config.go 读取 os.Getenv("PORT"),env 名不匹配。且 process.env 中 PORT 泄露时被误读为 9090,与 DevTools 端口冲突
- **修复**: config.js 将 PLUGIN_MANAGER_PORT → PORT,使 env 名与代码一致 (c3055f4)
### 2. 历史消息刷新后消失
- **症状**: 浏览器刷新后聊天历史清空
- **排查**: WebSocket history_response handler 中 if (msg.messages) 对空数组 [] 为 truthy
- **根因**: 后端返回空的 history_response (缓存为空) 时,空数组覆盖了 HTTP 已加载的消息
- **修复**: useWebSocket.ts 改为 if (msg.messages && msg.messages.length > 0),空数组走 else-if 分支仅打日志,不覆盖已有消息
### 3. Phase 6 多模型配置系统
- Gateway: ModelsConfigStore (JSON文件持久化) + Admin CRUD API (providers/models/routing)
- ai-core: ModelSelector 支持按 purpose 选择 + fallback_chain,无配置时回退 .env
- DevTools: 模型配置管理面板 (Providers/Models/Routing 三Tab)、在线模型查询代理、路由表单 checkbox 多选、关键词搜索过滤
- .gitignore: models.json + platform_configs.json
### 4. 多端客户端追踪
- Hub 新增 knownClients 映射 (clientID → KnownClient),在线/离线状态追踪
- 客户端备注持久化到 PostgreSQL
- DevTools 客户端管理面板
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -40,6 +40,8 @@ data/
|
||||
# ========== 环境与敏感配置 ==========
|
||||
.env
|
||||
backend/.env
|
||||
models.json
|
||||
platform_configs.json
|
||||
.claude/
|
||||
|
||||
# ========== 文档 (项目规范:docs/ 不纳入版本管理) ==========
|
||||
|
||||
@@ -15,6 +15,7 @@ import (
|
||||
"github.com/joho/godotenv"
|
||||
|
||||
"github.com/yourname/cyrene-ai/ai-core/internal/background"
|
||||
aiConfig "github.com/yourname/cyrene-ai/ai-core/internal/config"
|
||||
ctxbuild "github.com/yourname/cyrene-ai/ai-core/internal/context"
|
||||
"github.com/yourname/cyrene-ai/ai-core/internal/llm"
|
||||
"github.com/yourname/cyrene-ai/ai-core/internal/memory"
|
||||
@@ -50,16 +51,31 @@ func main() {
|
||||
}
|
||||
log.Printf("已加载 %d 个人格: %v", len(personaLoader.List()), personaLoader.List())
|
||||
|
||||
// 初始化LLM适配器
|
||||
llmProvider := llm.NewOpenAIProvider(llm.OpenAIConfig{
|
||||
// 初始化模型配置加载器 (Phase 6: 多模型配置)
|
||||
configPath := getEnv("MODELS_CONFIG_PATH", "../models.json")
|
||||
configLoader, err := aiConfig.NewLoader(configPath)
|
||||
if err != nil {
|
||||
log.Printf("⚠ 模型配置加载失败,回退到 .env: %v", err)
|
||||
configLoader = nil
|
||||
}
|
||||
|
||||
// 构建 .env 回退配置
|
||||
envFallback := llm.OpenAIConfig{
|
||||
BaseURL: cfg.LLMBaseURL,
|
||||
APIKey: cfg.LLMAPIKey,
|
||||
Model: cfg.LLMModel,
|
||||
FallbackModel: cfg.LLMFallbackModel,
|
||||
Timeout: 120 * time.Second,
|
||||
})
|
||||
llmAdapter := llm.NewAdapter(llmProvider)
|
||||
log.Printf("LLM适配器已就绪: 模型=%s", llmAdapter.ModelName())
|
||||
}
|
||||
|
||||
// 创建 ModelSelector (优先使用 models.json,回退到 .env)
|
||||
modelSelector := llm.NewModelSelector(configLoader, envFallback)
|
||||
llmAdapter := modelSelector.DefaultAdapter()
|
||||
if configLoader != nil && configLoader.HasConfig() {
|
||||
log.Printf("LLM适配器已就绪: models.json 驱动")
|
||||
} else {
|
||||
log.Printf("LLM适配器已就绪: .env 驱动 (模型=%s)", llmAdapter.ModelName())
|
||||
}
|
||||
|
||||
// 初始化记忆系统
|
||||
var memStore *memory.Store
|
||||
|
||||
@@ -0,0 +1,119 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// ProviderData mirrors the Gateway ProviderConfig JSON shape.
|
||||
type ProviderData struct {
|
||||
Name string `json:"name"`
|
||||
BaseURL string `json:"base_url"`
|
||||
APIKey string `json:"api_key"`
|
||||
TimeoutSec int `json:"timeout_sec"`
|
||||
MaxRetries int `json:"max_retries"`
|
||||
APIVersion string `json:"api_version,omitempty"`
|
||||
ExtraHeaders map[string]string `json:"extra_headers,omitempty"`
|
||||
}
|
||||
|
||||
// ModelData mirrors the Gateway ModelConfig JSON shape.
|
||||
type ModelData struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Provider string `json:"provider"`
|
||||
Description string `json:"description"`
|
||||
Priority int `json:"priority"`
|
||||
Tags []string `json:"tags"`
|
||||
Params map[string]interface{} `json:"params"`
|
||||
Enabled bool `json:"enabled"`
|
||||
}
|
||||
|
||||
// RoutingData mirrors the Gateway RoutingRule JSON shape.
|
||||
type RoutingData struct {
|
||||
Purpose string `json:"purpose"`
|
||||
FallbackChain []string `json:"fallback_chain"`
|
||||
Required bool `json:"required"`
|
||||
}
|
||||
|
||||
// ModelsConfigData is the top-level config document (read-only mirror).
|
||||
type ModelsConfigData struct {
|
||||
Version string `json:"version"`
|
||||
Providers map[string]*ProviderData `json:"providers"`
|
||||
Models map[string]*ModelData `json:"models"`
|
||||
Routing map[string]*RoutingData `json:"routing"`
|
||||
}
|
||||
|
||||
// Loader provides read-only access to models.json.
|
||||
type Loader struct {
|
||||
mu sync.RWMutex
|
||||
path string
|
||||
config *ModelsConfigData
|
||||
}
|
||||
|
||||
// NewLoader reads models.json and returns a Loader. Returns nil config if file doesn't exist.
|
||||
func NewLoader(path string) (*Loader, error) {
|
||||
l := &Loader{
|
||||
path: path,
|
||||
config: &ModelsConfigData{
|
||||
Version: "1.0",
|
||||
Providers: make(map[string]*ProviderData),
|
||||
Models: make(map[string]*ModelData),
|
||||
Routing: make(map[string]*RoutingData),
|
||||
},
|
||||
}
|
||||
if err := l.load(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return l, nil
|
||||
}
|
||||
|
||||
func (l *Loader) load() error {
|
||||
data, err := os.ReadFile(l.path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
l.config = nil // Signal: use .env fallback.
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("read model config: %w", err)
|
||||
}
|
||||
if len(data) == 0 {
|
||||
l.config = nil
|
||||
return nil
|
||||
}
|
||||
var cfg ModelsConfigData
|
||||
if err := json.Unmarshal(data, &cfg); err != nil {
|
||||
return fmt.Errorf("parse model config: %w", err)
|
||||
}
|
||||
if cfg.Providers == nil {
|
||||
cfg.Providers = make(map[string]*ProviderData)
|
||||
}
|
||||
if cfg.Models == nil {
|
||||
cfg.Models = make(map[string]*ModelData)
|
||||
}
|
||||
if cfg.Routing == nil {
|
||||
cfg.Routing = make(map[string]*RoutingData)
|
||||
}
|
||||
l.config = &cfg
|
||||
return nil
|
||||
}
|
||||
|
||||
// HasConfig returns true if models.json exists and contains data.
|
||||
func (l *Loader) HasConfig() bool {
|
||||
l.mu.RLock()
|
||||
defer l.mu.RUnlock()
|
||||
return l.config != nil && (len(l.config.Providers) > 0 || len(l.config.Models) > 0)
|
||||
}
|
||||
|
||||
// Reload re-reads the config file. Used for config updates without restart.
|
||||
func (l *Loader) Reload() error {
|
||||
return l.load()
|
||||
}
|
||||
|
||||
// GetConfig returns the current config (read-only).
|
||||
func (l *Loader) GetConfig() *ModelsConfigData {
|
||||
l.mu.RLock()
|
||||
defer l.mu.RUnlock()
|
||||
return l.config
|
||||
}
|
||||
@@ -0,0 +1,131 @@
|
||||
package llm
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/yourname/cyrene-ai/ai-core/internal/config"
|
||||
)
|
||||
|
||||
// ModelPurpose identifies the kind of LLM task.
|
||||
type ModelPurpose string
|
||||
|
||||
const (
|
||||
PurposeChat ModelPurpose = "chat"
|
||||
PurposeDeepThinking ModelPurpose = "deep_thinking"
|
||||
PurposeIntentAnalysis ModelPurpose = "intent_analysis"
|
||||
PurposeToolCalling ModelPurpose = "tool_calling"
|
||||
PurposeMemoryExtraction ModelPurpose = "memory_extraction"
|
||||
)
|
||||
|
||||
// ErrModelNotRequired is returned when an optional model is unavailable.
|
||||
var ErrModelNotRequired = fmt.Errorf("model not required, caller should degrade gracefully")
|
||||
|
||||
// ModelSelector routes requests to the best available LLMProvider based on purpose.
|
||||
type ModelSelector struct {
|
||||
loader *config.Loader
|
||||
envCfg OpenAIConfig
|
||||
mu sync.RWMutex
|
||||
cache map[string]LLMProvider
|
||||
}
|
||||
|
||||
// NewModelSelector creates a ModelSelector. If loader is nil or has no config,
|
||||
// all calls fall back to envCfg.
|
||||
func NewModelSelector(loader *config.Loader, envFallback OpenAIConfig) *ModelSelector {
|
||||
return &ModelSelector{
|
||||
loader: loader,
|
||||
envCfg: envFallback,
|
||||
cache: make(map[string]LLMProvider),
|
||||
}
|
||||
}
|
||||
|
||||
// Select returns an LLMProvider for the given purpose. Falls back through the
|
||||
// routing fallback chain; returns the env provider if nothing matches.
|
||||
func (s *ModelSelector) Select(ctx context.Context, purpose ModelPurpose) (LLMProvider, error) {
|
||||
if s.loader == nil || !s.loader.HasConfig() {
|
||||
return s.envProvider(), nil
|
||||
}
|
||||
|
||||
cfg := s.loader.GetConfig()
|
||||
if cfg == nil {
|
||||
return s.envProvider(), nil
|
||||
}
|
||||
|
||||
route, ok := cfg.Routing[string(purpose)]
|
||||
if !ok || len(route.FallbackChain) == 0 {
|
||||
return s.envProvider(), nil
|
||||
}
|
||||
|
||||
for _, modelID := range route.FallbackChain {
|
||||
provider, err := s.getOrCreateProvider(modelID, cfg)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
return provider, nil
|
||||
}
|
||||
|
||||
if route.Required {
|
||||
return nil, fmt.Errorf("all models unavailable for purpose %s", purpose)
|
||||
}
|
||||
return s.envProvider(), nil
|
||||
}
|
||||
|
||||
// DefaultAdapter returns an *Adapter backed by the chat-purpose provider.
|
||||
// This is the backward-compatible entry point: all existing consumers
|
||||
// (Orchestrator, Synthesizer, BackgroundThinker, etc.) use this.
|
||||
func (s *ModelSelector) DefaultAdapter() *Adapter {
|
||||
provider, _ := s.Select(context.Background(), PurposeChat)
|
||||
return NewAdapter(provider)
|
||||
}
|
||||
|
||||
func (s *ModelSelector) envProvider() LLMProvider {
|
||||
return NewOpenAIProvider(s.envCfg)
|
||||
}
|
||||
|
||||
func (s *ModelSelector) getOrCreateProvider(modelID string, cfg *config.ModelsConfigData) (LLMProvider, error) {
|
||||
s.mu.RLock()
|
||||
if p, ok := s.cache[modelID]; ok {
|
||||
s.mu.RUnlock()
|
||||
return p, nil
|
||||
}
|
||||
s.mu.RUnlock()
|
||||
|
||||
modelCfg, ok := cfg.Models[modelID]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("model %s not found", modelID)
|
||||
}
|
||||
if !modelCfg.Enabled {
|
||||
return nil, fmt.Errorf("model %s is disabled", modelID)
|
||||
}
|
||||
|
||||
provCfg, ok := cfg.Providers[modelCfg.Provider]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("provider %s not found for model %s", modelCfg.Provider, modelID)
|
||||
}
|
||||
|
||||
timeout := time.Duration(provCfg.TimeoutSec) * time.Second
|
||||
if timeout <= 0 {
|
||||
timeout = 120 * time.Second
|
||||
}
|
||||
maxRetries := provCfg.MaxRetries
|
||||
if maxRetries <= 0 {
|
||||
maxRetries = 3
|
||||
}
|
||||
|
||||
provider := NewOpenAIProvider(OpenAIConfig{
|
||||
BaseURL: provCfg.BaseURL,
|
||||
APIKey: provCfg.APIKey,
|
||||
Model: modelCfg.Name,
|
||||
FallbackModel: modelCfg.Name,
|
||||
MaxRetries: maxRetries,
|
||||
Timeout: timeout,
|
||||
})
|
||||
|
||||
s.mu.Lock()
|
||||
s.cache[modelID] = provider
|
||||
s.mu.Unlock()
|
||||
|
||||
return provider, nil
|
||||
}
|
||||
@@ -174,7 +174,18 @@ func main() {
|
||||
if sessionStore != nil {
|
||||
db = sessionStore.DB()
|
||||
}
|
||||
router.Setup(r, hub, cfg, sessionStore, reminderStore, briefingStore, automationStore, fileStore, ruleEngine, knowledgeStore, nil, db)
|
||||
// 初始化模型配置存储 (Phase 6)
|
||||
modelConfigStore, err := config.NewModelsConfigStore("../models.json")
|
||||
if err != nil {
|
||||
logger.Printf("[WARN] 模型配置存储初始化失败 (将仅使用 .env 回退): %v", err)
|
||||
modelConfigStore = nil
|
||||
} else if modelConfigStore.HasConfig() {
|
||||
logger.Println("[INFO] 模型配置文件已加载 (models.json)")
|
||||
} else {
|
||||
logger.Println("[INFO] 模型配置文件不存在,回退到 .env LLM 配置")
|
||||
}
|
||||
|
||||
router.Setup(r, hub, cfg, sessionStore, reminderStore, briefingStore, automationStore, fileStore, ruleEngine, knowledgeStore, nil, db, modelConfigStore)
|
||||
|
||||
// 启动提醒调度器
|
||||
if reminderStore != nil {
|
||||
|
||||
@@ -0,0 +1,234 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ModelsConfigStore manages persistence of model configuration to a JSON file.
|
||||
type ModelsConfigStore struct {
|
||||
mu sync.RWMutex
|
||||
path string
|
||||
config *ModelsConfig
|
||||
}
|
||||
|
||||
// NewModelsConfigStore creates a ModelsConfigStore, creating an empty config file if it doesn't exist.
|
||||
func NewModelsConfigStore(path string) (*ModelsConfigStore, error) {
|
||||
s := &ModelsConfigStore{
|
||||
path: path,
|
||||
config: &ModelsConfig{
|
||||
Version: "1.0",
|
||||
Providers: make(map[string]*ProviderConfig),
|
||||
Models: make(map[string]*ModelConfig),
|
||||
Routing: make(map[string]*RoutingRule),
|
||||
},
|
||||
}
|
||||
if err := s.load(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) load() error {
|
||||
data, err := os.ReadFile(s.path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return s.save() // Initialize empty file.
|
||||
}
|
||||
return fmt.Errorf("read model config file: %w", err)
|
||||
}
|
||||
if len(data) == 0 {
|
||||
return nil
|
||||
}
|
||||
var cfg ModelsConfig
|
||||
if err := json.Unmarshal(data, &cfg); err != nil {
|
||||
return fmt.Errorf("parse model config file: %w", err)
|
||||
}
|
||||
if cfg.Providers == nil {
|
||||
cfg.Providers = make(map[string]*ProviderConfig)
|
||||
}
|
||||
if cfg.Models == nil {
|
||||
cfg.Models = make(map[string]*ModelConfig)
|
||||
}
|
||||
if cfg.Routing == nil {
|
||||
cfg.Routing = make(map[string]*RoutingRule)
|
||||
}
|
||||
if cfg.Version == "" {
|
||||
cfg.Version = "1.0"
|
||||
}
|
||||
s.config = &cfg
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) save() error {
|
||||
data, err := json.MarshalIndent(s.config, "", " ")
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal model config: %w", err)
|
||||
}
|
||||
tmpPath := s.path + ".tmp"
|
||||
if err := os.WriteFile(tmpPath, data, 0640); err != nil {
|
||||
return fmt.Errorf("write model config file: %w", err)
|
||||
}
|
||||
return os.Rename(tmpPath, s.path)
|
||||
}
|
||||
|
||||
// HasConfig returns true if there are any providers or models configured.
|
||||
func (s *ModelsConfigStore) HasConfig() bool {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return len(s.config.Providers) > 0 || len(s.config.Models) > 0
|
||||
}
|
||||
|
||||
// ---- Providers ----
|
||||
|
||||
func (s *ModelsConfigStore) ListProviders() []*ProviderConfig {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
result := make([]*ProviderConfig, 0, len(s.config.Providers))
|
||||
for _, p := range s.config.Providers {
|
||||
result = append(result, p)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) GetProvider(name string) (*ProviderConfig, error) {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
p, ok := s.config.Providers[name]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("provider not found: %s", name)
|
||||
}
|
||||
return p, nil
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) SetProvider(cfg *ProviderConfig) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if cfg.Name == "" {
|
||||
return fmt.Errorf("provider name is required")
|
||||
}
|
||||
if cfg.BaseURL == "" {
|
||||
return fmt.Errorf("provider base_url is required")
|
||||
}
|
||||
cfg.UpdatedAt = time.Now()
|
||||
s.config.Providers[cfg.Name] = cfg
|
||||
return s.save()
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) DeleteProvider(name string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if _, ok := s.config.Providers[name]; !ok {
|
||||
return fmt.Errorf("provider not found: %s", name)
|
||||
}
|
||||
delete(s.config.Providers, name)
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// ---- Models ----
|
||||
|
||||
func (s *ModelsConfigStore) ListModels() []*ModelConfig {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
result := make([]*ModelConfig, 0, len(s.config.Models))
|
||||
for _, m := range s.config.Models {
|
||||
result = append(result, m)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) GetModel(id string) (*ModelConfig, error) {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
m, ok := s.config.Models[id]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("model not found: %s", id)
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) SetModel(cfg *ModelConfig) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if cfg.ID == "" {
|
||||
return fmt.Errorf("model id is required")
|
||||
}
|
||||
if cfg.Provider == "" {
|
||||
return fmt.Errorf("model provider is required")
|
||||
}
|
||||
cfg.UpdatedAt = time.Now()
|
||||
if cfg.Params == nil {
|
||||
cfg.Params = make(map[string]interface{})
|
||||
}
|
||||
if cfg.Tags == nil {
|
||||
cfg.Tags = []string{}
|
||||
}
|
||||
s.config.Models[cfg.ID] = cfg
|
||||
return s.save()
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) DeleteModel(id string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if _, ok := s.config.Models[id]; !ok {
|
||||
return fmt.Errorf("model not found: %s", id)
|
||||
}
|
||||
delete(s.config.Models, id)
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// ---- Routing ----
|
||||
|
||||
func (s *ModelsConfigStore) ListRouting() []*RoutingRule {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
result := make([]*RoutingRule, 0, len(s.config.Routing))
|
||||
for _, r := range s.config.Routing {
|
||||
result = append(result, r)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) GetRouting(purpose string) (*RoutingRule, error) {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
r, ok := s.config.Routing[purpose]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("routing not found: %s", purpose)
|
||||
}
|
||||
return r, nil
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) SetRouting(rule *RoutingRule) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if rule.Purpose == "" {
|
||||
return fmt.Errorf("routing purpose is required")
|
||||
}
|
||||
if len(rule.FallbackChain) == 0 {
|
||||
return fmt.Errorf("routing fallback_chain is required")
|
||||
}
|
||||
s.config.Routing[rule.Purpose] = rule
|
||||
return s.save()
|
||||
}
|
||||
|
||||
func (s *ModelsConfigStore) DeleteRouting(purpose string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if _, ok := s.config.Routing[purpose]; !ok {
|
||||
return fmt.Errorf("routing not found: %s", purpose)
|
||||
}
|
||||
delete(s.config.Routing, purpose)
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// GetConfig returns a copy of the full config (for ai-core loader compatibility).
|
||||
func (s *ModelsConfigStore) GetConfig() *ModelsConfig {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
// Return shallow copy; callers should treat as read-only.
|
||||
return s.config
|
||||
}
|
||||
@@ -0,0 +1,43 @@
|
||||
package config
|
||||
|
||||
import "time"
|
||||
|
||||
// ProviderConfig defines an LLM service provider (e.g. deepseek, openai).
|
||||
type ProviderConfig struct {
|
||||
Name string `json:"name"`
|
||||
BaseURL string `json:"base_url"`
|
||||
APIKey string `json:"api_key"`
|
||||
TimeoutSec int `json:"timeout_sec"`
|
||||
MaxRetries int `json:"max_retries"`
|
||||
APIVersion string `json:"api_version,omitempty"`
|
||||
ExtraHeaders map[string]string `json:"extra_headers,omitempty"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// ModelConfig defines a specific model under a provider.
|
||||
type ModelConfig struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Provider string `json:"provider"`
|
||||
Description string `json:"description"`
|
||||
Priority int `json:"priority"`
|
||||
Tags []string `json:"tags"`
|
||||
Params map[string]interface{} `json:"params"`
|
||||
Enabled bool `json:"enabled"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// RoutingRule maps a purpose to an ordered fallback chain of model IDs.
|
||||
type RoutingRule struct {
|
||||
Purpose string `json:"purpose"`
|
||||
FallbackChain []string `json:"fallback_chain"`
|
||||
Required bool `json:"required"`
|
||||
}
|
||||
|
||||
// ModelsConfig is the top-level configuration document.
|
||||
type ModelsConfig struct {
|
||||
Version string `json:"version"`
|
||||
Providers map[string]*ProviderConfig `json:"providers"`
|
||||
Models map[string]*ModelConfig `json:"models"`
|
||||
Routing map[string]*RoutingRule `json:"routing"`
|
||||
}
|
||||
@@ -7,8 +7,8 @@ import (
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"github.com/yourname/cyrene-ai/pkg/logger"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
@@ -50,6 +50,9 @@ func (h *ChatHandler) HandleWebSocket(c *gin.Context) {
|
||||
// 从query参数获取token和session_id
|
||||
token := c.Query("token")
|
||||
sessionID := c.Query("session_id")
|
||||
clientID := c.Query("client_id")
|
||||
deviceName := c.Query("device_name")
|
||||
userAgent := c.Request.UserAgent()
|
||||
|
||||
if token == "" {
|
||||
// 也尝试从Authorization头读取
|
||||
@@ -93,7 +96,7 @@ func (h *ChatHandler) HandleWebSocket(c *gin.Context) {
|
||||
}
|
||||
|
||||
// 创建客户端
|
||||
client := ws.NewClient(h.hub, conn, userID, sessionID)
|
||||
client := ws.NewClient(h.hub, conn, userID, sessionID, clientID, deviceName, userAgent)
|
||||
|
||||
// 注册到Hub
|
||||
h.hub.Register(client)
|
||||
@@ -128,7 +131,7 @@ func (h *ChatHandler) handleChatMessage(client *ws.Client, msg ws.ClientMessage)
|
||||
|
||||
// 持久化用户消息到数据库(在 WebSocket 发送之前)
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, "user", "chat", msg.Content); err != nil {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, "user", "chat", msg.Content, client.ClientID); err != nil {
|
||||
logger.Printf("[chat] 持久化用户消息失败: %v", err)
|
||||
}
|
||||
}
|
||||
@@ -168,6 +171,10 @@ func (h *ChatHandler) handleChatMessage(client *ws.Client, msg ws.ClientMessage)
|
||||
Role: "user",
|
||||
Content: msg.Content,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
ClientInfo: &ws.ClientInfo{
|
||||
ClientID: client.ClientID,
|
||||
DeviceName: client.DeviceName,
|
||||
},
|
||||
}
|
||||
if len(msg.Attachments) > 0 {
|
||||
userMsg.Attachments = msg.Attachments
|
||||
@@ -229,13 +236,13 @@ func (h *ChatHandler) streamResponse(client *ws.Client, mode string, reqBody []b
|
||||
// 增大 scanner buffer 以处理大块 SSE 数据
|
||||
scanner.Buffer(make([]byte, 0, 64*1024), 1024*1024)
|
||||
|
||||
// 通知前端 AI 开始生成回复
|
||||
client.SendMessage(ws.ServerMessage{
|
||||
Type: "stream_start",
|
||||
MessageID: "msg_" + generateID(),
|
||||
SessionID: client.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
})
|
||||
// 通知前端 AI 开始生成回复
|
||||
client.SendMessage(ws.ServerMessage{
|
||||
Type: "stream_start",
|
||||
MessageID: "msg_" + generateID(),
|
||||
SessionID: client.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
})
|
||||
|
||||
var fullText string
|
||||
var msgID string
|
||||
@@ -312,24 +319,30 @@ func (h *ChatHandler) streamResponse(client *ws.Client, mode string, reqBody []b
|
||||
reviewMsgID := fmt.Sprintf("%s_r%d", msgID, i)
|
||||
// 持久化每条审查消息
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, role, msgType, rm.Content); err != nil {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, role, msgType, rm.Content, client.ClientID); err != nil {
|
||||
logger.Printf("[chat] 持久化审查消息失败: %v", err)
|
||||
}
|
||||
}
|
||||
clientInfo := &ws.ClientInfo{
|
||||
ClientID: client.ClientID,
|
||||
DeviceName: client.DeviceName,
|
||||
}
|
||||
h.hub.CacheMessage(client.UserID, client.SessionID, ws.Message{
|
||||
ID: reviewMsgID,
|
||||
Role: role,
|
||||
Content: rm.Content,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
ID: reviewMsgID,
|
||||
Role: role,
|
||||
Content: rm.Content,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
ClientInfo: clientInfo,
|
||||
})
|
||||
client.SendMessage(ws.ServerMessage{
|
||||
Type: "response",
|
||||
MessageID: reviewMsgID,
|
||||
Content: rm.Content,
|
||||
Role: role,
|
||||
MsgType: msgType,
|
||||
SessionID: client.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
Type: "response",
|
||||
MessageID: reviewMsgID,
|
||||
Content: rm.Content,
|
||||
Role: role,
|
||||
MsgType: msgType,
|
||||
SessionID: client.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
ClientInfo: clientInfo,
|
||||
})
|
||||
// 使用 MessageScheduler 计算的 per-message 延迟
|
||||
if rm.DelayMs > 0 {
|
||||
@@ -416,7 +429,7 @@ func (h *ChatHandler) streamResponse(client *ws.Client, mode string, reqBody []b
|
||||
// 如果有审查消息,每条已单独持久化,跳过 fullText 以避免重复
|
||||
if !hasReview && fullText != "" {
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, "assistant", "chat", fullText); err != nil {
|
||||
if err := h.sessionStore.AddMessage(client.SessionID, "assistant", "chat", fullText, client.ClientID); err != nil {
|
||||
logger.Printf("[chat] 持久化 AI 回复失败: %v", err)
|
||||
}
|
||||
}
|
||||
@@ -426,6 +439,10 @@ func (h *ChatHandler) streamResponse(client *ws.Client, mode string, reqBody []b
|
||||
Role: "assistant",
|
||||
Content: fullText,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
ClientInfo: &ws.ClientInfo{
|
||||
ClientID: client.ClientID,
|
||||
DeviceName: client.DeviceName,
|
||||
},
|
||||
})
|
||||
}
|
||||
// RecordMessage 使用不带 [review] 标记的文本
|
||||
@@ -451,7 +468,6 @@ func (h *ChatHandler) handleVoiceInput(client *ws.Client, msg ws.ClientMessage)
|
||||
client.SendMessage(response)
|
||||
}
|
||||
|
||||
|
||||
// handleHistoryRequest 处理历史消息请求
|
||||
func (h *ChatHandler) handleHistoryRequest(client *ws.Client, msg ws.ClientMessage) {
|
||||
// 优先使用请求中的 session_id,否则使用客户端的 session_id
|
||||
@@ -469,19 +485,28 @@ func (h *ChatHandler) handleHistoryRequest(client *ws.Client, msg ws.ClientMessa
|
||||
logger.Printf("[history] 从数据库恢复会话历史: session=%s, %d 条消息", sessionID, len(dbMessages))
|
||||
// 恢复到内存缓存
|
||||
for _, dbMsg := range dbMessages {
|
||||
var ci *ws.ClientInfo
|
||||
if dbMsg.ClientID != "" {
|
||||
ci = h.hub.ClientInfo(dbMsg.ClientID)
|
||||
if ci == nil {
|
||||
ci = &ws.ClientInfo{ClientID: dbMsg.ClientID}
|
||||
}
|
||||
}
|
||||
messages = append(messages, ws.Message{
|
||||
ID: fmt.Sprintf("db_%d", dbMsg.ID),
|
||||
Role: dbMsg.Role,
|
||||
MsgType: dbMsg.MsgType,
|
||||
Content: dbMsg.Content,
|
||||
Timestamp: dbMsg.CreatedAt.UnixMilli(),
|
||||
ID: fmt.Sprintf("db_%d", dbMsg.ID),
|
||||
Role: dbMsg.Role,
|
||||
MsgType: dbMsg.MsgType,
|
||||
Content: dbMsg.Content,
|
||||
Timestamp: dbMsg.CreatedAt.UnixMilli(),
|
||||
ClientInfo: ci,
|
||||
})
|
||||
h.hub.CacheMessage(client.UserID, sessionID, ws.Message{
|
||||
ID: fmt.Sprintf("db_%d", dbMsg.ID),
|
||||
Role: dbMsg.Role,
|
||||
MsgType: dbMsg.MsgType,
|
||||
Content: dbMsg.Content,
|
||||
Timestamp: dbMsg.CreatedAt.UnixMilli(),
|
||||
ID: fmt.Sprintf("db_%d", dbMsg.ID),
|
||||
Role: dbMsg.Role,
|
||||
MsgType: dbMsg.MsgType,
|
||||
Content: dbMsg.Content,
|
||||
Timestamp: dbMsg.CreatedAt.UnixMilli(),
|
||||
ClientInfo: ci,
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -534,72 +559,216 @@ func (h *ChatHandler) HandleProactiveMessage(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
// 检查用户是否在线
|
||||
// Parse content to split (action) from chat text.
|
||||
segments := parseProactiveContent(req.Content)
|
||||
|
||||
// Check online status.
|
||||
onlineCount := h.hub.UserClientCount(req.UserID)
|
||||
if onlineCount == 0 {
|
||||
// Phase 2: 离线时排队,等待用户重连后推送
|
||||
data, _ := json.Marshal(ws.ServerMessage{
|
||||
Type: "response",
|
||||
MessageID: "proactive_" + generateID(),
|
||||
Content: req.Content,
|
||||
Role: "assistant",
|
||||
MsgType: "proactive",
|
||||
SessionID: req.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
})
|
||||
h.hub.QueueProactiveMessage(req.UserID, data)
|
||||
logger.Printf("[proactive] 用户离线,消息已排队: user=%s", req.UserID)
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"success": true,
|
||||
"reason": "queued",
|
||||
"message": "用户离线,消息已排队等待重连后推送",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 构建主动消息
|
||||
msgID := "proactive_" + generateID()
|
||||
msg := ws.ServerMessage{
|
||||
Type: "response",
|
||||
MessageID: msgID,
|
||||
Content: req.Content,
|
||||
Role: "assistant",
|
||||
MsgType: "proactive",
|
||||
SessionID: req.SessionID,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
}
|
||||
|
||||
data, err := json.Marshal(msg)
|
||||
if err != nil {
|
||||
logger.Printf("[proactive] 序列化消息失败: %v", err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "内部错误"})
|
||||
return
|
||||
}
|
||||
|
||||
h.hub.SendToUser(req.UserID, data)
|
||||
|
||||
// 同时缓存到对话历史(使用 admin 的主 session)
|
||||
sessionID := req.SessionID
|
||||
if sessionID == "" {
|
||||
sessionID = "session_admin_main"
|
||||
}
|
||||
h.hub.CacheMessage(req.UserID, sessionID, ws.Message{
|
||||
ID: msgID,
|
||||
Role: "assistant",
|
||||
Content: req.Content,
|
||||
Timestamp: time.Now().UnixMilli(),
|
||||
})
|
||||
h.hub.RecordMessage(sessionID, "assistant", req.Content)
|
||||
|
||||
logger.Printf("[proactive] 主动消息已推送: user=%s, online=%d, content_len=%d", req.UserID, onlineCount, len(req.Content))
|
||||
timestamp := time.Now().UnixMilli()
|
||||
|
||||
for i, seg := range segments {
|
||||
msgID := fmt.Sprintf("proactive_%s_%d", generateID(), i)
|
||||
msgType := "chat"
|
||||
role := "assistant"
|
||||
if seg.msgType == "action" {
|
||||
msgType = "action"
|
||||
role = "action"
|
||||
}
|
||||
|
||||
msg := ws.ServerMessage{
|
||||
Type: "response",
|
||||
MessageID: msgID,
|
||||
Content: seg.content,
|
||||
Role: role,
|
||||
MsgType: msgType,
|
||||
SessionID: sessionID,
|
||||
Timestamp: timestamp + int64(i),
|
||||
}
|
||||
|
||||
data, err := json.Marshal(msg)
|
||||
if err != nil {
|
||||
logger.Printf("[proactive] 序列化消息失败: %v", err)
|
||||
continue
|
||||
}
|
||||
|
||||
if onlineCount == 0 {
|
||||
h.hub.QueueProactiveMessage(req.UserID, data)
|
||||
} else {
|
||||
h.hub.SendToUser(req.UserID, data)
|
||||
if i < len(segments)-1 {
|
||||
delay := 200 + int(time.Now().UnixNano()%200)
|
||||
time.Sleep(time.Duration(delay) * time.Millisecond)
|
||||
}
|
||||
}
|
||||
|
||||
// Persist to database so proactive messages survive restarts.
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
if err := h.sessionStore.AddMessage(sessionID, role, msgType, seg.content, ""); err != nil {
|
||||
logger.Printf("[proactive] 持久化消息失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Cache each segment to conversation history.
|
||||
h.hub.CacheMessage(req.UserID, sessionID, ws.Message{
|
||||
ID: msgID,
|
||||
Role: role,
|
||||
MsgType: msgType,
|
||||
Content: seg.content,
|
||||
Timestamp: timestamp,
|
||||
})
|
||||
h.hub.RecordMessage(sessionID, role, seg.content)
|
||||
}
|
||||
|
||||
logger.Printf("[proactive] 主动消息已推送: user=%s, online=%d, segments=%d", req.UserID, onlineCount, len(segments))
|
||||
|
||||
reason := "delivered"
|
||||
if onlineCount == 0 {
|
||||
reason = "queued"
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"success": true,
|
||||
"message": "消息已推送",
|
||||
"segments": len(segments),
|
||||
"delivered": onlineCount,
|
||||
"reason": reason,
|
||||
})
|
||||
}
|
||||
|
||||
// proactiveSegment holds a parsed piece of a proactive message.
|
||||
type proactiveSegment struct {
|
||||
msgType string // "chat" or "action"
|
||||
content string
|
||||
}
|
||||
|
||||
// parseProactiveContent splits text by (parenthesized actions).
|
||||
// "(笑) 你好呀 (调暗灯光) 今天过得如何" →
|
||||
// [action: "笑", chat: "你好呀", action: "调暗灯光", chat: "今天过得如何"]
|
||||
func parseProactiveContent(text string) []proactiveSegment {
|
||||
if text == "" {
|
||||
return nil
|
||||
}
|
||||
|
||||
var segments []proactiveSegment
|
||||
remaining := []rune(text)
|
||||
|
||||
for len(remaining) > 0 {
|
||||
actionStart := -1 // index in remaining
|
||||
actionEnd := -1 // index after closing paren
|
||||
var actionContent string
|
||||
|
||||
for i, r := range remaining {
|
||||
if r == '(' || r == '(' {
|
||||
actionStart = i
|
||||
closeRune := ')'
|
||||
if r == '(' {
|
||||
closeRune = ')'
|
||||
}
|
||||
for j := i + 1; j < len(remaining); j++ {
|
||||
if remaining[j] == closeRune {
|
||||
actionEnd = j + 1
|
||||
actionContent = string(remaining[i+1 : j])
|
||||
break
|
||||
}
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if actionStart >= 0 {
|
||||
if actionStart > 0 {
|
||||
prefix := strings.TrimSpace(string(remaining[:actionStart]))
|
||||
if prefix != "" {
|
||||
segments = append(segments, proactiveSegment{msgType: "chat", content: prefix})
|
||||
}
|
||||
}
|
||||
content := strings.TrimSpace(actionContent)
|
||||
if content != "" {
|
||||
segments = append(segments, proactiveSegment{msgType: "action", content: content})
|
||||
}
|
||||
remaining = remaining[actionEnd:]
|
||||
} else {
|
||||
text := strings.TrimSpace(string(remaining))
|
||||
if text != "" {
|
||||
segments = append(segments, proactiveSegment{msgType: "chat", content: text})
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if len(segments) == 0 && text != "" {
|
||||
segments = append(segments, proactiveSegment{msgType: "chat", content: strings.TrimSpace(text)})
|
||||
}
|
||||
|
||||
return segments
|
||||
}
|
||||
|
||||
// ========== 多端客户端管理 API ==========
|
||||
|
||||
// HandleListClients returns all known clients for the authenticated user.
|
||||
// GET /api/v1/admin/clients
|
||||
func (h *ChatHandler) HandleListClients(c *gin.Context) {
|
||||
userID := c.Query("user_id")
|
||||
if userID == "" {
|
||||
userID = "admin"
|
||||
}
|
||||
clients := h.hub.GetKnownClients(userID)
|
||||
|
||||
// Merge with persisted notes from DB
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
dbClients, err := h.sessionStore.GetClients(userID)
|
||||
if err == nil {
|
||||
noteByID := make(map[string]string)
|
||||
for _, dc := range dbClients {
|
||||
noteByID[dc.ClientID] = dc.Note
|
||||
}
|
||||
for i := range clients {
|
||||
if note, ok := noteByID[clients[i].ClientID]; ok && note != "" {
|
||||
clients[i].Note = note
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"clients": clients,
|
||||
"total": len(clients),
|
||||
})
|
||||
}
|
||||
|
||||
// HandleUpdateClientNote sets a label/note on a client.
|
||||
// PUT /api/v1/admin/clients/:id/note
|
||||
func (h *ChatHandler) HandleUpdateClientNote(c *gin.Context) {
|
||||
clientID := c.Param("id")
|
||||
var req struct {
|
||||
Note string `json:"note"`
|
||||
}
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "请求参数无效"})
|
||||
return
|
||||
}
|
||||
|
||||
// Update in-memory
|
||||
if !h.hub.UpdateClientNote(clientID, req.Note) {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "客户端未找到"})
|
||||
return
|
||||
}
|
||||
|
||||
// Persist to DB
|
||||
if h.sessionStore != nil && h.sessionStore.IsAvailable() {
|
||||
if err := h.sessionStore.UpdateClientNote(clientID, req.Note); err != nil {
|
||||
logger.Printf("[clients] 持久化备注失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"status": "ok", "client_id": clientID, "note": req.Note})
|
||||
}
|
||||
|
||||
func generateID() string {
|
||||
return time.Now().Format("20060102150405") + randomStr(6)
|
||||
}
|
||||
@@ -609,7 +778,7 @@ func randomStr(n int) string {
|
||||
if _, err := rand.Read(b); err != nil {
|
||||
// fallback: deterministic but hard to predict
|
||||
for i := range b {
|
||||
b[i] = byte(time.Now().UnixNano()%256)
|
||||
b[i] = byte(time.Now().UnixNano() % 256)
|
||||
}
|
||||
}
|
||||
return hex.EncodeToString(b)[:n]
|
||||
@@ -637,4 +806,3 @@ func parseMultiMessage(text string) []string {
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,297 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/yourname/cyrene-ai/gateway/internal/config"
|
||||
)
|
||||
|
||||
// ModelConfigHandler exposes admin CRUD endpoints for model configuration.
|
||||
type ModelConfigHandler struct {
|
||||
store *config.ModelsConfigStore
|
||||
}
|
||||
|
||||
func NewModelConfigHandler(store *config.ModelsConfigStore) *ModelConfigHandler {
|
||||
return &ModelConfigHandler{store: store}
|
||||
}
|
||||
|
||||
// ---- Providers ----
|
||||
|
||||
func (h *ModelConfigHandler) ListProviders(c *gin.Context) {
|
||||
providers := h.store.ListProviders()
|
||||
if providers == nil {
|
||||
providers = []*config.ProviderConfig{}
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"providers": providers, "total": len(providers)})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) GetProvider(c *gin.Context) {
|
||||
name := c.Param("name")
|
||||
p, err := h.store.GetProvider(name)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, p)
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) SetProvider(c *gin.Context) {
|
||||
name := c.Param("name")
|
||||
var body config.ProviderConfig
|
||||
if err := c.ShouldBindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON: " + err.Error()})
|
||||
return
|
||||
}
|
||||
body.Name = name
|
||||
if err := h.store.SetProvider(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "saved", "name": name})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) DeleteProvider(c *gin.Context) {
|
||||
name := c.Param("name")
|
||||
if err := h.store.DeleteProvider(name); err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "deleted", "name": name})
|
||||
}
|
||||
|
||||
// ---- Models ----
|
||||
|
||||
func (h *ModelConfigHandler) ListModels(c *gin.Context) {
|
||||
models := h.store.ListModels()
|
||||
if models == nil {
|
||||
models = []*config.ModelConfig{}
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"models": models, "total": len(models)})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) GetModel(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
m, err := h.store.GetModel(id)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, m)
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) SetModel(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
var body config.ModelConfig
|
||||
if err := c.ShouldBindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON: " + err.Error()})
|
||||
return
|
||||
}
|
||||
body.ID = id
|
||||
if err := h.store.SetModel(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "saved", "id": id})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) DeleteModel(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
if err := h.store.DeleteModel(id); err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "deleted", "id": id})
|
||||
}
|
||||
|
||||
// ---- Routing ----
|
||||
|
||||
func (h *ModelConfigHandler) ListRouting(c *gin.Context) {
|
||||
routing := h.store.ListRouting()
|
||||
if routing == nil {
|
||||
routing = []*config.RoutingRule{}
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"routing": routing, "total": len(routing)})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) GetRouting(c *gin.Context) {
|
||||
purpose := c.Param("purpose")
|
||||
r, err := h.store.GetRouting(purpose)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, r)
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) SetRouting(c *gin.Context) {
|
||||
purpose := c.Param("purpose")
|
||||
var body config.RoutingRule
|
||||
if err := c.ShouldBindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON: " + err.Error()})
|
||||
return
|
||||
}
|
||||
body.Purpose = purpose
|
||||
if err := h.store.SetRouting(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "saved", "purpose": purpose})
|
||||
}
|
||||
|
||||
func (h *ModelConfigHandler) DeleteRouting(c *gin.Context) {
|
||||
purpose := c.Param("purpose")
|
||||
if err := h.store.DeleteRouting(purpose); err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"status": "deleted", "purpose": purpose})
|
||||
}
|
||||
|
||||
// ---- Health Check ----
|
||||
|
||||
func (h *ModelConfigHandler) TestProvider(c *gin.Context) {
|
||||
var body struct {
|
||||
Provider string `json:"provider"`
|
||||
}
|
||||
if err := c.ShouldBindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON: " + err.Error()})
|
||||
return
|
||||
}
|
||||
p, err := h.store.GetProvider(body.Provider)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"provider": p.Name,
|
||||
"base_url": p.BaseURL,
|
||||
"message": "Provider 配置已保存,连接测试请通过实际 LLM 调用验证",
|
||||
})
|
||||
}
|
||||
|
||||
// ---- Remote Model List Proxy ----
|
||||
|
||||
// ProxyListModels forwards a request to the provider's models endpoint using the stored API key.
|
||||
func (h *ModelConfigHandler) ProxyListModels(c *gin.Context) {
|
||||
providerName := c.Param("name")
|
||||
modelsURL := c.Query("url")
|
||||
if modelsURL == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "missing 'url' query parameter"})
|
||||
return
|
||||
}
|
||||
|
||||
p, err := h.store.GetProvider(providerName)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
if p.APIKey == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "provider 未配置 API Key"})
|
||||
return
|
||||
}
|
||||
|
||||
client := &http.Client{Timeout: 15 * time.Second}
|
||||
req, err := http.NewRequest("GET", modelsURL, nil)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "创建请求失败: " + err.Error()})
|
||||
return
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+p.APIKey)
|
||||
req.Header.Set("Accept", "application/json")
|
||||
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadGateway, gin.H{"error": "请求模型列表失败: " + err.Error()})
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, err := io.ReadAll(io.LimitReader(resp.Body, 2<<20)) // 2 MB limit
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadGateway, gin.H{"error": "读取响应失败: " + err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
c.JSON(http.StatusBadGateway, gin.H{
|
||||
"error": fmt.Sprintf("Provider API 返回错误 (HTTP %d)", resp.StatusCode),
|
||||
"body": string(body),
|
||||
"models_url": modelsURL,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// Parse the response body which may use different formats:
|
||||
// OpenAI: {"object":"list","data":[{"id":"...","object":"model",...}]}
|
||||
// DashScope: {"request_id":"...","data":{"models":[{"model_id":"..."}]}}
|
||||
// Generic: {"data":[{"id":"..."}]} or {"data":[{"model_id":"..."}]}
|
||||
ids := parseModelListResponse(body)
|
||||
if len(ids) == 0 {
|
||||
c.JSON(http.StatusBadGateway, gin.H{
|
||||
"error": "无法从 Provider 响应中解析模型列表 (不支持的格式)",
|
||||
"raw": string(body),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"provider": providerName,
|
||||
"url": modelsURL,
|
||||
"models": ids,
|
||||
"total": len(ids),
|
||||
})
|
||||
}
|
||||
|
||||
// parseModelListResponse attempts to extract model IDs from various provider response formats.
|
||||
// Supported formats:
|
||||
// - OpenAI-compatible: {"object":"list","data":[{"id":"gpt-4o",...}]}
|
||||
// - DashScope: {"data":{"models":[{"model_id":"qwen-turbo",...}]}}
|
||||
// - Generic: {"data":[{"id":"..."}]} or {"data":[{"model_id":"..."}]}
|
||||
func parseModelListResponse(body []byte) []string {
|
||||
var raw map[string]interface{}
|
||||
if err := json.Unmarshal(body, &raw); err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Strategy 1: data is an array of objects — try "id" then "model_id"
|
||||
if dataArr, ok := raw["data"].([]interface{}); ok {
|
||||
ids := extractIDs(dataArr, "id")
|
||||
if len(ids) > 0 {
|
||||
return ids
|
||||
}
|
||||
return extractIDs(dataArr, "model_id")
|
||||
}
|
||||
|
||||
// Strategy 2: data is an object with a "models" array (DashScope format)
|
||||
if dataObj, ok := raw["data"].(map[string]interface{}); ok {
|
||||
if modelsArr, ok := dataObj["models"].([]interface{}); ok {
|
||||
ids := extractIDs(modelsArr, "model_id")
|
||||
if len(ids) > 0 {
|
||||
return ids
|
||||
}
|
||||
return extractIDs(modelsArr, "id")
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func extractIDs(items []interface{}, key string) []string {
|
||||
ids := make([]string, 0, len(items))
|
||||
for _, item := range items {
|
||||
if obj, ok := item.(map[string]interface{}); ok {
|
||||
if v, ok := obj[key]; ok {
|
||||
if s, ok := v.(string); ok && s != "" {
|
||||
ids = append(ids, s)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return ids
|
||||
}
|
||||
@@ -322,8 +322,9 @@ func (h *SessionHandler) GetMessages(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
// 降级:从 Hub 内存缓存读取
|
||||
messages := h.hub.GetConversation("", sessionID)
|
||||
// 降级:从 Hub 内存缓存读取(使用当前认证用户的 ID 作为缓存键前缀)
|
||||
userID := middleware.GetUserID(c)
|
||||
messages := h.hub.GetConversation(userID, sessionID)
|
||||
if messages == nil {
|
||||
messages = []ws.Message{}
|
||||
}
|
||||
|
||||
@@ -15,7 +15,7 @@ import (
|
||||
)
|
||||
|
||||
// Setup 注册所有路由
|
||||
func Setup(r *gin.Engine, hub *ws.Hub, cfg *config.Config, sessionStore *store.SessionStore, reminderStore *store.ReminderStore, briefingStore *store.BriefingStore, automationStore *store.AutomationStore, fileStore *store.FileStore, ruleEngine *engine.RuleEngine, knowledgeStore *store.KnowledgeStore, imageHandler *handler.ImageHandler, db interface{}) {
|
||||
func Setup(r *gin.Engine, hub *ws.Hub, cfg *config.Config, sessionStore *store.SessionStore, reminderStore *store.ReminderStore, briefingStore *store.BriefingStore, automationStore *store.AutomationStore, fileStore *store.FileStore, ruleEngine *engine.RuleEngine, knowledgeStore *store.KnowledgeStore, imageHandler *handler.ImageHandler, db interface{}, modelConfigStore *config.ModelsConfigStore) {
|
||||
// 限流器
|
||||
rateLimiter := middleware.NewRateLimiter(10, 20) // 每秒10个请求,突发20
|
||||
|
||||
@@ -36,6 +36,7 @@ func Setup(r *gin.Engine, hub *ws.Hub, cfg *config.Config, sessionStore *store.S
|
||||
fileHandler := handler.NewFileHandler(fileStore)
|
||||
automationHandler := handler.NewAutomationHandler(automationStore, ruleEngine)
|
||||
knowledgeHandler := handler.NewKnowledgeHandler(knowledgeStore, fileStore)
|
||||
modelConfigHandler := handler.NewModelConfigHandler(modelConfigStore)
|
||||
if imageHandler == nil {
|
||||
imageHandler = handler.NewImageHandler(cfg, fileStore)
|
||||
}
|
||||
@@ -199,6 +200,29 @@ func Setup(r *gin.Engine, hub *ws.Hub, cfg *config.Config, sessionStore *store.S
|
||||
admin.GET("/sessions", sessionHandler.ListActiveSessions)
|
||||
admin.GET("/sessions/active", sessionHandler.GetActiveSessions)
|
||||
admin.GET("/sessions/:id", sessionHandler.GetSession)
|
||||
|
||||
// 多端客户端管理
|
||||
admin.GET("/clients", chatHandler.HandleListClients)
|
||||
admin.PUT("/clients/:id/note", chatHandler.HandleUpdateClientNote)
|
||||
|
||||
// 模型配置管理
|
||||
models := admin.Group("/models")
|
||||
{
|
||||
models.GET("/providers", modelConfigHandler.ListProviders)
|
||||
models.GET("/providers/:name", modelConfigHandler.GetProvider)
|
||||
models.POST("/providers/:name", modelConfigHandler.SetProvider)
|
||||
models.DELETE("/providers/:name", modelConfigHandler.DeleteProvider)
|
||||
models.GET("/models", modelConfigHandler.ListModels)
|
||||
models.GET("/models/:id", modelConfigHandler.GetModel)
|
||||
models.POST("/models/:id", modelConfigHandler.SetModel)
|
||||
models.DELETE("/models/:id", modelConfigHandler.DeleteModel)
|
||||
models.GET("/routing", modelConfigHandler.ListRouting)
|
||||
models.GET("/routing/:purpose", modelConfigHandler.GetRouting)
|
||||
models.POST("/routing/:purpose", modelConfigHandler.SetRouting)
|
||||
models.DELETE("/routing/:purpose", modelConfigHandler.DeleteRouting)
|
||||
models.POST("/health-check", modelConfigHandler.TestProvider)
|
||||
models.GET("/fetch-models/:name", modelConfigHandler.ProxyListModels)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -26,9 +26,21 @@ type Message struct {
|
||||
Role string `json:"role"`
|
||||
MsgType string `json:"msg_type"`
|
||||
Content string `json:"content"`
|
||||
ClientID string `json:"client_id"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
// ClientRecord 客户端记录 (持久化)
|
||||
type ClientRecord struct {
|
||||
ClientID string `json:"client_id"`
|
||||
UserID string `json:"user_id"`
|
||||
DeviceName string `json:"device_name"`
|
||||
UserAgent string `json:"user_agent"`
|
||||
Note string `json:"note"`
|
||||
FirstSeenAt time.Time `json:"first_seen_at"`
|
||||
LastSeenAt time.Time `json:"last_seen_at"`
|
||||
}
|
||||
|
||||
// SessionStore 会话持久化存储
|
||||
type SessionStore struct {
|
||||
db *sql.DB
|
||||
@@ -92,6 +104,19 @@ func (s *SessionStore) migrate() error {
|
||||
|
||||
// 为已存在的数据库添加 msg_type 列 (Phase 0.1)
|
||||
`ALTER TABLE messages ADD COLUMN IF NOT EXISTS msg_type VARCHAR(16) DEFAULT 'chat'`,
|
||||
// 为已存在的数据库添加 client_id 列 (Phase 5: 多端客户端追踪)
|
||||
`ALTER TABLE messages ADD COLUMN IF NOT EXISTS client_id VARCHAR(128) DEFAULT ''`,
|
||||
|
||||
`CREATE TABLE IF NOT EXISTS clients (
|
||||
client_id VARCHAR(128) PRIMARY KEY,
|
||||
user_id VARCHAR(128) NOT NULL,
|
||||
device_name VARCHAR(256) DEFAULT '',
|
||||
user_agent VARCHAR(512) DEFAULT '',
|
||||
note VARCHAR(256) DEFAULT '',
|
||||
first_seen_at TIMESTAMP DEFAULT NOW(),
|
||||
last_seen_at TIMESTAMP DEFAULT NOW()
|
||||
)`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_clients_user_id ON clients(user_id)`,
|
||||
}
|
||||
|
||||
for _, q := range queries {
|
||||
@@ -205,10 +230,10 @@ func (s *SessionStore) DeleteAllUserSessions(userID string) error {
|
||||
}
|
||||
|
||||
// AddMessage 添加一条消息到会话
|
||||
func (s *SessionStore) AddMessage(sessionID, role, msgType, content string) error {
|
||||
func (s *SessionStore) AddMessage(sessionID, role, msgType, content, clientID string) error {
|
||||
_, err := s.db.Exec(
|
||||
`INSERT INTO messages (session_id, role, msg_type, content) VALUES ($1, $2, $3, $4)`,
|
||||
sessionID, role, msgType, content,
|
||||
`INSERT INTO messages (session_id, role, msg_type, content, client_id) VALUES ($1, $2, $3, $4, $5)`,
|
||||
sessionID, role, msgType, content, clientID,
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("添加消息失败: %w", err)
|
||||
@@ -226,7 +251,7 @@ func (s *SessionStore) GetMessages(sessionID string, limit, offset int) ([]Messa
|
||||
}
|
||||
|
||||
rows, err := s.db.Query(
|
||||
`SELECT id, session_id, role, COALESCE(msg_type, 'chat'), content, created_at
|
||||
`SELECT id, session_id, role, COALESCE(msg_type, 'chat'), content, COALESCE(client_id, ''), created_at
|
||||
FROM messages WHERE session_id = $1
|
||||
ORDER BY created_at ASC
|
||||
LIMIT $2 OFFSET $3`,
|
||||
@@ -240,7 +265,7 @@ func (s *SessionStore) GetMessages(sessionID string, limit, offset int) ([]Messa
|
||||
var messages []Message
|
||||
for rows.Next() {
|
||||
var msg Message
|
||||
if err := rows.Scan(&msg.ID, &msg.SessionID, &msg.Role, &msg.MsgType, &msg.Content, &msg.CreatedAt); err != nil {
|
||||
if err := rows.Scan(&msg.ID, &msg.SessionID, &msg.Role, &msg.MsgType, &msg.Content, &msg.ClientID, &msg.CreatedAt); err != nil {
|
||||
return nil, fmt.Errorf("扫描消息行失败: %w", err)
|
||||
}
|
||||
messages = append(messages, msg)
|
||||
@@ -341,3 +366,57 @@ func (s *SessionStore) IsAvailable() bool {
|
||||
}
|
||||
return s.db.Ping() == nil
|
||||
}
|
||||
|
||||
// ========== 多端客户端追踪 ==========
|
||||
|
||||
// UpsertClient inserts or updates a client record.
|
||||
func (s *SessionStore) UpsertClient(clientID, userID, deviceName, userAgent string) error {
|
||||
_, err := s.db.Exec(
|
||||
`INSERT INTO clients (client_id, user_id, device_name, user_agent, first_seen_at, last_seen_at)
|
||||
VALUES ($1, $2, $3, $4, NOW(), NOW())
|
||||
ON CONFLICT (client_id) DO UPDATE SET
|
||||
device_name = EXCLUDED.device_name,
|
||||
user_agent = EXCLUDED.user_agent,
|
||||
last_seen_at = NOW()`,
|
||||
clientID, userID, deviceName, userAgent,
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("upsert client failed: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetClients returns all known clients for a user.
|
||||
func (s *SessionStore) GetClients(userID string) ([]ClientRecord, error) {
|
||||
rows, err := s.db.Query(
|
||||
`SELECT client_id, user_id, device_name, user_agent, note, first_seen_at, last_seen_at
|
||||
FROM clients WHERE user_id = $1 ORDER BY last_seen_at DESC`,
|
||||
userID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("query clients failed: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var result []ClientRecord
|
||||
for rows.Next() {
|
||||
var cr ClientRecord
|
||||
if err := rows.Scan(&cr.ClientID, &cr.UserID, &cr.DeviceName, &cr.UserAgent, &cr.Note, &cr.FirstSeenAt, &cr.LastSeenAt); err != nil {
|
||||
return nil, fmt.Errorf("scan client row failed: %w", err)
|
||||
}
|
||||
result = append(result, cr)
|
||||
}
|
||||
if result == nil {
|
||||
result = []ClientRecord{}
|
||||
}
|
||||
return result, rows.Err()
|
||||
}
|
||||
|
||||
// UpdateClientNote sets the user-defined note for a client.
|
||||
func (s *SessionStore) UpdateClientNote(clientID, note string) error {
|
||||
_, err := s.db.Exec(`UPDATE clients SET note = $1 WHERE client_id = $2`, note, clientID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("update client note failed: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -24,21 +24,27 @@ const (
|
||||
|
||||
// Client WebSocket客户端
|
||||
type Client struct {
|
||||
Hub *Hub
|
||||
Conn *websocket.Conn
|
||||
Send chan []byte
|
||||
UserID string
|
||||
SessionID string
|
||||
Hub *Hub
|
||||
Conn *websocket.Conn
|
||||
Send chan []byte
|
||||
UserID string
|
||||
SessionID string
|
||||
ClientID string
|
||||
DeviceName string
|
||||
UserAgent string
|
||||
}
|
||||
|
||||
// NewClient 创建WebSocket客户端
|
||||
func NewClient(hub *Hub, conn *websocket.Conn, userID, sessionID string) *Client {
|
||||
func NewClient(hub *Hub, conn *websocket.Conn, userID, sessionID, clientID, deviceName, userAgent string) *Client {
|
||||
return &Client{
|
||||
Hub: hub,
|
||||
Conn: conn,
|
||||
Send: make(chan []byte, 256),
|
||||
UserID: userID,
|
||||
SessionID: sessionID,
|
||||
Hub: hub,
|
||||
Conn: conn,
|
||||
Send: make(chan []byte, 256),
|
||||
UserID: userID,
|
||||
SessionID: sessionID,
|
||||
ClientID: clientID,
|
||||
DeviceName: deviceName,
|
||||
UserAgent: userAgent,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -39,10 +39,23 @@ type Message struct {
|
||||
MsgType string `json:"msg_type,omitempty"`
|
||||
Attachments []MessageAttachment `json:"attachments,omitempty"`
|
||||
Timestamp int64 `json:"timestamp"`
|
||||
ClientInfo *ClientInfo `json:"client_info,omitempty"`
|
||||
}
|
||||
|
||||
const maxRecentMessages = 20
|
||||
|
||||
// KnownClient tracks a device that has ever connected (online or offline).
|
||||
type KnownClient struct {
|
||||
ClientID string `json:"client_id"`
|
||||
UserID string `json:"user_id"`
|
||||
DeviceName string `json:"device_name"`
|
||||
UserAgent string `json:"user_agent"`
|
||||
Note string `json:"note"` // user-assigned label
|
||||
Online bool `json:"online"`
|
||||
LastSeenAt time.Time `json:"last_seen_at"`
|
||||
FirstSeenAt time.Time `json:"first_seen_at"`
|
||||
}
|
||||
|
||||
// Hub WebSocket连接池
|
||||
type Hub struct {
|
||||
mu sync.RWMutex
|
||||
@@ -76,6 +89,9 @@ type Hub struct {
|
||||
pendingProactive map[string][]json.RawMessage // userID -> queued messages
|
||||
aiCoreURL string
|
||||
internalToken string
|
||||
|
||||
// 多端客户端追踪: clientID -> KnownClient (在线+离线)
|
||||
knownClients map[string]*KnownClient
|
||||
}
|
||||
|
||||
// SetStore 设置持久化存储 (可选)
|
||||
@@ -100,6 +116,7 @@ func NewHub() *Hub {
|
||||
iotStopCh: make(chan struct{}),
|
||||
idleTimeout: 30 * time.Minute, // 默认30分钟
|
||||
pendingProactive: make(map[string][]json.RawMessage),
|
||||
knownClients: make(map[string]*KnownClient),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -254,6 +271,34 @@ func (h *Hub) Run() {
|
||||
MessageCount: 0,
|
||||
}
|
||||
}
|
||||
|
||||
// 多端客户端追踪
|
||||
if client.ClientID != "" {
|
||||
now := time.Now()
|
||||
if kc, ok := h.knownClients[client.ClientID]; ok {
|
||||
kc.Online = true
|
||||
kc.LastSeenAt = now
|
||||
kc.DeviceName = client.DeviceName
|
||||
kc.UserAgent = client.UserAgent
|
||||
} else {
|
||||
h.knownClients[client.ClientID] = &KnownClient{
|
||||
ClientID: client.ClientID,
|
||||
UserID: client.UserID,
|
||||
DeviceName: client.DeviceName,
|
||||
UserAgent: client.UserAgent,
|
||||
Online: true,
|
||||
LastSeenAt: now,
|
||||
FirstSeenAt: now,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 持久化客户端记录到数据库
|
||||
if client.ClientID != "" && h.store != nil && h.store.IsAvailable() {
|
||||
if err := h.store.UpsertClient(client.ClientID, client.UserID, client.DeviceName, client.UserAgent); err != nil {
|
||||
logger.Printf("[WS] 持久化客户端记录失败: %v", err)
|
||||
}
|
||||
}
|
||||
// Phase 2: 检测是否为重连 (之前处于离线状态)
|
||||
wasOffline := len(h.userClients[client.UserID]) == 1 // 刚加入,之前为0
|
||||
h.mu.Unlock()
|
||||
@@ -308,6 +353,23 @@ func (h *Hub) Run() {
|
||||
s.State = "idle"
|
||||
}
|
||||
}
|
||||
|
||||
// 多端客户端追踪: 检查同一 clientID 是否还有其他连接
|
||||
if client.ClientID != "" {
|
||||
hasOtherClientConn := false
|
||||
for c := range h.clients {
|
||||
if c.ClientID == client.ClientID {
|
||||
hasOtherClientConn = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !hasOtherClientConn {
|
||||
if kc, ok := h.knownClients[client.ClientID]; ok {
|
||||
kc.Online = false
|
||||
kc.LastSeenAt = time.Now()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
h.mu.Unlock()
|
||||
|
||||
@@ -362,6 +424,23 @@ func (h *Hub) Run() {
|
||||
s.State = "idle"
|
||||
}
|
||||
}
|
||||
|
||||
// 多端客户端追踪
|
||||
if client.ClientID != "" {
|
||||
hasOtherClientConn := false
|
||||
for c := range h.clients {
|
||||
if c.ClientID == client.ClientID {
|
||||
hasOtherClientConn = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !hasOtherClientConn {
|
||||
if kc, ok := h.knownClients[client.ClientID]; ok {
|
||||
kc.Online = false
|
||||
kc.LastSeenAt = time.Now()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
h.mu.Unlock()
|
||||
|
||||
@@ -766,3 +845,62 @@ func (h *Hub) DeleteConversation(userID, sessionID string) {
|
||||
key := cacheKey(userID, sessionID)
|
||||
h.conversationCache.Delete(key)
|
||||
}
|
||||
|
||||
// ========== 多端客户端追踪 ==========
|
||||
|
||||
// GetKnownClients returns all known clients (online + offline).
|
||||
func (h *Hub) GetKnownClients(userID string) []KnownClient {
|
||||
h.mu.RLock()
|
||||
defer h.mu.RUnlock()
|
||||
|
||||
result := make([]KnownClient, 0)
|
||||
for _, kc := range h.knownClients {
|
||||
if userID == "" || kc.UserID == userID {
|
||||
cp := *kc
|
||||
cp.UserAgent = "" // don't leak UA in list
|
||||
result = append(result, cp)
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// UpdateClientNote sets a user-defined note/label on a client.
|
||||
func (h *Hub) UpdateClientNote(clientID, note string) bool {
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
|
||||
kc, ok := h.knownClients[clientID]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
kc.Note = note
|
||||
return true
|
||||
}
|
||||
|
||||
// ClientInfo returns the ClientInfo for a given client.
|
||||
func (h *Hub) ClientInfo(clientID string) *ClientInfo {
|
||||
h.mu.RLock()
|
||||
defer h.mu.RUnlock()
|
||||
|
||||
kc, ok := h.knownClients[clientID]
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
return &ClientInfo{
|
||||
ClientID: kc.ClientID,
|
||||
DeviceName: kc.DeviceName,
|
||||
UserAgent: kc.UserAgent,
|
||||
}
|
||||
}
|
||||
|
||||
// buildClientInfo builds a ClientInfo from a Client.
|
||||
func buildClientInfo(c *Client) *ClientInfo {
|
||||
if c.ClientID == "" {
|
||||
return nil
|
||||
}
|
||||
return &ClientInfo{
|
||||
ClientID: c.ClientID,
|
||||
DeviceName: c.DeviceName,
|
||||
UserAgent: c.UserAgent,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -21,6 +21,9 @@ type ClientMessage struct {
|
||||
AudioData string `json:"audio_data,omitempty"` // base64
|
||||
Attachments []MessageAttachment `json:"attachments,omitempty"` // 图片等附件
|
||||
Timestamp int64 `json:"timestamp"`
|
||||
ClientID string `json:"client_id,omitempty"` // 客户端唯一标识 (多端区分)
|
||||
DeviceName string `json:"device_name,omitempty"` // 设备备注名称
|
||||
UserAgent string `json:"user_agent,omitempty"` // 浏览器 UA
|
||||
}
|
||||
|
||||
// ReviewMessage 审查后的结构化消息(动作/聊天分离)
|
||||
@@ -30,6 +33,13 @@ type ReviewMessage struct {
|
||||
DelayMs int `json:"delay_ms,omitempty"` // ms to wait before sending (0 = immediate)
|
||||
}
|
||||
|
||||
// ClientInfo carries the originating client's device metadata.
|
||||
type ClientInfo struct {
|
||||
ClientID string `json:"client_id,omitempty"`
|
||||
DeviceName string `json:"device_name,omitempty"`
|
||||
UserAgent string `json:"user_agent,omitempty"`
|
||||
}
|
||||
|
||||
// 服务端 → 客户端消息
|
||||
type ServerMessage struct {
|
||||
Type string `json:"type"` // response | segment | audio | error | device_update | pong | history_response | stream_chunk | stream_end | background_thinking | notification | multi_message | stream_segments | review | thinking | tool_progress | system_info
|
||||
@@ -55,6 +65,7 @@ type ServerMessage struct {
|
||||
ToolProgress *ToolProgressInfo `json:"tool_progress,omitempty"` // 工具执行进度
|
||||
SystemInfo *SystemInfoPayload `json:"system_info,omitempty"` // 系统通知信息
|
||||
ProtocolVersion int `json:"protocol_version,omitempty"` // 协议版本
|
||||
ClientInfo *ClientInfo `json:"client_info,omitempty"` // 消息来源客户端信息
|
||||
}
|
||||
|
||||
// ToolProgressInfo 工具执行进度
|
||||
|
||||
@@ -20,12 +20,28 @@ import (
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/bridge"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/config"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/handler"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/logging"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/permissions"
|
||||
)
|
||||
|
||||
func main() {
|
||||
cfg := config.Load()
|
||||
|
||||
// Config store for platform adapter configs.
|
||||
configStore, err := config.NewStore("platform_configs.json")
|
||||
if err != nil {
|
||||
fmt.Printf("FATAL: config store: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Message logger.
|
||||
msgLogger, err := logging.NewLogger("logs")
|
||||
if err != nil {
|
||||
fmt.Printf("FATAL: logger: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
defer msgLogger.Close()
|
||||
|
||||
// Core components.
|
||||
mapper := bridge.NewIdentityMapper()
|
||||
checker := permissions.NewChecker()
|
||||
@@ -34,23 +50,58 @@ func main() {
|
||||
// Seed default identities from environment.
|
||||
seedIdentities(mapper)
|
||||
|
||||
// Register platform adapters.
|
||||
adapters := []bridge.PlatformAdapter{
|
||||
qqadapter.NewAdapter(cfg.QQBotPort),
|
||||
telegramadapter.NewAdapter(cfg.TelegramToken, cfg.TelegramWebhookURL),
|
||||
webhookadapter.NewAdapter("webhook"),
|
||||
wechatstub.NewAdapter(),
|
||||
feishustub.NewAdapter(),
|
||||
discordstub.NewAdapter(),
|
||||
}
|
||||
|
||||
// Register platform adapters based on stored configs or defaults.
|
||||
adapters := createAdapters(cfg, configStore)
|
||||
for _, a := range adapters {
|
||||
router.RegisterAdapter(a)
|
||||
}
|
||||
|
||||
// Set message handler: forward to AI-Core.
|
||||
// Set message handler with logging.
|
||||
router.SetMessageHandler(func(msg *bridge.UnifiedMessage) (*bridge.UnifiedResponse, error) {
|
||||
return forwardToAICore(cfg, msg)
|
||||
// Log incoming.
|
||||
msgLogger.Log(logging.LogEntry{
|
||||
Timestamp: time.Now(),
|
||||
Direction: "incoming",
|
||||
Platform: msg.Platform,
|
||||
ChannelID: msg.ChannelID,
|
||||
SenderID: msg.SenderID,
|
||||
SenderName: msg.SenderName,
|
||||
Content: msg.Content,
|
||||
ContentType: msg.ContentType,
|
||||
MessageID: msg.MessageID,
|
||||
Success: true,
|
||||
})
|
||||
|
||||
response, err := forwardToAICore(cfg, msg)
|
||||
if err != nil {
|
||||
msgLogger.Log(logging.LogEntry{
|
||||
Timestamp: time.Now(),
|
||||
Direction: "outgoing",
|
||||
Platform: msg.Platform,
|
||||
ChannelID: msg.ChannelID,
|
||||
SenderID: msg.SenderID,
|
||||
Success: false,
|
||||
Error: err.Error(),
|
||||
})
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// Log outgoing.
|
||||
for _, rm := range response.Messages {
|
||||
msgLogger.Log(logging.LogEntry{
|
||||
Timestamp: time.Now(),
|
||||
Direction: "outgoing",
|
||||
Platform: msg.Platform,
|
||||
ChannelID: msg.ChannelID,
|
||||
SenderID: msg.SenderID,
|
||||
SenderName: "Cyrene",
|
||||
Content: rm.Content,
|
||||
ContentType: "text",
|
||||
Success: true,
|
||||
})
|
||||
}
|
||||
|
||||
return response, nil
|
||||
})
|
||||
|
||||
// Connect all adapters.
|
||||
@@ -68,6 +119,12 @@ func main() {
|
||||
bh := handler.NewBridgeHandler(router)
|
||||
bh.RegisterRoutes(mux)
|
||||
|
||||
// Config and log handlers.
|
||||
ch := handler.NewConfigHandler(configStore, router)
|
||||
ch.RegisterRoutes(mux)
|
||||
lh := handler.NewLogHandler(msgLogger)
|
||||
lh.RegisterRoutes(mux)
|
||||
|
||||
// Start QQ message reader loop.
|
||||
qq, _ := router.GetAdapter("qq")
|
||||
if qqa, ok := qq.(*qqadapter.Adapter); ok {
|
||||
@@ -80,13 +137,12 @@ func main() {
|
||||
fmt.Printf("[qq] route error: %v\n", err)
|
||||
continue
|
||||
}
|
||||
// Send response back through QQ adapter.
|
||||
msgs, err := router.SendResponse(response)
|
||||
if err != nil {
|
||||
fmt.Printf("[qq] send error: %v\n", err)
|
||||
continue
|
||||
}
|
||||
_ = msgs // QQ adapter handles sending via WebSocket
|
||||
_ = msgs
|
||||
}
|
||||
}()
|
||||
}
|
||||
@@ -115,6 +171,76 @@ func main() {
|
||||
fmt.Println("Platform Bridge stopped")
|
||||
}
|
||||
|
||||
// createAdapters builds platform adapters, preferring stored configs over defaults.
|
||||
func createAdapters(cfg *config.Config, store *config.Store) []bridge.PlatformAdapter {
|
||||
allNames := []string{"qq", "telegram", "webhook", "wechat", "feishu", "discord"}
|
||||
var adapters []bridge.PlatformAdapter
|
||||
|
||||
for _, name := range allNames {
|
||||
stored, _ := store.Get(name)
|
||||
if stored != nil && !stored.Enabled {
|
||||
fmt.Printf("Platform %s is disabled in config, skipping\n", name)
|
||||
continue
|
||||
}
|
||||
|
||||
var a bridge.PlatformAdapter
|
||||
fields := mergeFields(cfg, name, stored)
|
||||
|
||||
switch name {
|
||||
case "qq":
|
||||
port := cfg.QQBotPort
|
||||
if p, ok := fields["bot_port"]; ok && p != "" {
|
||||
port = p
|
||||
}
|
||||
a = qqadapter.NewAdapter(port)
|
||||
case "telegram":
|
||||
token := cfg.TelegramToken
|
||||
if t, ok := fields["bot_token"]; ok && t != "" {
|
||||
token = t
|
||||
}
|
||||
webhookURL := cfg.TelegramWebhookURL
|
||||
if w, ok := fields["webhook_url"]; ok && w != "" {
|
||||
webhookURL = w
|
||||
}
|
||||
a = telegramadapter.NewAdapter(token, webhookURL)
|
||||
case "webhook":
|
||||
a = webhookadapter.NewAdapter("webhook")
|
||||
case "wechat":
|
||||
a = wechatstub.NewAdapter()
|
||||
case "feishu":
|
||||
a = feishustub.NewAdapter()
|
||||
case "discord":
|
||||
a = discordstub.NewAdapter()
|
||||
}
|
||||
|
||||
if a != nil {
|
||||
adapters = append(adapters, a)
|
||||
}
|
||||
}
|
||||
return adapters
|
||||
}
|
||||
|
||||
// mergeFields returns fields from stored config, falling back to env defaults.
|
||||
func mergeFields(cfg *config.Config, name string, stored *config.PlatformConfig) map[string]string {
|
||||
fields := make(map[string]string)
|
||||
if stored != nil {
|
||||
for k, v := range stored.Fields {
|
||||
fields[k] = v
|
||||
}
|
||||
}
|
||||
// Apply env var defaults if fields are missing.
|
||||
if fields["bot_token"] == "" && cfg.TelegramToken != "" && name == "telegram" {
|
||||
fields["bot_token"] = cfg.TelegramToken
|
||||
}
|
||||
if fields["webhook_url"] == "" && cfg.TelegramWebhookURL != "" && name == "telegram" {
|
||||
fields["webhook_url"] = cfg.TelegramWebhookURL
|
||||
}
|
||||
if fields["bot_port"] == "" && cfg.QQBotPort != "" && name == "qq" {
|
||||
fields["bot_port"] = cfg.QQBotPort
|
||||
}
|
||||
return fields
|
||||
}
|
||||
|
||||
// forwardToAICore sends a unified message to AI-Core's chat endpoint and returns the response.
|
||||
func forwardToAICore(cfg *config.Config, msg *bridge.UnifiedMessage) (*bridge.UnifiedResponse, error) {
|
||||
reqBody, _ := json.Marshal(map[string]interface{}{
|
||||
@@ -145,13 +271,11 @@ func forwardToAICore(cfg *config.Config, msg *bridge.UnifiedMessage) (*bridge.Un
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// For simplicity, collect full text from SSE stream.
|
||||
var result struct {
|
||||
Content string `json:"content"`
|
||||
Error string `json:"error"`
|
||||
}
|
||||
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
|
||||
// Non-JSON response — read as raw text.
|
||||
buf := new(bytes.Buffer)
|
||||
buf.ReadFrom(resp.Body)
|
||||
return &bridge.UnifiedResponse{
|
||||
@@ -181,7 +305,6 @@ func forwardToAICore(cfg *config.Config, msg *bridge.UnifiedMessage) (*bridge.Un
|
||||
|
||||
// seedIdentities loads default identity mappings.
|
||||
func seedIdentities(m *bridge.IdentityMapper) {
|
||||
// Admin on QQ.
|
||||
if qqAdmin := os.Getenv("QQ_ADMIN_UID"); qqAdmin != "" {
|
||||
m.Register(permissions.PlatformIdentity{
|
||||
Platform: "qq",
|
||||
@@ -191,7 +314,6 @@ func seedIdentities(m *bridge.IdentityMapper) {
|
||||
PermissionLevel: "admin",
|
||||
})
|
||||
}
|
||||
// Admin on Telegram.
|
||||
if tgAdmin := os.Getenv("TELEGRAM_ADMIN_UID"); tgAdmin != "" {
|
||||
m.Register(permissions.PlatformIdentity{
|
||||
Platform: "telegram",
|
||||
|
||||
@@ -0,0 +1,120 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// PlatformConfig holds persistent configuration for one platform adapter.
|
||||
type PlatformConfig struct {
|
||||
Name string `json:"name"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Label string `json:"label"`
|
||||
Fields map[string]string `json:"fields"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// Store manages persistence of platform configs to a JSON file.
|
||||
type Store struct {
|
||||
mu sync.RWMutex
|
||||
path string
|
||||
configs map[string]*PlatformConfig
|
||||
}
|
||||
|
||||
// NewStore creates a Store, creating the config file if it doesn't exist.
|
||||
func NewStore(path string) (*Store, error) {
|
||||
s := &Store{
|
||||
path: path,
|
||||
configs: make(map[string]*PlatformConfig),
|
||||
}
|
||||
if err := s.load(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (s *Store) load() error {
|
||||
data, err := os.ReadFile(s.path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
// Initialize empty file.
|
||||
return s.save()
|
||||
}
|
||||
return fmt.Errorf("read config file: %w", err)
|
||||
}
|
||||
if len(data) == 0 {
|
||||
return nil
|
||||
}
|
||||
if err := json.Unmarshal(data, &s.configs); err != nil {
|
||||
return fmt.Errorf("parse config file: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Store) save() error {
|
||||
data, err := json.MarshalIndent(s.configs, "", " ")
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal configs: %w", err)
|
||||
}
|
||||
tmpPath := s.path + ".tmp"
|
||||
if err := os.WriteFile(tmpPath, data, 0640); err != nil {
|
||||
return fmt.Errorf("write config file: %w", err)
|
||||
}
|
||||
return os.Rename(tmpPath, s.path)
|
||||
}
|
||||
|
||||
// List returns all platform configs.
|
||||
func (s *Store) List() []PlatformConfig {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
result := make([]PlatformConfig, 0, len(s.configs))
|
||||
for _, c := range s.configs {
|
||||
result = append(result, *c)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// Get returns a single platform config.
|
||||
func (s *Store) Get(name string) (*PlatformConfig, error) {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
c, ok := s.configs[name]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("config not found: %s", name)
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
|
||||
// Set upserts a platform config and persists.
|
||||
func (s *Store) Set(cfg PlatformConfig) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if cfg.Fields == nil {
|
||||
cfg.Fields = make(map[string]string)
|
||||
}
|
||||
cfg.UpdatedAt = time.Now()
|
||||
s.configs[cfg.Name] = &cfg
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// Delete removes a platform config and persists.
|
||||
func (s *Store) Delete(name string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if _, ok := s.configs[name]; !ok {
|
||||
return fmt.Errorf("config not found: %s", name)
|
||||
}
|
||||
delete(s.configs, name)
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// HasConfig checks if a config exists for the given platform.
|
||||
func (s *Store) HasConfig(name string) bool {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
_, ok := s.configs[name]
|
||||
return ok
|
||||
}
|
||||
@@ -0,0 +1,179 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/bridge"
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/config"
|
||||
)
|
||||
|
||||
var knownPlatforms = map[string]bool{
|
||||
"qq": true, "telegram": true, "webhook": true,
|
||||
"wechat": true, "feishu": true, "discord": true,
|
||||
}
|
||||
|
||||
// ConfigHandler exposes CRUD endpoints for platform configs.
|
||||
type ConfigHandler struct {
|
||||
store *config.Store
|
||||
router *bridge.PlatformRouter
|
||||
}
|
||||
|
||||
func NewConfigHandler(store *config.Store, router *bridge.PlatformRouter) *ConfigHandler {
|
||||
return &ConfigHandler{store: store, router: router}
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) RegisterRoutes(mux *http.ServeMux) {
|
||||
mux.HandleFunc("/api/v1/configs", h.listConfigs)
|
||||
mux.HandleFunc("/api/v1/configs/", h.handleConfig)
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) listConfigs(w http.ResponseWriter, r *http.Request) {
|
||||
configs := h.store.List()
|
||||
|
||||
type configSummary struct {
|
||||
Name string `json:"name"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Label string `json:"label,omitempty"`
|
||||
Fields map[string]string `json:"fields"`
|
||||
UpdatedAt string `json:"updated_at"`
|
||||
Connected bool `json:"connected"`
|
||||
}
|
||||
|
||||
var result []configSummary
|
||||
for _, c := range configs {
|
||||
connected := false
|
||||
if a, err := h.router.GetAdapter(c.Name); err == nil {
|
||||
connected = a.IsConnected()
|
||||
}
|
||||
result = append(result, configSummary{
|
||||
Name: c.Name,
|
||||
Enabled: c.Enabled,
|
||||
Label: c.Label,
|
||||
Fields: c.Fields,
|
||||
UpdatedAt: c.UpdatedAt.Format("2006-01-02T15:04:05Z07:00"),
|
||||
Connected: connected,
|
||||
})
|
||||
}
|
||||
// Also include platforms that exist as adapters but have no config yet.
|
||||
for _, name := range h.router.ListAdapters() {
|
||||
found := false
|
||||
for _, c := range configs {
|
||||
if c.Name == name {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
connected := false
|
||||
if a, err := h.router.GetAdapter(name); err == nil {
|
||||
connected = a.IsConnected()
|
||||
}
|
||||
result = append(result, configSummary{
|
||||
Name: name,
|
||||
Enabled: false,
|
||||
Fields: map[string]string{},
|
||||
Connected: connected,
|
||||
})
|
||||
}
|
||||
}
|
||||
if result == nil {
|
||||
result = []configSummary{}
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"configs": result,
|
||||
"total": len(result),
|
||||
})
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) handleConfig(w http.ResponseWriter, r *http.Request) {
|
||||
name := r.URL.Path[len("/api/v1/configs/"):]
|
||||
if name == "" {
|
||||
writeJSON(w, http.StatusBadRequest, errResp("missing config name"))
|
||||
return
|
||||
}
|
||||
if !knownPlatforms[name] {
|
||||
writeJSON(w, http.StatusBadRequest, errResp("unknown platform: "+name))
|
||||
return
|
||||
}
|
||||
|
||||
switch r.Method {
|
||||
case "GET":
|
||||
h.getConfig(w, r, name)
|
||||
case "POST", "PUT":
|
||||
h.saveConfig(w, r, name)
|
||||
case "DELETE":
|
||||
h.deleteConfig(w, r, name)
|
||||
default:
|
||||
writeJSON(w, http.StatusMethodNotAllowed, errResp("method not allowed"))
|
||||
}
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) getConfig(w http.ResponseWriter, r *http.Request, name string) {
|
||||
cfg, err := h.store.Get(name)
|
||||
if err != nil {
|
||||
writeJSON(w, http.StatusNotFound, errResp(err.Error()))
|
||||
return
|
||||
}
|
||||
connected := false
|
||||
if a, err := h.router.GetAdapter(name); err == nil {
|
||||
connected = a.IsConnected()
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"name": cfg.Name,
|
||||
"enabled": cfg.Enabled,
|
||||
"label": cfg.Label,
|
||||
"fields": cfg.Fields,
|
||||
"updated_at": cfg.UpdatedAt.Format("2006-01-02T15:04:05Z07:00"),
|
||||
"connected": connected,
|
||||
})
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) saveConfig(w http.ResponseWriter, r *http.Request, name string) {
|
||||
var body struct {
|
||||
Enabled *bool `json:"enabled"`
|
||||
Label string `json:"label"`
|
||||
Fields map[string]string `json:"fields"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
||||
writeJSON(w, http.StatusBadRequest, errResp("invalid JSON: "+err.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
enabled := true
|
||||
if body.Enabled != nil {
|
||||
enabled = *body.Enabled
|
||||
}
|
||||
|
||||
fields := body.Fields
|
||||
if fields == nil {
|
||||
fields = make(map[string]string)
|
||||
}
|
||||
|
||||
cfg := config.PlatformConfig{
|
||||
Name: name,
|
||||
Enabled: enabled,
|
||||
Label: body.Label,
|
||||
Fields: fields,
|
||||
}
|
||||
if err := h.store.Set(cfg); err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, errResp(err.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"name": name,
|
||||
"enabled": enabled,
|
||||
"label": body.Label,
|
||||
"fields": fields,
|
||||
"status": "saved",
|
||||
})
|
||||
}
|
||||
|
||||
func (h *ConfigHandler) deleteConfig(w http.ResponseWriter, r *http.Request, name string) {
|
||||
if err := h.store.Delete(name); err != nil {
|
||||
writeJSON(w, http.StatusNotFound, errResp(err.Error()))
|
||||
return
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]string{"status": "deleted", "name": name})
|
||||
}
|
||||
@@ -0,0 +1,50 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
"github.com/yourname/cyrene-ai/platform-bridge/internal/logging"
|
||||
)
|
||||
|
||||
// LogHandler exposes message log retrieval endpoints.
|
||||
type LogHandler struct {
|
||||
logger *logging.Logger
|
||||
}
|
||||
|
||||
func NewLogHandler(logger *logging.Logger) *LogHandler {
|
||||
return &LogHandler{logger: logger}
|
||||
}
|
||||
|
||||
func (h *LogHandler) RegisterRoutes(mux *http.ServeMux) {
|
||||
mux.HandleFunc("/api/v1/logs/", h.handleLogs)
|
||||
}
|
||||
|
||||
func (h *LogHandler) handleLogs(w http.ResponseWriter, r *http.Request) {
|
||||
name := r.URL.Path[len("/api/v1/logs/"):]
|
||||
if name == "" {
|
||||
writeJSON(w, http.StatusBadRequest, errResp("missing platform name in path"))
|
||||
return
|
||||
}
|
||||
|
||||
limit := 100
|
||||
if l := r.URL.Query().Get("limit"); l != "" {
|
||||
if n, err := strconv.Atoi(l); err == nil && n > 0 && n <= 1000 {
|
||||
limit = n
|
||||
}
|
||||
}
|
||||
|
||||
entries, err := h.logger.ReadLogs(name, limit)
|
||||
if err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, errResp(err.Error()))
|
||||
return
|
||||
}
|
||||
if entries == nil {
|
||||
entries = []logging.LogEntry{}
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"platform": name,
|
||||
"total": len(entries),
|
||||
"logs": entries,
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,149 @@
|
||||
package logging
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// LogEntry represents one message log record.
|
||||
type LogEntry struct {
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
Direction string `json:"direction"` // "incoming" or "outgoing"
|
||||
Platform string `json:"platform"`
|
||||
ChannelID string `json:"channel_id"`
|
||||
SenderID string `json:"sender_id"`
|
||||
SenderName string `json:"sender_name"`
|
||||
Content string `json:"content"`
|
||||
ContentType string `json:"content_type"`
|
||||
MessageID string `json:"message_id,omitempty"`
|
||||
Success bool `json:"success"`
|
||||
Error string `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
// Logger writes message logs to per-platform JSONL files.
|
||||
type Logger struct {
|
||||
mu sync.Mutex
|
||||
dir string
|
||||
files map[string]*os.File
|
||||
}
|
||||
|
||||
// NewLogger creates a Logger, ensuring the log directory exists.
|
||||
func NewLogger(dir string) (*Logger, error) {
|
||||
if err := os.MkdirAll(dir, 0750); err != nil {
|
||||
return nil, fmt.Errorf("create log dir: %w", err)
|
||||
}
|
||||
return &Logger{
|
||||
dir: dir,
|
||||
files: make(map[string]*os.File),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Log writes a log entry to the appropriate platform log file.
|
||||
func (l *Logger) Log(entry LogEntry) error {
|
||||
if entry.Timestamp.IsZero() {
|
||||
entry.Timestamp = time.Now()
|
||||
}
|
||||
|
||||
f, err := l.getOrCreateFile(entry.Platform)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
data, err := json.Marshal(entry)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal log entry: %w", err)
|
||||
}
|
||||
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
|
||||
if _, err := f.Write(append(data, '\n')); err != nil {
|
||||
return fmt.Errorf("write log: %w", err)
|
||||
}
|
||||
return f.Sync()
|
||||
}
|
||||
|
||||
// ReadLogs reads the last N log entries for a platform, newest first.
|
||||
func (l *Logger) ReadLogs(platform string, limit int) ([]LogEntry, error) {
|
||||
if limit <= 0 || limit > 1000 {
|
||||
limit = 1000
|
||||
}
|
||||
|
||||
l.mu.Lock()
|
||||
// Flush any pending writes to the file before reading.
|
||||
if f, ok := l.files[platform]; ok {
|
||||
f.Sync()
|
||||
}
|
||||
l.mu.Unlock()
|
||||
|
||||
path := filepath.Join(l.dir, platform+".log")
|
||||
f, err := os.Open(path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return []LogEntry{}, nil
|
||||
}
|
||||
return nil, fmt.Errorf("open log file: %w", err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
// Read all lines, keep only the last `limit`.
|
||||
var lines []string
|
||||
scanner := bufio.NewScanner(f)
|
||||
scanner.Buffer(make([]byte, 0, 64*1024), 1024*1024)
|
||||
for scanner.Scan() {
|
||||
lines = append(lines, scanner.Text())
|
||||
}
|
||||
if err := scanner.Err(); err != nil {
|
||||
return nil, fmt.Errorf("read log file: %w", err)
|
||||
}
|
||||
|
||||
// Take last N lines and reverse.
|
||||
start := len(lines) - limit
|
||||
if start < 0 {
|
||||
start = 0
|
||||
}
|
||||
lines = lines[start:]
|
||||
|
||||
entries := make([]LogEntry, 0, len(lines))
|
||||
for i := len(lines) - 1; i >= 0; i-- {
|
||||
var entry LogEntry
|
||||
if err := json.Unmarshal([]byte(lines[i]), &entry); err != nil {
|
||||
continue // Skip corrupted lines.
|
||||
}
|
||||
entries = append(entries, entry)
|
||||
}
|
||||
return entries, nil
|
||||
}
|
||||
|
||||
// Close closes all open log file handles.
|
||||
func (l *Logger) Close() error {
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
for _, f := range l.files {
|
||||
f.Close()
|
||||
}
|
||||
l.files = make(map[string]*os.File)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (l *Logger) getOrCreateFile(platform string) (*os.File, error) {
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
|
||||
if f, ok := l.files[platform]; ok {
|
||||
return f, nil
|
||||
}
|
||||
|
||||
path := filepath.Join(l.dir, platform+".log")
|
||||
f, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0640)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("open log file %s: %w", path, err)
|
||||
}
|
||||
l.files[platform] = f
|
||||
return f, nil
|
||||
}
|
||||
@@ -22,7 +22,7 @@ func (h *PluginHandler) RegisterRoutes(mux *http.ServeMux) {
|
||||
mux.HandleFunc("/api/v1/plugins/", h.pluginRoute)
|
||||
mux.HandleFunc("/api/v1/tools", h.listTools)
|
||||
mux.HandleFunc("/api/v1/tools/", h.toolRoute)
|
||||
mux.HandleFunc("/health", h.health)
|
||||
mux.HandleFunc("/api/v1/health", h.health)
|
||||
}
|
||||
|
||||
func (h *PluginHandler) health(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
+888
-2
@@ -684,6 +684,16 @@ input[type="range"] { accent-color: var(--accent); padding: 0; }
|
||||
<button class="nav-item" data-panel="timeline">
|
||||
<span class="nav-icon">⏱️</span><span class="nav-label">记忆时间线</span>
|
||||
</button>
|
||||
<button class="nav-item" data-panel="chatPlatforms">
|
||||
<span class="nav-icon">💬</span><span class="nav-label">第三方聊天</span>
|
||||
</button>
|
||||
<button class="nav-item" data-panel="clients">
|
||||
<span class="nav-icon">📱</span><span class="nav-label">客户端管理</span>
|
||||
<span class="nav-badge" id="clients-badge" style="display:none">0</span>
|
||||
</button>
|
||||
<button class="nav-item" data-panel="modelConfig">
|
||||
<span class="nav-icon">🤖</span><span class="nav-label">模型配置</span>
|
||||
</button>
|
||||
</nav>
|
||||
<div class="sidebar-footer">
|
||||
<span id="ws-dot" class="disconnected"></span>
|
||||
@@ -720,6 +730,11 @@ input[type="range"] { accent-color: var(--accent); padding: 0; }
|
||||
<div class="panel" id="panel-thinking"></div>
|
||||
<!-- 记忆时间线 -->
|
||||
<div class="panel" id="panel-timeline"></div>
|
||||
<!-- 第三方聊天 -->
|
||||
<div class="panel" id="panel-chatPlatforms"></div>
|
||||
<!-- 客户端管理 -->
|
||||
<div class="panel" id="panel-clients"></div>
|
||||
<div class="panel" id="panel-modelConfig"></div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
@@ -772,8 +787,19 @@ const STATE = {
|
||||
timelineFilterType: 'all',
|
||||
timelineAutoRefresh: null,
|
||||
timelineLimit: 100,
|
||||
// 第三方聊天
|
||||
chatConfigsAutoRefresh: null,
|
||||
chatConfigs: [],
|
||||
chatActivePlatform: null,
|
||||
chatLogLimit: 100,
|
||||
// 自主思考面板:记录展开的日志 ID
|
||||
expandedThinkingLogs: {},
|
||||
// 模型配置
|
||||
modelConfigTab: 'providers',
|
||||
modelConfigProviders: [],
|
||||
modelConfigModels: [],
|
||||
modelConfigRouting: [],
|
||||
fetchedModels: [],
|
||||
expandedThinkingLogs: {},
|
||||
};
|
||||
|
||||
// ========== WebSocket ==========
|
||||
@@ -968,6 +994,11 @@ document.getElementById('toggle-sidebar').addEventListener('click', () => {
|
||||
function switchPanel(name) {
|
||||
STATE.activePanel = name;
|
||||
|
||||
// Update URL hash (without triggering hashchange).
|
||||
if (location.hash !== '#' + name) {
|
||||
history.replaceState(null, '', '#' + name);
|
||||
}
|
||||
|
||||
// 更新侧边栏
|
||||
document.querySelectorAll('.nav-item').forEach(b => b.classList.remove('active'));
|
||||
const navBtn = document.querySelector(`.nav-item[data-panel="${name}"]`);
|
||||
@@ -978,6 +1009,9 @@ function switchPanel(name) {
|
||||
dashboard: '🏠 仪表盘', memory: '🧠 记忆管理', sessions: '💬 会话监看',
|
||||
services: '🖥 服务管理', iot: '🏠 IoT 设备控制', performance: '📊 性能监控', database: '🗄️ 数据库监看',
|
||||
toolCalls: '🔧 工具调用记录', stt: '🎤 语音识别日志', thinking: '💭 自主思考', timeline: '⏱️ 记忆时间线',
|
||||
chatPlatforms: '💬 第三方聊天配置与消息日志',
|
||||
clients: '📱 客户端管理',
|
||||
modelConfig: '🤖 模型配置管理',
|
||||
};
|
||||
document.getElementById('panel-title').textContent = titles[name] || name;
|
||||
|
||||
@@ -1001,6 +1035,9 @@ function switchPanel(name) {
|
||||
case 'stt': renderSTTPanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); stopTimelineAutoRefresh(); break;
|
||||
case 'thinking': renderThinkingPanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); stopTimelineAutoRefresh(); break;
|
||||
case 'timeline': renderTimelinePanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); startTimelineAutoRefresh(); break;
|
||||
case 'chatPlatforms': renderChatPlatformsPanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); stopTimelineAutoRefresh(); startChatAutoRefresh(); break;
|
||||
case 'clients': renderClientsPanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); stopTimelineAutoRefresh(); break;
|
||||
case 'modelConfig': renderModelConfigPanel(); stopSessionsAutoRefresh(); stopDashboardAutoRefresh(); stopDbAutoRefresh(); stopIoTRefresh(); stopTimelineAutoRefresh(); break;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -3439,13 +3476,862 @@ function toggleTimelineAutoRefresh(on) {
|
||||
}
|
||||
}
|
||||
|
||||
// ========== 面板: 第三方聊天配置 ==========
|
||||
|
||||
var PLATFORM_FIELDS = {
|
||||
qq: [{ key: 'bot_port', label: 'Bot WebSocket 端口', placeholder: '8096' }],
|
||||
telegram: [
|
||||
{ key: 'bot_token', label: 'Bot Token', placeholder: '123456:ABC-DEF...' },
|
||||
{ key: 'webhook_url', label: 'Webhook URL', placeholder: 'https://your-domain.com' }
|
||||
],
|
||||
webhook: [
|
||||
{ key: 'webhook_url', label: 'Webhook URL', placeholder: 'https://hook.example.com/chat' },
|
||||
{ key: 'secret', label: 'Secret Token', placeholder: '(可选)' }
|
||||
],
|
||||
wechat: [
|
||||
{ key: 'corp_id', label: '企业ID (Corp ID)', placeholder: 'ww...' },
|
||||
{ key: 'corp_secret', label: '应用Secret', placeholder: '' },
|
||||
{ key: 'agent_id', label: 'Agent ID', placeholder: '1000001' },
|
||||
{ key: 'webhook_url', label: 'Webhook URL', placeholder: 'https://...' }
|
||||
],
|
||||
feishu: [
|
||||
{ key: 'app_id', label: 'App ID', placeholder: 'cli_...' },
|
||||
{ key: 'app_secret', label: 'App Secret', placeholder: '' },
|
||||
{ key: 'verification_token', label: 'Verification Token', placeholder: '' },
|
||||
{ key: 'webhook_url', label: 'Webhook URL', placeholder: 'https://...' }
|
||||
],
|
||||
discord: [
|
||||
{ key: 'bot_token', label: 'Bot Token', placeholder: 'MT...' },
|
||||
{ key: 'application_id', label: 'Application ID', placeholder: '123456789...' }
|
||||
]
|
||||
};
|
||||
|
||||
var PLATFORM_ICONS = { qq: '🐧', telegram: '✈️', webhook: '🪝', wechat: '💚', feishu: '🕊️', discord: '🎮' };
|
||||
var PLATFORM_LABELS = { qq: 'QQ', telegram: 'Telegram', webhook: 'Webhook', wechat: 'WeChat', feishu: 'Feishu', discord: 'Discord' };
|
||||
|
||||
function startChatAutoRefresh() {
|
||||
stopChatAutoRefresh();
|
||||
STATE.chatConfigsAutoRefresh = setInterval(function() {
|
||||
if (STATE.activePanel === 'chatPlatforms') {
|
||||
loadChatConfigs();
|
||||
if (STATE.chatActivePlatform) refreshChatLogs(STATE.chatActivePlatform);
|
||||
}
|
||||
}, 10000);
|
||||
}
|
||||
|
||||
function stopChatAutoRefresh() {
|
||||
if (STATE.chatConfigsAutoRefresh) { clearInterval(STATE.chatConfigsAutoRefresh); STATE.chatConfigsAutoRefresh = null; }
|
||||
}
|
||||
|
||||
function renderChatPlatformsPanel() {
|
||||
if (STATE.chatActivePlatform) { renderChatPlatformDetail(STATE.chatActivePlatform); return; }
|
||||
var panel = document.getElementById('panel-chatPlatforms');
|
||||
panel.innerHTML = '<div class="card"><div class="card-header"><span class="card-title">🔗 平台配置列表</span>' +
|
||||
'<button class="btn btn-sm btn-accent" onclick="showChatConfigForm()">+ 添加配置</button></div>' +
|
||||
'<div class="table-wrap"><table id="chat-configs-table"><thead><tr>' +
|
||||
'<th>平台</th><th>启用</th><th>连接</th><th>关键配置</th><th>更新时间</th><th>操作</th>' +
|
||||
'</tr></thead><tbody id="chat-configs-tbody">' +
|
||||
'<tr><td colspan="6"><div class="empty-state"><div class="icon">💬</div>加载中...</div></td></tr></tbody></table></div></div>';
|
||||
document.getElementById('panel-actions').innerHTML = '<button class="btn btn-sm" onclick="refreshChatConfigs()">🔄 刷新</button>';
|
||||
loadChatConfigs();
|
||||
}
|
||||
|
||||
async function loadChatConfigs() {
|
||||
var data = await api('/api/chat-platforms/configs');
|
||||
var tbody = document.getElementById('chat-configs-tbody');
|
||||
if (!tbody) return;
|
||||
if (data.error) { tbody.innerHTML = '<tr><td colspan="6"><div class="empty-state"><div class="icon">⚠️</div>' + escHtml(data.error) + '</div></td></tr>'; return; }
|
||||
STATE.chatConfigs = data.configs || [];
|
||||
renderChatConfigsTable();
|
||||
}
|
||||
|
||||
function renderChatConfigsTable() {
|
||||
var tbody = document.getElementById('chat-configs-tbody');
|
||||
if (!tbody) return;
|
||||
var configs = STATE.chatConfigs;
|
||||
if (configs.length === 0) {
|
||||
tbody.innerHTML = '<tr><td colspan="6"><div class="empty-state"><div class="icon">💬</div>暂无配置,点击「添加配置」创建</div></td></tr>';
|
||||
return;
|
||||
}
|
||||
tbody.innerHTML = configs.map(function(c) {
|
||||
var icon = PLATFORM_ICONS[c.name] || '🔗';
|
||||
var label = c.label || PLATFORM_LABELS[c.name] || c.name;
|
||||
var connBadge = c.connected ? '<span class="badge badge-running">已连接</span>' : '<span class="badge badge-stopped">未连接</span>';
|
||||
var enabledBadge = c.enabled !== false ? '<span class="badge badge-running">启用</span>' : '<span class="badge badge-stopped">禁用</span>';
|
||||
var keys = (c.fields && Object.keys(c.fields).length > 0)
|
||||
? Object.keys(c.fields).map(function(k) { return k + '=' + (c.fields[k] ? '***' : '(空)'); }).join(', ')
|
||||
: '—';
|
||||
var updated = c.updated_at ? timeAgo(c.updated_at) : '—';
|
||||
return '<tr>' +
|
||||
'<td><strong>' + icon + ' ' + escHtml(label) + '</strong></td>' +
|
||||
'<td>' + enabledBadge + '</td>' +
|
||||
'<td>' + connBadge + '</td>' +
|
||||
'<td style="max-width:200px;overflow:hidden;text-overflow:ellipsis;white-space:nowrap">' + escHtml(keys) + '</td>' +
|
||||
'<td>' + updated + '</td>' +
|
||||
'<td><div class="btn-group">' +
|
||||
'<button class="btn btn-xs" onclick="editChatConfig(\'' + escHtml(c.name) + '\')">✏️ 编辑</button>' +
|
||||
'<button class="btn btn-xs btn-red" onclick="deleteChatConfig(\'' + escHtml(c.name) + '\')">🗑</button>' +
|
||||
'</div></td></tr>';
|
||||
}).join('');
|
||||
}
|
||||
|
||||
function refreshChatConfigs() { loadChatConfigs(); }
|
||||
|
||||
function showChatConfigForm() {
|
||||
var panel = document.getElementById('panel-chatPlatforms');
|
||||
var options = ['qq', 'telegram', 'webhook', 'wechat', 'feishu', 'discord'];
|
||||
panel.innerHTML = '<div class="card"><div class="card-header"><span class="card-title">➕ 选择要配置的平台</span>' +
|
||||
'<button class="btn btn-sm" onclick="STATE.chatActivePlatform=null;renderChatPlatformsPanel();">← 取消</button></div>' +
|
||||
'<div class="cards-grid cards-3">' +
|
||||
options.map(function(p) {
|
||||
return '<div class="card" style="cursor:pointer;text-align:center;padding:20px" onclick="startNewConfig(\'' + p + '\')">' +
|
||||
'<div style="font-size:32px;margin-bottom:8px">' + (PLATFORM_ICONS[p] || '🔗') + '</div>' +
|
||||
'<div style="font-weight:600">' + (PLATFORM_LABELS[p] || p) + '</div></div>';
|
||||
}).join('') + '</div></div>';
|
||||
document.getElementById('panel-actions').innerHTML = '';
|
||||
}
|
||||
|
||||
function startNewConfig(name) { STATE.chatActivePlatform = name; renderChatPlatformsPanel(); }
|
||||
|
||||
function editChatConfig(name) {
|
||||
if (!STATE.chatConfigs.some(function(c) { return c.name === name; })) {
|
||||
STATE.chatActivePlatform = name;
|
||||
renderChatPlatformsPanel();
|
||||
} else {
|
||||
STATE.chatActivePlatform = name;
|
||||
renderChatPlatformsPanel();
|
||||
}
|
||||
}
|
||||
|
||||
function renderChatPlatformDetail(name) {
|
||||
var cfg = null;
|
||||
for (var i = 0; i < STATE.chatConfigs.length; i++) {
|
||||
if (STATE.chatConfigs[i].name === name) { cfg = STATE.chatConfigs[i]; break; }
|
||||
}
|
||||
var icon = PLATFORM_ICONS[name] || '🔗';
|
||||
var panel = document.getElementById('panel-chatPlatforms');
|
||||
panel.innerHTML =
|
||||
'<div style="margin-bottom:14px"><button class="btn btn-sm" onclick="STATE.chatActivePlatform=null;renderChatPlatformsPanel();">← 返回列表</button></div>' +
|
||||
'<div class="card"><div class="card-header"><span class="card-title">' + icon + ' ' + escHtml(name) + ' 配置</span><span id="cfg-save-status"></span></div>' +
|
||||
'<div class="card-body" id="chat-config-form"></div></div>' +
|
||||
'<div class="card" style="margin-top:14px"><div class="card-header"><span class="card-title">📋 消息日志 (最近 ' + STATE.chatLogLimit + ' 条)</span>' +
|
||||
'<div class="btn-group">' +
|
||||
'<button class="btn btn-xs" onclick="refreshChatLogs(\'' + escHtml(name) + '\')">🔄 刷新</button>' +
|
||||
'<select id="chat-log-limit" onchange="STATE.chatLogLimit=parseInt(this.value);refreshChatLogs(\'' + escHtml(name) + '\')" ' +
|
||||
'style="width:auto;padding:4px 8px;font-size:11px;background:var(--bg3);color:var(--text);border:1px solid var(--border);border-radius:4px">' +
|
||||
'<option value="50">50条</option><option value="100" selected>100条</option><option value="200">200条</option><option value="500">500条</option></select></div></div>' +
|
||||
'<div id="chat-log-container" style="max-height:400px;overflow-y:auto;background:var(--bg);border:1px solid var(--border);border-radius:var(--radius-sm);padding:8px">' +
|
||||
'<div class="empty-state"><div class="icon">📝</div>加载中...</div></div></div>';
|
||||
document.getElementById('panel-actions').innerHTML = '';
|
||||
renderChatConfigForm(name, cfg);
|
||||
refreshChatLogs(name);
|
||||
}
|
||||
|
||||
function renderChatConfigForm(name, cfg) {
|
||||
var fields = PLATFORM_FIELDS[name] || [];
|
||||
var container = document.getElementById('chat-config-form');
|
||||
if (!container) return;
|
||||
var currentFields = (cfg && cfg.fields) || {};
|
||||
var enabled = cfg ? (cfg.enabled !== false) : true;
|
||||
var fieldsHTML = fields.map(function(f) {
|
||||
var val = currentFields[f.key] || '';
|
||||
return '<div class="form-group"><label>' + escHtml(f.label) + '</label>' +
|
||||
'<input type="text" id="cfg-field-' + escHtml(f.key) + '" value="' + escHtml(val) + '" placeholder="' + escHtml(f.placeholder || '') + '"></div>';
|
||||
}).join('');
|
||||
container.innerHTML =
|
||||
'<div class="form-group"><label style="display:flex;align-items:center;gap:8px;cursor:pointer">' +
|
||||
'<input type="checkbox" id="cfg-field-enabled" ' + (enabled ? 'checked' : '') + ' style="width:auto"><span>启用此平台</span></label></div>' +
|
||||
fieldsHTML +
|
||||
'<div class="form-group"><label>显示名称</label>' +
|
||||
'<input type="text" id="cfg-field-label" value="' + escHtml((cfg && cfg.label) || '') + '" placeholder="' + escHtml(name) + '"></div>' +
|
||||
'<div class="btn-group" style="margin-top:12px"><button class="btn btn-sm btn-accent" onclick="saveChatConfig(\'' + escHtml(name) + '\')">💾 保存配置</button></div>';
|
||||
}
|
||||
|
||||
async function saveChatConfig(name) {
|
||||
var fields = {};
|
||||
var fieldDefs = PLATFORM_FIELDS[name] || [];
|
||||
fieldDefs.forEach(function(f) {
|
||||
var el = document.getElementById('cfg-field-' + f.key);
|
||||
if (el) fields[f.key] = el.value;
|
||||
});
|
||||
var enabledEl = document.getElementById('cfg-field-enabled');
|
||||
var enabled = enabledEl ? enabledEl.checked : true;
|
||||
var labelEl = document.getElementById('cfg-field-label');
|
||||
var label = labelEl ? labelEl.value : '';
|
||||
var data = await api('/api/chat-platforms/configs/' + encodeURIComponent(name), {
|
||||
method: 'POST',
|
||||
body: JSON.stringify({ name: name, enabled: enabled, label: label, fields: fields })
|
||||
});
|
||||
if (data.error) { showToast('保存失败: ' + data.error, 'error'); return; }
|
||||
showToast('配置已保存 (需重启平台桥接服务生效)', 'success');
|
||||
await loadChatConfigs();
|
||||
renderChatPlatformDetail(name);
|
||||
}
|
||||
|
||||
async function deleteChatConfig(name) {
|
||||
if (!confirm('确认删除 ' + name + ' 的配置?')) return;
|
||||
var data = await api('/api/chat-platforms/configs/' + encodeURIComponent(name), { method: 'DELETE' });
|
||||
if (data.error) { showToast('删除失败: ' + data.error, 'error'); return; }
|
||||
showToast('配置已删除', 'success');
|
||||
STATE.chatActivePlatform = null;
|
||||
await loadChatConfigs();
|
||||
renderChatPlatformsPanel();
|
||||
}
|
||||
|
||||
async function refreshChatLogs(name) {
|
||||
var limit = STATE.chatLogLimit || 100;
|
||||
var data = await api('/api/chat-platforms/logs/' + encodeURIComponent(name) + '?limit=' + limit);
|
||||
var container = document.getElementById('chat-log-container');
|
||||
if (!container) return;
|
||||
if (data.error) { container.innerHTML = '<div class="empty-state"><div class="icon">⚠️</div>' + escHtml(data.error) + '</div>'; return; }
|
||||
var logs = data.logs || [];
|
||||
STATE.chatLogs = STATE.chatLogs || {};
|
||||
STATE.chatLogs[name] = logs;
|
||||
if (logs.length === 0) { container.innerHTML = '<div class="empty-state"><div class="icon">📝</div>暂无消息日志</div>'; return; }
|
||||
container.innerHTML = logs.map(function(l) {
|
||||
var arrow = l.direction === 'incoming' ? '← 收到' : '→ 发送';
|
||||
var color = l.direction === 'incoming' ? 'var(--blue)' : 'var(--green)';
|
||||
var time = new Date(l.timestamp).toLocaleString('zh-CN', { hour12: false });
|
||||
var content = (l.content || '').length > 300 ? (l.content || '').substring(0, 297) + '...' : (l.content || '');
|
||||
return '<div style="padding:6px 10px;border-bottom:1px solid var(--border);font-size:12px">' +
|
||||
'<span style="color:' + color + ';font-weight:600">' + arrow + '</span> ' +
|
||||
'<span style="color:var(--text3)">' + time + '</span> ' +
|
||||
'<span style="color:var(--text2)">[' + escHtml(l.sender_name || l.sender_id || '-') + ']</span> ' +
|
||||
'<span>' + escHtml(content) + '</span>' +
|
||||
(l.error ? ' <span style="color:var(--red)">⚠ ' + escHtml(l.error) + '</span>' : '') +
|
||||
'</div>';
|
||||
}).join('');
|
||||
}
|
||||
|
||||
// ========== 模型配置管理面板 ==========
|
||||
|
||||
function renderModelConfigPanel() {
|
||||
var panel = document.getElementById('panel-modelConfig');
|
||||
var activeTab = STATE.modelConfigTab || 'providers';
|
||||
|
||||
var tabs = [
|
||||
{ id: 'providers', label: '🔌 模型提供商' },
|
||||
{ id: 'models', label: '🧠 模型定义' },
|
||||
{ id: 'routing', label: '🔀 用途路由' },
|
||||
];
|
||||
|
||||
var tabBar = '<div class="tab-bar" style="margin-bottom:14px;display:flex;gap:6px;flex-wrap:wrap">' +
|
||||
tabs.map(function(t) {
|
||||
return '<button class="btn btn-sm' + (activeTab === t.id ? ' btn-accent' : '') +
|
||||
'" onclick="STATE.modelConfigTab=\'' + t.id + '\';renderModelConfigPanel();">' + t.label + '</button>';
|
||||
}).join('') + '</div>';
|
||||
|
||||
panel.innerHTML = tabBar + '<div id="model-config-content" class="card"><div class="card-body">加载中...</div></div>';
|
||||
document.getElementById('panel-actions').innerHTML = '';
|
||||
|
||||
switch (activeTab) {
|
||||
case 'providers': renderProvidersTab(); break;
|
||||
case 'models': renderModelsTab(); break;
|
||||
case 'routing': renderRoutingTab(); break;
|
||||
}
|
||||
}
|
||||
|
||||
// ---- Providers tab ----
|
||||
|
||||
async function renderProvidersTab() {
|
||||
var container = document.getElementById('model-config-content');
|
||||
var data = await api('/api/model-config/providers');
|
||||
STATE.modelConfigProviders = data.providers || [];
|
||||
|
||||
var rows = STATE.modelConfigProviders.length === 0
|
||||
? '<tr><td colspan="5"><div class="empty-state"><div class="icon">🔌</div>暂无模型提供商,请添加</div></td></tr>'
|
||||
: STATE.modelConfigProviders.map(function(p) {
|
||||
var updated = p.updated_at ? timeAgo(p.updated_at) : '—';
|
||||
return '<tr>' +
|
||||
'<td><strong>' + escHtml(p.name) + '</strong></td>' +
|
||||
'<td style="max-width:280px;overflow:hidden;text-overflow:ellipsis;white-space:nowrap">' + escHtml(p.base_url) + '</td>' +
|
||||
'<td>' + (p.timeout_sec || '—') + 's</td>' +
|
||||
'<td>' + updated + '</td>' +
|
||||
'<td><div class="btn-group">' +
|
||||
'<button class="btn btn-xs" onclick="showProviderForm(\'' + escHtml(p.name) + '\')">✏️</button>' +
|
||||
'<button class="btn btn-xs btn-red" onclick="deleteModelProvider(\'' + escHtml(p.name) + '\')">🗑</button>' +
|
||||
'</div></td></tr>';
|
||||
}).join('');
|
||||
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">🔌 模型提供商</span>' +
|
||||
'<button class="btn btn-sm btn-accent" onclick="showProviderForm()">+ 添加</button></div>' +
|
||||
'<div class="table-wrap"><table><thead><tr>' +
|
||||
'<th>名称</th><th>Base URL</th><th>超时</th><th>更新时间</th><th>操作</th>' +
|
||||
'</tr></thead><tbody>' + rows + '</tbody></table></div>';
|
||||
}
|
||||
|
||||
var PROVIDER_TEMPLATES = [
|
||||
{ name: 'deepseek', label: 'DeepSeek', base_url: 'https://api.deepseek.com/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.deepseek.com/models' },
|
||||
{ name: 'dashscope', label: '阿里百炼 (DashScope)', base_url: 'https://dashscope.aliyuncs.com/compatible-mode/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://dashscope.aliyuncs.com/compatible-mode/v1/models' },
|
||||
{ name: 'zhipu', label: '智谱 AI (GLM)', base_url: 'https://open.bigmodel.cn/api/paas/v4', timeout_sec: 120, max_retries: 3, models_url: 'https://open.bigmodel.cn/api/paas/v4/models' },
|
||||
{ name: 'moonshot', label: 'Moonshot (Kimi)', base_url: 'https://api.moonshot.cn/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.moonshot.cn/v1/models' },
|
||||
{ name: 'siliconflow', label: '硅基流动 (SiliconFlow)', base_url: 'https://api.siliconflow.cn/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.siliconflow.cn/v1/models' },
|
||||
{ name: 'lingyi', label: '零一万物', base_url: 'https://api.lingyiwanwu.com/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.lingyiwanwu.com/v1/models' },
|
||||
{ name: 'qianfan', label: '百度千帆', base_url: 'https://qianfan.baidubce.com/v2', timeout_sec: 120, max_retries: 3, models_url: 'https://qianfan.baidubce.com/v2/models' },
|
||||
{ name: 'xfyun', label: '讯飞星火', base_url: 'https://spark-api-open.xf-yun.com/v1', timeout_sec: 120, max_retries: 3, models_url: '' },
|
||||
{ name: 'minimax', label: 'MiniMax', base_url: 'https://api.minimax.chat/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.minimax.chat/v1/models' },
|
||||
{ name: 'openai', label: 'OpenAI', base_url: 'https://api.openai.com/v1', timeout_sec: 120, max_retries: 3, models_url: 'https://api.openai.com/v1/models' },
|
||||
{ name: 'custom', label: '💡 自定义...', base_url: '', timeout_sec: 120, max_retries: 3, models_url: '' },
|
||||
];
|
||||
|
||||
var MODEL_TEMPLATES = {
|
||||
deepseek: ['deepseek-chat', 'deepseek-reasoner'],
|
||||
dashscope: ['qwen-turbo', 'qwen-plus', 'qwen-max', 'qwen-max-longcontext', 'qwen-vl-plus', 'qwen-coder-turbo'],
|
||||
zhipu: ['glm-4-flash', 'glm-4-plus', 'glm-4-long', 'glm-4v-flash', 'glm-4-air'],
|
||||
moonshot: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
|
||||
siliconflow: ['Qwen/Qwen3-235B-A22B', 'Qwen/Qwen2.5-72B-Instruct', 'Qwen/Qwen2.5-7B-Instruct', 'deepseek-ai/DeepSeek-V3', 'Pro/THUDM/glm-4-9b-chat'],
|
||||
lingyi: ['yi-large', 'yi-medium', 'yi-lightning', 'yi-vision'],
|
||||
qianfan: ['ernie-speed-128k', 'ernie-4.0-8k', 'ernie-3.5-8k', 'ernie-speed-pro-128k'],
|
||||
xfyun: ['spark-lite', 'spark-pro-128k', 'spark-max', 'spark-4.0-ultra'],
|
||||
minimax: ['abab6.5s-chat', 'abab6.5-chat'],
|
||||
openai: ['gpt-4o', 'gpt-4o-mini', 'gpt-4.1', 'o3-mini'],
|
||||
};
|
||||
|
||||
function showProviderForm(name) {
|
||||
var existing = null;
|
||||
if (name) {
|
||||
for (var i = 0; i < STATE.modelConfigProviders.length; i++) {
|
||||
if (STATE.modelConfigProviders[i].name === name) { existing = STATE.modelConfigProviders[i]; break; }
|
||||
}
|
||||
}
|
||||
var isEdit = !!existing;
|
||||
var formTitle = isEdit ? '✏️ 编辑 ' + escHtml(name) : '➕ 添加模型提供商';
|
||||
var defaults = existing || { name: '', base_url: 'https://api.deepseek.com/v1', api_key: '', timeout_sec: 120, max_retries: 3 };
|
||||
|
||||
var templateOptions = PROVIDER_TEMPLATES.map(function(t) {
|
||||
return '<option value="' + t.name + '" data-url="' + escHtml(t.base_url) + '" data-timeout="' + t.timeout_sec + '" data-retries="' + t.max_retries + '">' + escHtml(t.label) + '</option>';
|
||||
}).join('');
|
||||
|
||||
var container = document.getElementById('model-config-content');
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">' + formTitle + '</span>' +
|
||||
'<button class="btn btn-sm" onclick="renderProvidersTab()">← 返回</button></div>' +
|
||||
'<div class="card-body"><form onsubmit="event.preventDefault();saveProviderForm(\'' + escHtml(name || '') + '\');">' +
|
||||
(isEdit ? '' :
|
||||
'<div class="form-row"><label>📋 快速模板</label>' +
|
||||
'<select id="prov-template" class="input" onchange="applyProviderTemplate(this.value)" style="background:var(--bg3)">' +
|
||||
'<option value="">-- 选择提供商模板自动填充 --</option>' + templateOptions + '</select></div>') +
|
||||
'<div class="form-row"><label>Provider 名称 ' + (isEdit ? '' : '<span style="color:var(--red)">*</span>') + '</label>' +
|
||||
'<input id="prov-name" class="input" value="' + escHtml(defaults.name) + '" ' + (isEdit ? 'readonly' : 'placeholder="如 deepseek, openai"') + ' required></div>' +
|
||||
'<div class="form-row"><label>Base URL <span style="color:var(--red)">*</span></label>' +
|
||||
'<input id="prov-url" class="input" value="' + escHtml(defaults.base_url) + '" placeholder="https://api.deepseek.com/v1" required></div>' +
|
||||
'<div class="form-row"><label>API Key</label>' +
|
||||
'<input id="prov-key" class="input" type="password" value="' + escHtml(defaults.api_key || '') + '" placeholder="sk-xxx"></div>' +
|
||||
'<div class="form-row" style="display:flex;gap:12px"><div style="flex:1"><label>超时 (秒)</label>' +
|
||||
'<input id="prov-timeout" class="input" type="number" value="' + (defaults.timeout_sec || 120) + '"></div>' +
|
||||
'<div style="flex:1"><label>最大重试</label>' +
|
||||
'<input id="prov-retries" class="input" type="number" value="' + (defaults.max_retries || 3) + '"></div></div>' +
|
||||
'<div style="margin-top:14px"><button type="submit" class="btn btn-accent">💾 保存</button></div>' +
|
||||
'</form></div>';
|
||||
}
|
||||
|
||||
function applyProviderTemplate(templateName) {
|
||||
if (!templateName || templateName === 'custom') return;
|
||||
var sel = document.getElementById('prov-template');
|
||||
var opt = sel.options[sel.selectedIndex];
|
||||
document.getElementById('prov-name').value = opt.value;
|
||||
document.getElementById('prov-url').value = opt.getAttribute('data-url') || '';
|
||||
var timeout = parseInt(opt.getAttribute('data-timeout')) || 120;
|
||||
var retries = parseInt(opt.getAttribute('data-retries')) || 3;
|
||||
var timeoutEl = document.getElementById('prov-timeout');
|
||||
var retriesEl = document.getElementById('prov-retries');
|
||||
if (timeoutEl && !timeoutEl.value) timeoutEl.value = timeout;
|
||||
if (retriesEl && !retriesEl.value) retriesEl.value = retries;
|
||||
}
|
||||
|
||||
function updateModelTemplateOptions() {
|
||||
var provider = document.getElementById('model-provider').value;
|
||||
var area = document.getElementById('model-template-area');
|
||||
if (!area) return;
|
||||
STATE.fetchedModels = [];
|
||||
var models = MODEL_TEMPLATES[provider] || [];
|
||||
area.innerHTML =
|
||||
'<select id="model-template" class="input" onchange="applyModelTemplate(this.value)" style="background:var(--bg3)">' +
|
||||
'<option value="">-- 选择模型模板 / 查询获取 --</option>' +
|
||||
models.map(function(m) { return '<option value="' + escHtml(m) + '">' + escHtml(m) + '</option>'; }).join('') +
|
||||
'</select>';
|
||||
var btn = document.getElementById('btn-fetch-models');
|
||||
if (!btn) return;
|
||||
var tmpl = null;
|
||||
for (var i = 0; i < PROVIDER_TEMPLATES.length; i++) {
|
||||
if (PROVIDER_TEMPLATES[i].name === provider) { tmpl = PROVIDER_TEMPLATES[i]; break; }
|
||||
}
|
||||
btn.disabled = !(tmpl && tmpl.models_url);
|
||||
}
|
||||
|
||||
|
||||
function applyModelTemplate(modelName) {
|
||||
if (!modelName) return;
|
||||
document.getElementById('model-name').value = modelName;
|
||||
var idEl = document.getElementById('model-id');
|
||||
if (idEl && !idEl.value) idEl.value = modelName.replace(/[^a-zA-Z0-9_-]/g, '_').toLowerCase();
|
||||
}
|
||||
|
||||
async function fetchProviderModels() {
|
||||
var provider = document.getElementById('model-provider').value;
|
||||
if (!provider) { alert('请先选择 Provider'); return; }
|
||||
var tmpl = null;
|
||||
for (var i = 0; i < PROVIDER_TEMPLATES.length; i++) {
|
||||
if (PROVIDER_TEMPLATES[i].name === provider) { tmpl = PROVIDER_TEMPLATES[i]; break; }
|
||||
}
|
||||
if (!tmpl || !tmpl.models_url) { alert('该 Provider 不支持在线查询模型列表(讯飞星火等使用非标准接口)'); return; }
|
||||
var btn = document.getElementById('btn-fetch-models');
|
||||
if (btn) { btn.disabled = true; btn.textContent = '⏳ 查询中...'; }
|
||||
try {
|
||||
var result = await api('/api/model-config/fetch-models/' + encodeURIComponent(provider) + '?url=' + encodeURIComponent(tmpl.models_url));
|
||||
if (result.error) { alert('查询失败: ' + result.error + (result.body ? '\n' + result.body.substring(0, 200) : '')); return; }
|
||||
var models = result.models || [];
|
||||
if (models.length === 0) { alert('该 Provider 未返回任何模型'); return; }
|
||||
STATE.fetchedModels = models;
|
||||
renderFetchedModelList(models, '');
|
||||
} catch(e) {
|
||||
alert('查询模型列表出错: ' + e.message);
|
||||
} finally {
|
||||
if (btn) { btn.disabled = false; btn.textContent = '🔍 查询'; }
|
||||
}
|
||||
}
|
||||
|
||||
function renderFetchedModelList(models, filter) {
|
||||
var area = document.getElementById('model-template-area');
|
||||
if (!area) return;
|
||||
var filterLower = (filter || '').toLowerCase();
|
||||
var filtered = filterLower ? models.filter(function(m) { return m.toLowerCase().indexOf(filterLower) >= 0; }) : models;
|
||||
var countInfo = filterLower ? '\uff08' + filtered.length + '/' + models.length + '\uff09' : '\uff08共 ' + models.length + ' \u4e2a\uff09';
|
||||
var html = '<div style="display:flex;gap:8px;margin-bottom:8px">' +
|
||||
'<input id="model-search-input" class="input" type="text" placeholder="\U0001f50d \u641c\u7d22\u6a21\u578b\u540d\u79f0...' + countInfo + '" value="' + escHtml(filter) + '"' +
|
||||
' oninput="renderFetchedModelList(STATE.fetchedModels, this.value)" style="flex:1;background:var(--bg);font-size:12px">' +
|
||||
'<button type="button" class="btn btn-xs" onclick="var s=document.getElementById("model-search-input");if(s)s.value="";renderFetchedModelList(STATE.fetchedModels,"");" title="\u6e05\u9664\u641c\u7d22">\u2715</button></div>' +
|
||||
'<div id="fetched-model-list" style="max-height:220px;overflow-y:auto;border:1px solid var(--border);border-radius:var(--radius-sm);background:var(--bg)">';
|
||||
if (filtered.length === 0) {
|
||||
html += '<div class="empty-state" style="padding:12px"><div class="icon">\U0001f50d</div>\u65e0\u5339\u914d\u6a21\u578b</div>';
|
||||
} else {
|
||||
html += filtered.map(function(m) {
|
||||
return '<div class="fetched-model-item"' +
|
||||
' data-model="' + escHtml(m) + '"' +
|
||||
' onclick="var mn=this.getAttribute("data-model");selectFetchedModel(mn);"' +
|
||||
' style="padding:6px 12px;cursor:pointer;font-size:12px;border-bottom:1px solid var(--border);transition:background .12s"' +
|
||||
' onmouseenter="this.style.background="var(--bg3)""' +
|
||||
' onmouseleave="this.style.background=""">' + escHtml(m) + '</div>';
|
||||
}).join('');
|
||||
}
|
||||
html += '</div>';
|
||||
area.innerHTML = html;
|
||||
}
|
||||
|
||||
|
||||
function selectFetchedModel(modelName) {
|
||||
document.getElementById('model-name').value = modelName;
|
||||
var idEl = document.getElementById('model-id');
|
||||
if (idEl && !idEl.value) idEl.value = modelName.replace(/[^a-zA-Z0-9_-]/g, '_').toLowerCase();
|
||||
var items = document.querySelectorAll('.fetched-model-item');
|
||||
for (var i = 0; i < items.length; i++) {
|
||||
items[i].style.background = (items[i].textContent === modelName) ? 'var(--accent-bg)' : '';
|
||||
}
|
||||
}
|
||||
|
||||
function resetModelTemplateArea() {
|
||||
var area = document.getElementById('model-template-area');
|
||||
if (!area) return;
|
||||
STATE.fetchedModels = [];
|
||||
var provider = document.getElementById('model-provider').value;
|
||||
var models = MODEL_TEMPLATES[provider] || [];
|
||||
area.innerHTML =
|
||||
'<select id="model-template" class="input" onchange="applyModelTemplate(this.value)" style="background:var(--bg3)">' +
|
||||
'<option value="">-- 选择模型模板 / 查询获取 --</option>' +
|
||||
models.map(function(m) { return '<option value="' + escHtml(m) + '">' + escHtml(m) + '</option>'; }).join('') +
|
||||
'</select>';
|
||||
}
|
||||
|
||||
async function saveProviderForm(name) {
|
||||
var data = {
|
||||
name: document.getElementById('prov-name').value.trim(),
|
||||
base_url: document.getElementById('prov-url').value.trim(),
|
||||
api_key: document.getElementById('prov-key').value,
|
||||
timeout_sec: parseInt(document.getElementById('prov-timeout').value) || 120,
|
||||
max_retries: parseInt(document.getElementById('prov-retries').value) || 3,
|
||||
};
|
||||
var saveName = name || data.name;
|
||||
if (!saveName || !data.base_url) { alert('名称和 Base URL 为必填项'); return; }
|
||||
var result = await api('/api/model-config/providers/' + encodeURIComponent(saveName), { method: 'POST', body: JSON.stringify(data) });
|
||||
if (result.error) { alert('保存失败: ' + result.error); return; }
|
||||
STATE.modelConfigTab = 'providers';
|
||||
renderModelConfigPanel();
|
||||
}
|
||||
|
||||
async function deleteModelProvider(name) {
|
||||
if (!confirm('确定删除 Provider "' + name + '"?\n注意:关联的模型和路由也会受影响。')) return;
|
||||
var result = await api('/api/model-config/providers/' + encodeURIComponent(name), { method: 'DELETE' });
|
||||
if (result.error) { alert('删除失败: ' + result.error); return; }
|
||||
renderProvidersTab();
|
||||
}
|
||||
|
||||
// ---- Models tab ----
|
||||
|
||||
async function renderModelsTab() {
|
||||
var container = document.getElementById('model-config-content');
|
||||
var data = await api('/api/model-config/models');
|
||||
STATE.modelConfigModels = data.models || [];
|
||||
|
||||
var rows = STATE.modelConfigModels.length === 0
|
||||
? '<tr><td colspan="6"><div class="empty-state"><div class="icon">🧠</div>暂无模型定义,请添加</div></td></tr>'
|
||||
: STATE.modelConfigModels.map(function(m) {
|
||||
var enabledBadge = m.enabled !== false ? '<span class="badge badge-running">启用</span>' : '<span class="badge badge-stopped">禁用</span>';
|
||||
var tags = (m.tags && m.tags.length > 0) ? m.tags.join(', ') : '—';
|
||||
var updated = m.updated_at ? timeAgo(m.updated_at) : '—';
|
||||
return '<tr>' +
|
||||
'<td><strong>' + escHtml(m.id) + '</strong></td>' +
|
||||
'<td>' + escHtml(m.name) + '</td>' +
|
||||
'<td>' + escHtml(m.provider) + '</td>' +
|
||||
'<td>' + enabledBadge + '</td>' +
|
||||
'<td style="max-width:150px;overflow:hidden;text-overflow:ellipsis;white-space:nowrap">' + escHtml(tags) + '</td>' +
|
||||
'<td><div class="btn-group">' +
|
||||
'<button class="btn btn-xs" onclick="showModelForm(\'' + escHtml(m.id) + '\')">✏️</button>' +
|
||||
'<button class="btn btn-xs btn-red" onclick="deleteModelConfig(\'' + escHtml(m.id) + '\')">🗑</button>' +
|
||||
'</div></td></tr>';
|
||||
}).join('');
|
||||
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">🧠 模型定义</span>' +
|
||||
'<button class="btn btn-sm btn-accent" onclick="showModelForm()">+ 添加</button></div>' +
|
||||
'<div class="table-wrap"><table><thead><tr>' +
|
||||
'<th>ID</th><th>模型名</th><th>Provider</th><th>状态</th><th>标签</th><th>操作</th>' +
|
||||
'</tr></thead><tbody>' + rows + '</tbody></table></div>';
|
||||
}
|
||||
|
||||
function showModelForm(id) {
|
||||
var existing = null;
|
||||
if (id) {
|
||||
for (var i = 0; i < STATE.modelConfigModels.length; i++) {
|
||||
if (STATE.modelConfigModels[i].id === id) { existing = STATE.modelConfigModels[i]; break; }
|
||||
}
|
||||
}
|
||||
var isEdit = !!existing;
|
||||
var formTitle = isEdit ? '✏️ 编辑模型 ' + escHtml(id) : '➕ 添加模型';
|
||||
var defaults = existing || { id: '', name: '', provider: '', description: '', priority: 0, tags: [], params: {}, enabled: true };
|
||||
var tagsStr = (defaults.tags && defaults.tags.length > 0) ? defaults.tags.join(', ') : '';
|
||||
var paramsStr = JSON.stringify(defaults.params || {}, null, 2);
|
||||
|
||||
var container = document.getElementById('model-config-content');
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">' + formTitle + '</span>' +
|
||||
'<button class="btn btn-sm" onclick="renderModelsTab()">← 返回</button></div>' +
|
||||
'<div class="card-body"><form onsubmit="event.preventDefault();saveModelForm(\'' + escHtml(id || '') + '\');">' +
|
||||
'<div class="form-row"><label>模型 ID ' + (isEdit ? '' : '<span style="color:var(--red)">*</span>') + '</label>' +
|
||||
'<input id="model-id" class="input" value="' + escHtml(defaults.id) + '" ' + (isEdit ? 'readonly' : 'placeholder="如 primary_chat"') + ' required></div>' +
|
||||
'<div class="form-row"><label>模型名称 <span style="color:var(--red)">*</span></label>' +
|
||||
'<input id="model-name" class="input" value="' + escHtml(defaults.name) + '" placeholder="deepseek-v4-flash" required></div>' +
|
||||
'<div class="form-row"><label>Provider <span style="color:var(--red)">*</span></label>' +
|
||||
'<select id="model-provider" class="input" required onchange="updateModelTemplateOptions()"><option value="">-- 选择 Provider --</option>' +
|
||||
STATE.modelConfigProviders.map(function(p) {
|
||||
return '<option value="' + escHtml(p.name) + '"' + (defaults.provider === p.name ? ' selected' : '') + '>' + escHtml(p.name) + '</option>';
|
||||
}).join('') + '</select></div>' +
|
||||
(isEdit ? '' :
|
||||
'<div class="form-row"><label>📋 快速模板</label>' +
|
||||
'<div style="display:flex;gap:8px">' +
|
||||
'<div id="model-template-area" style="flex:1"><select id="model-template" class="input" onchange="applyModelTemplate(this.value)" style="background:var(--bg3)">' +
|
||||
'<option value="">-- 选择模型模板 / 查询获取 --</option></select></div>' +
|
||||
'<button type="button" class="btn btn-sm" id="btn-fetch-models" onclick="fetchProviderModels()" style="white-space:nowrap" disabled>🔍 查询</button></div>' +
|
||||
'<div style="font-size:11px;color:var(--text3);margin-top:4px">选择 Provider 后可用模板或点击查询在线获取模型列表</div></div>') +
|
||||
'<div class="form-row"><label>描述</label>' +
|
||||
'<input id="model-desc" class="input" value="' + escHtml(defaults.description || '') + '" placeholder="用于日常对话的模型"></div>' +
|
||||
'<div class="form-row" style="display:flex;gap:12px"><div style="flex:1"><label>优先级</label>' +
|
||||
'<input id="model-priority" class="input" type="number" value="' + (defaults.priority || 0) + '"></div>' +
|
||||
'<div style="flex:1;display:flex;align-items:flex-end;padding-bottom:4px"><label style="display:flex;align-items:center;gap:6px;cursor:pointer">' +
|
||||
'<input type="checkbox" id="model-enabled"' + (defaults.enabled !== false ? ' checked' : '') + '> 启用</label></div></div>' +
|
||||
'<div class="form-row"><label>标签 (逗号分隔)</label>' +
|
||||
'<input id="model-tags" class="input" value="' + escHtml(tagsStr) + '" placeholder="chat, fast"></div>' +
|
||||
'<div class="form-row"><label>模型参数 (JSON)</label>' +
|
||||
'<textarea id="model-params" class="input" rows="3" style="font-family:monospace;font-size:12px">' + escHtml(paramsStr) + '</textarea></div>' +
|
||||
'<div style="margin-top:14px"><button type="submit" class="btn btn-accent">💾 保存</button></div>' +
|
||||
'</form></div>';
|
||||
}
|
||||
|
||||
async function saveModelForm(id) {
|
||||
var tagsStr = document.getElementById('model-tags').value.trim();
|
||||
var tags = tagsStr ? tagsStr.split(',').map(function(t) { return t.trim(); }).filter(Boolean) : [];
|
||||
var paramsStr = document.getElementById('model-params').value.trim();
|
||||
var params = {};
|
||||
try { if (paramsStr) params = JSON.parse(paramsStr); } catch(e) { alert('模型参数 JSON 格式错误: ' + e.message); return; }
|
||||
|
||||
var data = {
|
||||
id: document.getElementById('model-id').value.trim(),
|
||||
name: document.getElementById('model-name').value.trim(),
|
||||
provider: document.getElementById('model-provider').value,
|
||||
description: document.getElementById('model-desc').value.trim(),
|
||||
priority: parseInt(document.getElementById('model-priority').value) || 0,
|
||||
enabled: document.getElementById('model-enabled').checked,
|
||||
tags: tags,
|
||||
params: params,
|
||||
};
|
||||
var saveId = id || data.id;
|
||||
if (!saveId || !data.name || !data.provider) { alert('模型 ID、名称和 Provider 为必填项'); return; }
|
||||
var result = await api('/api/model-config/models/' + encodeURIComponent(saveId), { method: 'POST', body: JSON.stringify(data) });
|
||||
if (result.error) { alert('保存失败: ' + result.error); return; }
|
||||
STATE.modelConfigTab = 'models';
|
||||
renderModelConfigPanel();
|
||||
}
|
||||
|
||||
async function deleteModelConfig(id) {
|
||||
if (!confirm('确定删除模型 "' + id + '"?')) return;
|
||||
var result = await api('/api/model-config/models/' + encodeURIComponent(id), { method: 'DELETE' });
|
||||
if (result.error) { alert('删除失败: ' + result.error); return; }
|
||||
renderModelsTab();
|
||||
}
|
||||
|
||||
// ---- Routing tab ----
|
||||
|
||||
async function renderRoutingTab() {
|
||||
var container = document.getElementById('model-config-content');
|
||||
var data = await api('/api/model-config/routing');
|
||||
STATE.modelConfigRouting = data.routing || [];
|
||||
|
||||
var rows = STATE.modelConfigRouting.length === 0
|
||||
? '<tr><td colspan="4"><div class="empty-state"><div class="icon">🔀</div>暂无路由规则,请添加</div></td></tr>'
|
||||
: STATE.modelConfigRouting.map(function(r) {
|
||||
var chain = (r.fallback_chain && r.fallback_chain.length > 0) ? r.fallback_chain.join(' → ') : '—';
|
||||
var requiredBadge = r.required ? '<span class="badge badge-running">必需</span>' : '<span class="badge badge-stopped">可选</span>';
|
||||
return '<tr>' +
|
||||
'<td><strong>' + escHtml(r.purpose) + '</strong></td>' +
|
||||
'<td style="max-width:300px;overflow:hidden;text-overflow:ellipsis;white-space:nowrap">' + escHtml(chain) + '</td>' +
|
||||
'<td>' + requiredBadge + '</td>' +
|
||||
'<td><div class="btn-group">' +
|
||||
'<button class="btn btn-xs" onclick="showRoutingForm(\'' + escHtml(r.purpose) + '\')">✏️</button>' +
|
||||
'<button class="btn btn-xs btn-red" onclick="deleteRoutingRule(\'' + escHtml(r.purpose) + '\')">🗑</button>' +
|
||||
'</div></td></tr>';
|
||||
}).join('');
|
||||
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">🔀 用途路由</span>' +
|
||||
'<button class="btn btn-sm btn-accent" onclick="showRoutingForm()">+ 添加</button></div>' +
|
||||
'<div class="table-wrap"><table><thead><tr>' +
|
||||
'<th>用途</th><th>回退链</th><th>必需性</th><th>操作</th>' +
|
||||
'</tr></thead><tbody>' + rows + '</tbody></table></div>';
|
||||
}
|
||||
|
||||
function showRoutingForm(purpose) {
|
||||
var existing = null;
|
||||
if (purpose) {
|
||||
for (var i = 0; i < STATE.modelConfigRouting.length; i++) {
|
||||
if (STATE.modelConfigRouting[i].purpose === purpose) { existing = STATE.modelConfigRouting[i]; break; }
|
||||
}
|
||||
}
|
||||
var isEdit = !!existing;
|
||||
var formTitle = isEdit ? '✏️ 编辑路由 ' + escHtml(purpose) : '➕ 添加路由';
|
||||
var defaults = existing || { purpose: '', fallback_chain: [], required: false };
|
||||
var existingChain = defaults.fallback_chain || [];
|
||||
|
||||
var models = STATE.modelConfigModels;
|
||||
var modelCheckboxes = '';
|
||||
if (models.length === 0) {
|
||||
modelCheckboxes = '<div class="empty-state" style="padding:16px"><div class="icon">🧠</div>暂无模型定义,请先在「模型定义」标签中添加模型</div>';
|
||||
} else {
|
||||
modelCheckboxes = '<div style="display:grid;grid-template-columns:repeat(auto-fill,minmax(240px,1fr));gap:6px;max-height:260px;overflow-y:auto;padding:4px 0">' +
|
||||
models.map(function(m) {
|
||||
var checked = existingChain.indexOf(m.id) >= 0 ? ' checked' : '';
|
||||
var providerLabel = m.provider ? ' <span style="color:var(--text3);font-size:11px">(' + escHtml(m.provider) + ')</span>' : '';
|
||||
return '<label style="display:flex;align-items:center;gap:8px;padding:6px 10px;background:var(--bg3);border-radius:var(--radius-sm);cursor:pointer;font-size:12px;transition:background .15s" onmouseenter="this.style.background=\'var(--bg4)\'" onmouseleave="this.style.background=\'var(--bg3)\'">' +
|
||||
'<input type="checkbox" name="routing-model" value="' + escHtml(m.id) + '"' + checked + ' style="accent-color:var(--accent)">' +
|
||||
'<span style="flex:1"><strong>' + escHtml(m.name || m.id) + '</strong>' + providerLabel + '</span>' +
|
||||
'</label>';
|
||||
}).join('') + '</div>';
|
||||
}
|
||||
|
||||
var container = document.getElementById('model-config-content');
|
||||
container.innerHTML =
|
||||
'<div class="card-header"><span class="card-title">' + formTitle + '</span>' +
|
||||
'<button class="btn btn-sm" onclick="renderRoutingTab()">← 返回</button></div>' +
|
||||
'<div class="card-body"><form onsubmit="event.preventDefault();saveRoutingForm(\'' + escHtml(purpose || '') + '\');">' +
|
||||
'<div class="form-row"><label>用途 ID ' + (isEdit ? '' : '<span style="color:var(--red)">*</span>') + '</label>' +
|
||||
'<select id="routing-purpose" class="input" ' + (isEdit ? 'disabled' : 'required') + '>' +
|
||||
'<option value="">-- 选择用途 --</option>' +
|
||||
'<option value="chat"' + (defaults.purpose === 'chat' ? ' selected' : '') + '>chat (日常对话)</option>' +
|
||||
'<option value="deep_thinking"' + (defaults.purpose === 'deep_thinking' ? ' selected' : '') + '>deep_thinking (深度思考)</option>' +
|
||||
'<option value="intent_analysis"' + (defaults.purpose === 'intent_analysis' ? ' selected' : '') + '>intent_analysis (意图分析)</option>' +
|
||||
'<option value="tool_calling"' + (defaults.purpose === 'tool_calling' ? ' selected' : '') + '>tool_calling (工具调用)</option>' +
|
||||
'<option value="memory_extraction"' + (defaults.purpose === 'memory_extraction' ? ' selected' : '') + '>memory_extraction (记忆提取)</option>' +
|
||||
'</select></div>' +
|
||||
'<div class="form-row"><label>回退模型链 <span style="color:var(--text2);font-weight:400">(勾选即加入,顺序=表格显示顺序)</span></label>' +
|
||||
(models.length > 0 ? '<div class="btn-group" style="margin-bottom:8px">' +
|
||||
'<button type="button" class="btn btn-xs" onclick="var cbs=document.querySelectorAll(\'input[name=routing-model]\');cbs.forEach(function(c){c.checked=true})">全选</button>' +
|
||||
'<button type="button" class="btn btn-xs" onclick="var cbs=document.querySelectorAll(\'input[name=routing-model]\');cbs.forEach(function(c){c.checked=false})">取消全选</button>' +
|
||||
'</div>' : '') +
|
||||
modelCheckboxes + '</div>' +
|
||||
'<div class="form-row"><label style="display:flex;align-items:center;gap:6px;cursor:pointer">' +
|
||||
'<input type="checkbox" id="routing-required"' + (defaults.required ? ' checked' : '') + '> 必需 (所有模型不可用时返回错误,而非回退到 .env)</label></div>' +
|
||||
'<div style="margin-top:14px"><button type="submit" class="btn btn-accent">💾 保存</button></div>' +
|
||||
'</form></div>';
|
||||
}
|
||||
|
||||
async function saveRoutingForm(purpose) {
|
||||
// 收集所有勾选的模型 (按 DOM 顺序 = 表格显示顺序)
|
||||
var checkedCbs = document.querySelectorAll('input[name="routing-model"]:checked');
|
||||
var chain = [];
|
||||
for (var i = 0; i < checkedCbs.length; i++) {
|
||||
chain.push(checkedCbs[i].value);
|
||||
}
|
||||
|
||||
var data = {
|
||||
purpose: purpose || document.getElementById('routing-purpose').value,
|
||||
fallback_chain: chain,
|
||||
required: document.getElementById('routing-required').checked,
|
||||
};
|
||||
if (!data.purpose) { alert('请选择用途'); return; }
|
||||
if (chain.length === 0) { alert('回退模型链不能为空'); return; }
|
||||
var result = await api('/api/model-config/routing/' + encodeURIComponent(data.purpose), { method: 'POST', body: JSON.stringify(data) });
|
||||
if (result.error) { alert('保存失败: ' + result.error); return; }
|
||||
STATE.modelConfigTab = 'routing';
|
||||
renderModelConfigPanel();
|
||||
}
|
||||
|
||||
async function deleteRoutingRule(purpose) {
|
||||
if (!confirm('确定删除路由 "' + purpose + '"?')) return;
|
||||
var result = await api('/api/model-config/routing/' + encodeURIComponent(purpose), { method: 'DELETE' });
|
||||
if (result.error) { alert('删除失败: ' + result.error); return; }
|
||||
renderRoutingTab();
|
||||
}
|
||||
|
||||
// ========== 客户端管理面板 ==========
|
||||
|
||||
function renderClientsPanel() {
|
||||
var panel = document.getElementById('panel-clients');
|
||||
panel.innerHTML =
|
||||
'<div class="card">' +
|
||||
'<div class="card-header">' +
|
||||
'<span class="card-title">📱 已连接设备</span>' +
|
||||
'<div class="btn-group">' +
|
||||
'<button class="btn btn-sm" onclick="loadClients()">🔄 刷新</button>' +
|
||||
'</div>' +
|
||||
'</div>' +
|
||||
'<div id="clients-online"></div>' +
|
||||
'</div>' +
|
||||
'<div class="card">' +
|
||||
'<div class="card-header"><span class="card-title">📋 历史设备</span></div>' +
|
||||
'<div id="clients-offline"></div>' +
|
||||
'</div>' +
|
||||
'<div class="card">' +
|
||||
'<div class="card-header"><span class="card-title">💡 跨端说明</span></div>' +
|
||||
'<div style="font-size:12px;color:var(--text2);line-height:1.8">' +
|
||||
'<p>每个浏览器/设备首次连接时会分配唯一的 <b>Client ID</b>(存储在浏览器 localStorage)。</p>' +
|
||||
'<p>后续所有消息都会携带此 ID,昔涟可据此判断用户当前使用的设备。</p>' +
|
||||
'<p><b>设备名称</b> 自动从 User-Agent 推断,你也可以在下方为设备添加备注。</p>' +
|
||||
'<p>在线设备 = 当前 WebSocket 连接已建立;离线设备 = 曾连接过但当前断开。</p>' +
|
||||
'</div>' +
|
||||
'</div>';
|
||||
loadClients();
|
||||
}
|
||||
|
||||
async function loadClients() {
|
||||
var data = await api('/api/clients');
|
||||
var onlineDiv = document.getElementById('clients-online');
|
||||
var offlineDiv = document.getElementById('clients-offline');
|
||||
if (!onlineDiv || !offlineDiv) return;
|
||||
|
||||
if (data.error) {
|
||||
onlineDiv.innerHTML = '<div class="empty-state"><div class="icon">⚠️</div>' + escHtml(data.error) + '<br><span style="font-size:11px">Gateway 服务可能未启动</span></div>';
|
||||
offlineDiv.innerHTML = '';
|
||||
return;
|
||||
}
|
||||
|
||||
var clients = data.clients || [];
|
||||
var onlineClients = clients.filter(function(c) { return c.online; });
|
||||
var offlineClients = clients.filter(function(c) { return !c.online; });
|
||||
|
||||
// Update badge
|
||||
var badge = document.getElementById('clients-badge');
|
||||
if (badge) {
|
||||
var onlineCount = onlineClients.length;
|
||||
if (onlineCount > 0) {
|
||||
badge.textContent = onlineCount;
|
||||
badge.style.display = 'inline';
|
||||
} else {
|
||||
badge.style.display = 'none';
|
||||
}
|
||||
}
|
||||
|
||||
if (clients.length === 0) {
|
||||
onlineDiv.innerHTML = '<div class="empty-state"><div class="icon">📱</div>暂无已连接的客户端</div>';
|
||||
offlineDiv.innerHTML = '<div class="empty-state"><div class="icon">📋</div>暂无历史设备记录</div>';
|
||||
return;
|
||||
}
|
||||
|
||||
onlineDiv.innerHTML = onlineClients.length > 0
|
||||
? '<div class="table-wrap"><table>' +
|
||||
'<thead><tr><th>状态</th><th>设备名</th><th>Client ID</th><th>备注</th><th>首次连接</th><th>最后活跃</th><th>操作</th></tr></thead>' +
|
||||
'<tbody>' + onlineClients.map(function(c) { return clientRowHTML(c); }).join('') + '</tbody>' +
|
||||
'</table></div>'
|
||||
: '<div class="empty-state"><div class="icon">📱</div>暂无在线客户端</div>';
|
||||
|
||||
offlineDiv.innerHTML = offlineClients.length > 0
|
||||
? '<div class="table-wrap"><table>' +
|
||||
'<thead><tr><th>状态</th><th>设备名</th><th>Client ID</th><th>备注</th><th>首次连接</th><th>最后活跃</th><th>操作</th></tr></thead>' +
|
||||
'<tbody>' + offlineClients.map(function(c) { return clientRowHTML(c); }).join('') + '</tbody>' +
|
||||
'</table></div>'
|
||||
: '<div class="empty-state"><div class="icon">📋</div>暂无历史离线设备</div>';
|
||||
}
|
||||
|
||||
function clientRowHTML(c) {
|
||||
var statusClass = c.online ? 'badge-running' : 'badge-stopped';
|
||||
var statusText = c.online ? '在线' : '离线';
|
||||
return '<tr>' +
|
||||
'<td><span class="badge ' + statusClass + '">' + statusText + '</span></td>' +
|
||||
'<td><strong>' + escHtml(c.device_name || '未知设备') + '</strong></td>' +
|
||||
'<td style="font-family:\'JetBrains Mono\',monospace;font-size:11px;color:var(--text2)">' + escHtml(c.client_id || '') + '</td>' +
|
||||
'<td>' +
|
||||
'<span id="client-note-' + escHtml(c.client_id || '') + '" style="cursor:pointer;color:var(--accent)" title="点击编辑备注" onclick="editClientNote(\'' + escHtml(c.client_id || '') + '\')">' +
|
||||
(c.note ? escHtml(c.note) : '<span style="color:var(--text3);font-style:italic">点击添加备注</span>') +
|
||||
'</span>' +
|
||||
'</td>' +
|
||||
'<td style="font-size:11px;color:var(--text3)">' + formatTime(c.first_seen_at) + '</td>' +
|
||||
'<td style="font-size:11px;color:var(--text3)">' + formatTime(c.last_seen_at) + '</td>' +
|
||||
'<td><button class="btn btn-xs" onclick="editClientNote(\'' + escHtml(c.client_id || '') + '\')">✏️ 备注</button></td>' +
|
||||
'</tr>';
|
||||
}
|
||||
|
||||
async function editClientNote(clientID) {
|
||||
var currentNote = document.getElementById('client-note-' + clientID);
|
||||
var oldNote = (currentNote && currentNote.textContent !== '点击添加备注') ? currentNote.textContent : '';
|
||||
var note = prompt('为设备 ' + clientID + ' 输入备注:', oldNote);
|
||||
if (note === null) return; // cancelled
|
||||
|
||||
var resp = await api('/api/clients/' + encodeURIComponent(clientID) + '/note', {
|
||||
method: 'PUT',
|
||||
body: JSON.stringify({ note: note }),
|
||||
});
|
||||
if (resp.error) {
|
||||
showToast('更新备注失败: ' + resp.error, 'error');
|
||||
} else {
|
||||
showToast('备注已更新', 'success');
|
||||
loadClients();
|
||||
}
|
||||
}
|
||||
|
||||
</script>
|
||||
<script src="iot-panel.js"></script>
|
||||
<script>
|
||||
// ========== 初始化 ==========
|
||||
|
||||
// Listen for browser back/forward navigation.
|
||||
window.addEventListener('hashchange', function() {
|
||||
var hash = location.hash.replace('#', '');
|
||||
var validPanels = ['dashboard', 'memory', 'sessions', 'services', 'iot', 'performance', 'database', 'toolCalls', 'stt', 'thinking', 'timeline', 'chatPlatforms', 'clients', 'modelConfig'];
|
||||
if (hash && validPanels.indexOf(hash) >= 0 && hash !== STATE.activePanel) {
|
||||
switchPanel(hash);
|
||||
}
|
||||
});
|
||||
|
||||
connectWS();
|
||||
refreshStatus();
|
||||
renderDashboard();
|
||||
|
||||
// Restore last panel from URL hash, or default to dashboard.
|
||||
var initHash = location.hash.replace('#', '');
|
||||
var validPanels = ['dashboard', 'memory', 'sessions', 'services', 'iot', 'performance', 'database', 'toolCalls', 'stt', 'thinking', 'timeline', 'chatPlatforms', 'clients', 'modelConfig'];
|
||||
if (initHash && validPanels.indexOf(initHash) >= 0) {
|
||||
switchPanel(initHash);
|
||||
} else {
|
||||
switchPanel('dashboard');
|
||||
location.hash = '#dashboard';
|
||||
}
|
||||
|
||||
// 全局状态定时刷新
|
||||
STATE.statusInterval = setInterval(refreshStatus, 5000);
|
||||
|
||||
@@ -160,7 +160,7 @@ export const SERVICES = {
|
||||
cwd: path.join(ROOT, 'backend/plugin-manager'),
|
||||
command: './main',
|
||||
env: {
|
||||
PLUGIN_MANAGER_PORT: '8094',
|
||||
PORT: '8094',
|
||||
IOT_SERVICE_URL: process.env.IOT_SERVICE_URL || process.env.IOT_DEBUG_SERVICE_URL || 'http://localhost:8083',
|
||||
},
|
||||
healthUrl: 'http://localhost:8094/api/v1/health',
|
||||
|
||||
@@ -21,6 +21,7 @@ import { SERVICES, DEVTOOLS_PORT, LOGS_DIR, logFile, GATEWAY_URL, TOOL_ENGINE_UR
|
||||
|
||||
const MEMORY_SERVICE_URL = process.env.MEMORY_SERVICE_URL || 'http://localhost:8091';
|
||||
const VOICE_SERVICE_URL = process.env.VOICE_SERVICE_URL || 'http://localhost:8093';
|
||||
const PLATFORM_BRIDGE_URL = process.env.PLATFORM_BRIDGE_URL || 'http://localhost:8095';
|
||||
|
||||
const ROOT = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../..');
|
||||
|
||||
@@ -596,6 +597,182 @@ async function proxyToToolEngine(path, opts = {}) {
|
||||
}
|
||||
}
|
||||
|
||||
// ---- 第三方聊天平台配置代理 (转发到 platform-bridge) ----
|
||||
|
||||
/**
|
||||
* 代理请求到 Platform-Bridge
|
||||
* @param {string} path - Platform-Bridge API 路径
|
||||
* @param {object} opts - fetch 选项
|
||||
*/
|
||||
async function proxyToPlatformBridge(path, opts = {}) {
|
||||
const url = `${PLATFORM_BRIDGE_URL}${path}`;
|
||||
const logPrefix = `[PlatformBridge代理]`;
|
||||
try {
|
||||
console.log(`${logPrefix} ${opts.method || 'GET'} ${path}`);
|
||||
const resp = await fetch(url, {
|
||||
...opts,
|
||||
headers: { 'Content-Type': 'application/json', ...opts.headers },
|
||||
signal: AbortSignal.timeout(10000),
|
||||
});
|
||||
const body = await resp.json().catch(() => null);
|
||||
if (!resp.ok) {
|
||||
console.log(`${logPrefix} 请求失败 (HTTP ${resp.status}): ${path}`);
|
||||
}
|
||||
return { status: resp.status, body };
|
||||
} catch (err) {
|
||||
const isConnRefused = err.message?.includes('ECONNREFUSED') || err.cause?.code === 'ECONNREFUSED';
|
||||
console.error(`${logPrefix} 请求异常: ${path} - ${err.message}`);
|
||||
return {
|
||||
status: 502,
|
||||
body: {
|
||||
error: `Platform-Bridge 不可达: ${err.message}`,
|
||||
errorType: isConnRefused ? 'bridge_not_running' : 'bridge_unreachable',
|
||||
hint: isConnRefused
|
||||
? 'Platform-Bridge 服务未启动,请先在「服务管理」面板中启动该服务'
|
||||
: 'Platform-Bridge 服务无响应,请检查网络连接和服务状态',
|
||||
},
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// GET /api/chat-platforms/configs — 列出所有平台配置
|
||||
app.get('/api/chat-platforms/configs', async (_req, res) => {
|
||||
const result = await proxyToPlatformBridge('/api/v1/configs');
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// GET /api/chat-platforms/configs/:name — 获取单个配置
|
||||
app.get('/api/chat-platforms/configs/:name', async (req, res) => {
|
||||
const result = await proxyToPlatformBridge(`/api/v1/configs/${req.params.name}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// POST /api/chat-platforms/configs/:name — 创建或更新配置
|
||||
app.post('/api/chat-platforms/configs/:name', async (req, res) => {
|
||||
const result = await proxyToPlatformBridge(`/api/v1/configs/${req.params.name}`, {
|
||||
method: 'POST',
|
||||
body: JSON.stringify(req.body),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// DELETE /api/chat-platforms/configs/:name — 删除配置
|
||||
app.delete('/api/chat-platforms/configs/:name', async (req, res) => {
|
||||
const result = await proxyToPlatformBridge(`/api/v1/configs/${req.params.name}`, {
|
||||
method: 'DELETE',
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// GET /api/chat-platforms/logs/:name — 获取消息日志
|
||||
app.get('/api/chat-platforms/logs/:name', async (req, res) => {
|
||||
const limit = req.query.limit || '100';
|
||||
const result = await proxyToPlatformBridge(`/api/v1/logs/${req.params.name}?limit=${limit}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// ---- 多端客户端管理代理 (转发到 Gateway) ----
|
||||
|
||||
// GET /api/clients — 获取已知客户端列表
|
||||
app.get('/api/clients', async (req, res) => {
|
||||
const userID = req.query.user_id || 'admin';
|
||||
const result = await proxyToGateway(`/api/v1/admin/clients?user_id=${encodeURIComponent(userID)}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// PUT /api/clients/:id/note — 更新客户端备注
|
||||
app.put('/api/clients/:id/note', async (req, res) => {
|
||||
const { note } = req.body;
|
||||
const result = await proxyToGateway(`/api/v1/admin/clients/${req.params.id}/note`, {
|
||||
method: 'PUT',
|
||||
body: JSON.stringify({ note }),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// ---- 模型配置管理代理 (转发到 Gateway admin) ----
|
||||
|
||||
// Providers
|
||||
app.get('/api/model-config/providers', async (_req, res) => {
|
||||
const result = await proxyToGateway('/api/v1/admin/models/providers');
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.get('/api/model-config/providers/:name', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/providers/${req.params.name}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.post('/api/model-config/providers/:name', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/providers/${req.params.name}`, {
|
||||
method: 'POST', body: JSON.stringify(req.body),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.delete('/api/model-config/providers/:name', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/providers/${req.params.name}`, {
|
||||
method: 'DELETE',
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// Models
|
||||
app.get('/api/model-config/models', async (_req, res) => {
|
||||
const result = await proxyToGateway('/api/v1/admin/models/models');
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.get('/api/model-config/models/:id', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/models/${req.params.id}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.post('/api/model-config/models/:id', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/models/${req.params.id}`, {
|
||||
method: 'POST', body: JSON.stringify(req.body),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.delete('/api/model-config/models/:id', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/models/${req.params.id}`, {
|
||||
method: 'DELETE',
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// Routing
|
||||
app.get('/api/model-config/routing', async (_req, res) => {
|
||||
const result = await proxyToGateway('/api/v1/admin/models/routing');
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.get('/api/model-config/routing/:purpose', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/routing/${req.params.purpose}`);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.post('/api/model-config/routing/:purpose', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/routing/${req.params.purpose}`, {
|
||||
method: 'POST', body: JSON.stringify(req.body),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
app.delete('/api/model-config/routing/:purpose', async (req, res) => {
|
||||
const result = await proxyToGateway(`/api/v1/admin/models/routing/${req.params.purpose}`, {
|
||||
method: 'DELETE',
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// Health check
|
||||
app.post('/api/model-config/health-check', async (req, res) => {
|
||||
const result = await proxyToGateway('/api/v1/admin/models/health-check', {
|
||||
method: 'POST', body: JSON.stringify(req.body),
|
||||
});
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// GET /api/model-config/fetch-models/:name?url=... — 代理查询 Provider 模型列表
|
||||
app.get('/api/model-config/fetch-models/:name', async (req, res) => {
|
||||
const urlParam = req.query.url ? '?url=' + encodeURIComponent(req.query.url) : '';
|
||||
const result = await proxyToGateway('/api/v1/admin/models/fetch-models/' + encodeURIComponent(req.params.name) + urlParam);
|
||||
res.status(result.status).json(result.body);
|
||||
});
|
||||
|
||||
// GET /api/tool-calls — 查询工具调用记录
|
||||
app.get('/api/tool-calls', async (req, res) => {
|
||||
const { tool_name, page, limit } = req.query;
|
||||
|
||||
@@ -1,11 +1,13 @@
|
||||
import { useState, useEffect, useRef, useCallback } from 'react';
|
||||
import { CyreneAvatar } from '@/components/persona/CyreneAvatar';
|
||||
import { useAuthStore } from '@/store/authStore';
|
||||
import { useChatStore } from '@/store/chatStore';
|
||||
import { useSpeechSynthesis } from '@/hooks/useSpeechSynthesis';
|
||||
import type { MessageAttachment, MultiMessageItem, StreamSegment, MessageDisplayType } from '@/types/chat';
|
||||
import { ImageLightbox } from './ImageLightbox';
|
||||
|
||||
interface MessageBubbleProps {
|
||||
id: string;
|
||||
role: 'user' | 'assistant' | 'system' | 'action';
|
||||
content: string;
|
||||
timestamp: number;
|
||||
@@ -141,6 +143,7 @@ function AIMessageActions({ content }: { content: string }) {
|
||||
}
|
||||
|
||||
export function MessageBubble({
|
||||
id,
|
||||
role,
|
||||
content,
|
||||
timestamp,
|
||||
@@ -203,6 +206,19 @@ export function MessageBubble({
|
||||
// 判断是否还有未显示完的字符
|
||||
const hasMoreChars = isStreaming && displayedContent.length < content.length;
|
||||
|
||||
// When typewriter animation finishes, notify store to process next queued message.
|
||||
const typewriterComplete = isStreaming && content.length > 0 && displayedContent.length >= content.length;
|
||||
const doneNotifiedRef = useRef(false);
|
||||
useEffect(() => {
|
||||
if (typewriterComplete && !doneNotifiedRef.current) {
|
||||
doneNotifiedRef.current = true;
|
||||
const timer = setTimeout(() => {
|
||||
useChatStore.getState().onTypewriterDone(id);
|
||||
}, 200);
|
||||
return () => clearTimeout(timer);
|
||||
}
|
||||
}, [typewriterComplete, id]);
|
||||
|
||||
// 图片附件
|
||||
const imageAttachments = attachments?.filter((a) => a.type === 'image') ?? [];
|
||||
|
||||
|
||||
@@ -66,6 +66,7 @@ export function MessageList({
|
||||
{messages.map((msg) => (
|
||||
<MessageBubble
|
||||
key={msg.id}
|
||||
id={msg.id}
|
||||
role={msg.role}
|
||||
content={msg.content}
|
||||
timestamp={msg.timestamp}
|
||||
|
||||
@@ -249,7 +249,8 @@ function handleServerMessage(msg: WSServerMessage) {
|
||||
case 'history_response':
|
||||
// 防御性检查:仅当当前消息为空时才加载 WebSocket 历史响应
|
||||
// 避免 WebSocket 的 history_response (可能来自后端空缓存) 覆盖 HTTP loadMessagesFromServer 已加载的消息
|
||||
if (msg.messages) {
|
||||
// 注意:空数组 [] 在 JS 中是 truthy,必须显式检查 length > 0
|
||||
if (msg.messages && msg.messages.length > 0) {
|
||||
const sessionState = useSessionStore.getState();
|
||||
// 如果 sessionStore 或 chatStore 中已有消息,说明 HTTP 已加载完成,忽略 WS 的历史响应
|
||||
if (sessionState.messages.length > 0 || chatState.messages.length > 0) {
|
||||
@@ -263,9 +264,13 @@ function handleServerMessage(msg: WSServerMessage) {
|
||||
const msgsWithIds: Message[] = msg.messages.map((m, i) => ({
|
||||
...m,
|
||||
id: m.id || `hist_${i}_${Date.now()}`,
|
||||
// 规范化 msg_type → msgType 以保持与 HTTP 加载路径一致
|
||||
msgType: (m as any).msg_type || m.msgType,
|
||||
}));
|
||||
setMessages(msgsWithIds);
|
||||
useChatStore.getState().setMessages(msgsWithIds);
|
||||
} else if (msg.messages && msg.messages.length === 0) {
|
||||
console.log('[WS] 忽略空的 history_response(后端缓存和数据库均无消息),等待 HTTP 加载');
|
||||
}
|
||||
setTyping(false);
|
||||
break;
|
||||
|
||||
@@ -21,9 +21,16 @@ interface ChatStore {
|
||||
isLoadingHistory: boolean;
|
||||
historyPage: number;
|
||||
|
||||
// 多气泡消息队列:确保气泡依次出现 + 逐字动画
|
||||
messageQueue: Message[];
|
||||
|
||||
addMessage: (message: Message) => void;
|
||||
appendToLastMessage: (content: string) => void;
|
||||
finishStreaming: () => void;
|
||||
/** 将一个消息加入队列,等待上一个气泡的逐字动画完成后再展示 */
|
||||
enqueueMessage: (message: Message) => void;
|
||||
/** 当前气泡逐字动画完成后调用:关闭 isStreaming,出队下一个 */
|
||||
onTypewriterDone: (messageId: string) => void;
|
||||
setMessages: (messages: Message[]) => void;
|
||||
setTyping: (typing: boolean) => void;
|
||||
clearMessages: () => void;
|
||||
@@ -49,6 +56,7 @@ export const useChatStore = create<ChatStore>((set) => ({
|
||||
hasMoreMessages: false,
|
||||
isLoadingHistory: false,
|
||||
historyPage: 1,
|
||||
messageQueue: [],
|
||||
|
||||
addMessage: (message) =>
|
||||
set((state) => ({
|
||||
@@ -84,14 +92,48 @@ export const useChatStore = create<ChatStore>((set) => ({
|
||||
isStreaming: false,
|
||||
};
|
||||
}
|
||||
return { messages: msgs, isTyping: false };
|
||||
// Process queued messages after streaming finishes
|
||||
const queue = [...state.messageQueue];
|
||||
if (queue.length > 0) {
|
||||
const next = queue.shift()!;
|
||||
return {
|
||||
messages: [...msgs, { ...next, isStreaming: true }],
|
||||
messageQueue: queue,
|
||||
isTyping: false,
|
||||
};
|
||||
}
|
||||
return { messages: msgs, isTyping: false, messageQueue: queue };
|
||||
}),
|
||||
|
||||
enqueueMessage: (message) =>
|
||||
set((state) => ({
|
||||
messageQueue: [...state.messageQueue, message],
|
||||
})),
|
||||
|
||||
onTypewriterDone: (messageId) =>
|
||||
set((state) => {
|
||||
// Mark this specific message as no longer streaming
|
||||
const msgs = state.messages.map((m) =>
|
||||
m.id === messageId ? { ...m, isStreaming: false } : m
|
||||
);
|
||||
// Process next queued message
|
||||
const queue = [...state.messageQueue];
|
||||
if (queue.length > 0) {
|
||||
const next = queue.shift()!;
|
||||
return {
|
||||
messages: [...msgs, { ...next, isStreaming: true }],
|
||||
messageQueue: queue,
|
||||
isTyping: false,
|
||||
};
|
||||
}
|
||||
return { messages: msgs, isTyping: false, messageQueue: queue };
|
||||
}),
|
||||
|
||||
setMessages: (messages) => set({ messages, isTyping: false }),
|
||||
|
||||
setTyping: (typing) => set({ isTyping: typing }),
|
||||
|
||||
clearMessages: () => set({ messages: [], isTyping: false, hasMoreMessages: false, historyPage: 1 }),
|
||||
clearMessages: () => set({ messages: [], isTyping: false, hasMoreMessages: false, historyPage: 1, messageQueue: [] }),
|
||||
|
||||
setContinuousMode: (enabled) => set({ continuousMode: enabled }),
|
||||
|
||||
|
||||
@@ -38,6 +38,13 @@ export interface ReviewMessage {
|
||||
content: string;
|
||||
}
|
||||
|
||||
/** 客户端信息 (多端区分) */
|
||||
export interface ClientInfo {
|
||||
client_id: string;
|
||||
device_name?: string;
|
||||
user_agent?: string;
|
||||
}
|
||||
|
||||
/** 单条消息 */
|
||||
export interface Message {
|
||||
id: string;
|
||||
@@ -50,6 +57,8 @@ export interface Message {
|
||||
isStreaming?: boolean;
|
||||
/** 消息显示类型: 区分聊天消息与动作消息 */
|
||||
msgType?: MessageDisplayType;
|
||||
/** 消息来源客户端信息 (多端区分) */
|
||||
client_info?: ClientInfo;
|
||||
}
|
||||
|
||||
/** IoT 设备类型定义 */
|
||||
@@ -99,6 +108,9 @@ export interface WSClientMessage {
|
||||
audio_data?: string; // base64
|
||||
attachments?: MessageAttachment[];
|
||||
timestamp: number;
|
||||
client_id?: string;
|
||||
device_name?: string;
|
||||
user_agent?: string;
|
||||
}
|
||||
|
||||
/** 通知类型 */
|
||||
@@ -146,6 +158,7 @@ export interface WSServerMessage {
|
||||
system_info?: SystemInfoPayload;
|
||||
protocol_version?: number;
|
||||
timestamp: number;
|
||||
client_info?: ClientInfo;
|
||||
}
|
||||
|
||||
/** 工具进度信息 */
|
||||
|
||||
Reference in New Issue
Block a user