freedakgmail
c949204662
借鉴 odysseus 的能力设计，全程净室实现、零 AGPL 代码、不引入 AGPL 依赖。
T1 提示注入防护: pkg/promptguard 包裹外部/知识库内容为不可信数据，buildMessages 移出 system 指令区。 T2 安全 CI: .github/workflows(ci+security: govulncheck/gitleaks/actionlint/hadolint/trivy)+dependabot+.hadolint.yaml；go.mod 加 toolchain go1.25.11 修复 20 个 stdlib CVE。 T3 管理员 2FA: 迁移 000016 + RFC6238 TOTP/备份码(pkg/auth, 零依赖) + 登录流程集成(后端)。 T4 本地模型: LLM/embedding 支持本地 vLLM/Ollama(OpenAI 兼容, 鉴权头条件发送, NoAuth) + docs/local-deploy.md。 T6 深度研究: 迁移 000017 + Python research-worker(净室多步流水线, 检索避开 SearXNG) + Go research 服务/handler/路由。 T7 service 层: 新增 internal/service/{research,twofa}, 2FA 业务逻辑从胖 handler 下沉, 接口注入可单测。 T10 缓存/可观测性: internal/cache(Redis+内存, 优雅降级) 接入 store 热点列表; Prometheus 指标+/metrics; docs/openapi.yaml。 验证: go build/vet/test ./... 全绿(8 包); research-worker 12 单测过; 真实 PG 应用迁移并烟测。
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package research
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strconv"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
const (
|
|
taskQueueKey = "research:tasks"
|
|
statusKeyPrefix = "research:status:"
|
|
)
|
|
|
|
// redisBackend 同时实现 Queue 与 Cache：任务下发到列表队列、快速状态读自 hash。
|
|
// 与 research-worker 的 TASK_QUEUE / TASK_STATUS_PREFIX 约定保持一致。
|
|
type redisBackend struct {
|
|
rdb *redis.Client
|
|
}
|
|
|
|
func NewRedisBackend(rdb *redis.Client) *redisBackend {
|
|
return &redisBackend{rdb: rdb}
|
|
}
|
|
|
|
func (b *redisBackend) Enqueue(ctx context.Context, taskID string) error {
|
|
msg, _ := json.Marshal(map[string]string{"task_id": taskID})
|
|
return b.rdb.LPush(ctx, taskQueueKey, msg).Err()
|
|
}
|
|
|
|
func (b *redisBackend) GetStatus(ctx context.Context, taskID string) (*CachedStatus, bool) {
|
|
m, err := b.rdb.HGetAll(ctx, statusKeyPrefix+taskID).Result()
|
|
if err != nil || len(m) == 0 {
|
|
return nil, false
|
|
}
|
|
progress, _ := strconv.Atoi(m["progress"])
|
|
return &CachedStatus{
|
|
Status: m["status"],
|
|
Progress: progress,
|
|
Message: m["message"],
|
|
}, true
|
|
}
|