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:
2026-05-23 21:23:10 +08:00
parent 965cce7192
commit 0717928496
29 changed files with 3177 additions and 137 deletions
+2
View File
@@ -40,6 +40,8 @@ data/
# ========== 环境与敏感配置 ==========
.env
backend/.env
models.json
platform_configs.json
.claude/
# ========== 文档 (项目规范:docs/ 不纳入版本管理) ==========
+21 -5
View File
@@ -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
+119
View File
@@ -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
}
+131
View File
@@ -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
}
+12 -1
View File
@@ -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"`
}
+255 -87
View File
@@ -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{}
}
+25 -1
View File
@@ -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
}
+17 -11
View File
@@ -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,
}
}
+138
View File
@@ -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,
}
}
+11
View File
@@ -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 工具执行进度
+140 -18
View File
@@ -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
View File
@@ -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(&quot;model-search-input&quot;);if(s)s.value=&quot;&quot;;renderFetchedModelList(STATE.fetchedModels,&quot;&quot;);" 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(&quot;data-model&quot;);selectFetchedModel(mn);"' +
' style="padding:6px 12px;cursor:pointer;font-size:12px;border-bottom:1px solid var(--border);transition:background .12s"' +
' onmouseenter="this.style.background=&quot;var(--bg3)&quot;"' +
' onmouseleave="this.style.background=&quot;&quot;">' + 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);
+1 -1
View File
@@ -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',
+177
View File
@@ -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}
+6 -1
View File
@@ -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;
+44 -2
View File
@@ -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 }),
+13
View File
@@ -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;
}
/** 工具进度信息 */