Files
HomeAgent/pkg/types/types.go
root 239a22899b fix: 模型思考模式配置 + Unicode 截断 + 审计修复 (13 files)
模型模式:
- 新增 LLMConfig/Source.ThinkingEnabled 配置,通过 ExtraBody
  控制 DeepSeek thinking mode,默认关闭
- SeedDefaults/ToConfig 读写 core.llm.thinking_enabled
- deepseek.lua 移除硬编码 temperature=0

Unicode 截断:
- truncateStr 改按 rune 计数,修复中文截断乱码

审计修复 (Critical):
- graph.go: defer rows.Close 在 for 循环 → 显式 Close (连接池泄漏)
- cli/openclaw/plugin.go: bare type assertion → comma-ok (panic)
- channel.go: payload["type"].(string) → comma-ok (panic)
- webui/handler.go: .(string) → fmt.Sprint (panic)
- agent.go: 添加 nil provider 错误返回

审计修复 (High):
- events/bus.go: copy handler slice under RLock (data race)
- webui/handler.go: SSE 通过 channel 串行化写入 (data race)
- timer/plugin.go: time.Sleep → select with stopCh (Stop 阻塞)
- provider.go: stream ch <- 添加 select ctx.Done (goroutine 泄漏)
- main.go: outputCh goroutine 添加 ctx.Done 退出路径
2026-07-03 20:46:04 +08:00

132 lines
4.0 KiB
Go

package types
import "time"
type AgentState int
const (
AgentStateStopped AgentState = 0
AgentStateRunning AgentState = 1
AgentStateDegraded AgentState = 2
AgentStatePanic AgentState = 3
)
type HealthStatus int
const (
HealthUnknown HealthStatus = 0
HealthHealthy HealthStatus = 1
HealthUnstable HealthStatus = 2
HealthDown HealthStatus = 3
HealthDegraded HealthStatus = 4
)
type AgentID string
type SnapshotID string
type Snapshot struct {
ID SnapshotID `json:"id"`
AgentID AgentID `json:"agent_id"`
CreatedAt time.Time `json:"created_at"`
Reason string `json:"reason"`
Size int64 `json:"size_bytes"`
DockerImage string `json:"docker_image,omitempty"`
Valid bool `json:"valid"`
}
type Heartbeat struct {
AgentID AgentID `json:"agent_id"`
Timestamp time.Time `json:"timestamp"`
State AgentState `json:"state"`
Health HealthStatus `json:"health"`
Uptime time.Duration `json:"uptime"`
LLMConnected bool `json:"llm_connected"`
Error string `json:"error,omitempty"`
}
type NetworkCheckResult struct {
LLMAPIReachable bool `json:"llm_api_reachable"`
DNSResolving bool `json:"dns_resolving"`
Latency time.Duration `json:"latency_ms"`
Error string `json:"error,omitempty"`
}
type SnapshotPolicy struct {
Interval time.Duration `json:"interval"`
MaxSnapshots int `json:"max_snapshots"`
PreAction bool `json:"pre_action"`
PostAction bool `json:"post_action"`
}
type RollbackPolicy struct {
MaxRetries int `json:"max_retries"`
HealthThreshold HealthStatus `json:"health_threshold"`
CooldownPeriod time.Duration `json:"cooldown_period"`
AutoRollback bool `json:"auto_rollback"`
}
type AgentConfig struct {
ID AgentID `json:"id"`
Image string `json:"image"`
Name string `json:"name"`
LLMEndpoints []string `json:"llm_endpoints"`
SnapshotPolicy SnapshotPolicy `json:"snapshot_policy"`
RollbackPolicy RollbackPolicy `json:"rollback_policy"`
ResourceLimit ResourceLimit `json:"resource_limit"`
OpenClawEnabled bool `json:"openclaw_enabled"`
}
type ResourceLimit struct {
CPU string `json:"cpu"`
Memory string `json:"memory"`
Disk string `json:"disk"`
Network bool `json:"network"`
}
type OperationLog struct {
ID string `json:"id"`
AgentID AgentID `json:"agent_id"`
Timestamp time.Time `json:"timestamp"`
Action string `json:"action"`
SnapshotID SnapshotID `json:"snapshot_id,omitempty"`
Success bool `json:"success"`
}
type LLMSource struct {
Name string `json:"name"`
BaseURL string `json:"base_url"`
Model string `json:"model"`
APIKey string `json:"api_key,omitempty"`
Adapter string `json:"adapter"`
AdapterPath string `json:"adapter_path,omitempty"`
ThinkingEnabled bool `json:"thinking_enabled,omitempty"`
}
type LLMConfig struct {
Provider string `json:"provider"`
Model string `json:"model"`
BaseURL string `json:"base_url"`
APIKey string `json:"api_key"`
Adapter string `json:"adapter"`
Temperature float64 `json:"temperature"`
MaxTokens int `json:"max_tokens"`
ThinkingEnabled bool `json:"thinking_enabled"`
Sources []LLMSource `json:"sources,omitempty"`
}
type Config struct {
Daemon DaemonConfig `json:"daemon"`
LLM LLMConfig `json:"llm"`
Defaults AgentConfig `json:"defaults"`
Agents []AgentConfig `json:"agents"`
}
type DaemonConfig struct {
ListenAddr string `json:"listen_addr"`
DataDir string `json:"data_dir"`
HeartbeatInterval time.Duration `json:"heartbeat_interval"`
CheckInterval time.Duration `json:"check_interval"`
LogLevel string `json:"log_level"`
}