fix: 修复 AI 回复无法送达发送者 + 重复消息 + action角色泄露 + OS环境支持
广播逻辑重构: - AI 回复 (stream_start/response/stream_segments/multi_message/stream_end) 改用 broadcastToUser 发送给所有客户端 - 用户消息回显保持 broadcastToUserExcept 排除发送者 消息去重与角色修复: - CacheMessage(user) 移至回复生成后,避免本轮 LLM 调用出现重复用户消息 - action 角色消息在 DB 存储时映射为 assistant,DeepSeek 等模型不支持自定义角色 - stream_end defer 机制确保错误路径也会终止客户端思考指示器 OS 完整环境支持: - host 包重构为 HostBackend 接口 + Direct/WSL/Docker 三种后端 - 新增 os_exec/os_file/os_system 工具供 AI 在完整 Linux 环境中自由操作 其他: - 视觉模型注入 + 图片预处理后清空 Images 避免传给 Chat 模型 - 图片 URL 相对路径→绝对 URL 转换 - DevTools 链路追踪页面 + 重启修复 - 记忆搜索模糊匹配增强 - 后台思考定时调度支持 - 管理后台页面 (模型配置/用户管理等) - docs/api 更新广播机制说明 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,151 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// ScheduleRule defines a time-based interval rule.
|
||||
type ScheduleRule struct {
|
||||
Name string `json:"name"`
|
||||
Days []string `json:"days"` // monday, tuesday, wednesday, thursday, friday, saturday, sunday
|
||||
TimeRange string `json:"time_range"` // "HH:MM-HH:MM"
|
||||
Except []string `json:"except"` // ["HH:MM-HH:MM", ...]
|
||||
IntervalMinutes int `json:"interval_minutes"`
|
||||
}
|
||||
|
||||
// ThinkingScheduleConfig is the full schedule configuration.
|
||||
type ThinkingScheduleConfig struct {
|
||||
Version string `json:"version"`
|
||||
DefaultIntervalMinutes int `json:"default_interval_minutes"`
|
||||
Rules []ScheduleRule `json:"rules"`
|
||||
}
|
||||
|
||||
// DefaultThinkingScheduleConfig returns the default schedule with two rules.
|
||||
func DefaultThinkingScheduleConfig() *ThinkingScheduleConfig {
|
||||
return &ThinkingScheduleConfig{
|
||||
Version: "1.0",
|
||||
DefaultIntervalMinutes: 5,
|
||||
Rules: []ScheduleRule{
|
||||
{
|
||||
Name: "night",
|
||||
Days: []string{"monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"},
|
||||
TimeRange: "23:00-07:00",
|
||||
IntervalMinutes: 30,
|
||||
},
|
||||
{
|
||||
Name: "weekday_work",
|
||||
Days: []string{"monday", "tuesday", "wednesday", "thursday", "friday"},
|
||||
TimeRange: "09:00-17:00",
|
||||
Except: []string{"12:00-14:00", "15:00-15:30"},
|
||||
IntervalMinutes: 30,
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// ThinkingScheduleStore persists the schedule config to a JSON file.
|
||||
type ThinkingScheduleStore struct {
|
||||
mu sync.RWMutex
|
||||
path string
|
||||
config *ThinkingScheduleConfig
|
||||
}
|
||||
|
||||
// NewThinkingScheduleStore creates a store, creating the file with defaults if it does not exist.
|
||||
func NewThinkingScheduleStore(path string) (*ThinkingScheduleStore, error) {
|
||||
s := &ThinkingScheduleStore{
|
||||
path: path,
|
||||
config: nil,
|
||||
}
|
||||
if err := s.load(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (s *ThinkingScheduleStore) load() error {
|
||||
data, err := os.ReadFile(s.path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
s.config = DefaultThinkingScheduleConfig()
|
||||
return s.save()
|
||||
}
|
||||
return fmt.Errorf("read thinking schedule file: %w", err)
|
||||
}
|
||||
if len(data) == 0 {
|
||||
s.config = DefaultThinkingScheduleConfig()
|
||||
return s.save()
|
||||
}
|
||||
var cfg ThinkingScheduleConfig
|
||||
if err := json.Unmarshal(data, &cfg); err != nil {
|
||||
return fmt.Errorf("parse thinking schedule: %w", err)
|
||||
}
|
||||
if cfg.Version == "" {
|
||||
cfg.Version = "1.0"
|
||||
}
|
||||
if cfg.DefaultIntervalMinutes <= 0 {
|
||||
cfg.DefaultIntervalMinutes = 5
|
||||
}
|
||||
if cfg.Rules == nil {
|
||||
cfg.Rules = []ScheduleRule{}
|
||||
}
|
||||
s.config = &cfg
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *ThinkingScheduleStore) save() error {
|
||||
data, err := json.MarshalIndent(s.config, "", " ")
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal thinking schedule: %w", err)
|
||||
}
|
||||
tmpPath := s.path + ".tmp"
|
||||
if err := os.WriteFile(tmpPath, data, 0640); err != nil {
|
||||
return fmt.Errorf("write thinking schedule: %w", err)
|
||||
}
|
||||
return os.Rename(tmpPath, s.path)
|
||||
}
|
||||
|
||||
// GetConfig returns the current config (read-only).
|
||||
func (s *ThinkingScheduleStore) GetConfig() *ThinkingScheduleConfig {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return s.config
|
||||
}
|
||||
|
||||
// SetConfig validates and persists a new config.
|
||||
func (s *ThinkingScheduleStore) SetConfig(cfg *ThinkingScheduleConfig) error {
|
||||
if cfg == nil {
|
||||
return fmt.Errorf("配置不能为空")
|
||||
}
|
||||
if cfg.DefaultIntervalMinutes <= 0 {
|
||||
cfg.DefaultIntervalMinutes = 5
|
||||
}
|
||||
if cfg.Version == "" {
|
||||
cfg.Version = "1.0"
|
||||
}
|
||||
if cfg.Rules == nil {
|
||||
cfg.Rules = []ScheduleRule{}
|
||||
}
|
||||
for _, r := range cfg.Rules {
|
||||
if r.IntervalMinutes <= 0 {
|
||||
return fmt.Errorf("规则 %q 间隔分钟必须大于 0", r.Name)
|
||||
}
|
||||
if r.TimeRange == "" {
|
||||
return fmt.Errorf("规则 %q 缺少 time_range", r.Name)
|
||||
}
|
||||
}
|
||||
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.config = cfg
|
||||
return s.save()
|
||||
}
|
||||
|
||||
// HasConfig returns true if a config is loaded.
|
||||
func (s *ThinkingScheduleStore) HasConfig() bool {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return s.config != nil
|
||||
}
|
||||
Reference in New Issue
Block a user