AskaEth
0624d9ee36
按 DevDocs/design/22 批次 1 后端（§2/§3/§4.1-4.5/§4.7/§5 验收 1-6）。
- 迁移 0011：chat_sessions.persona_id + chat_messages.persona_id（均 DEFAULT ''，无回填 → 升级零风险）；同步包内幂等 DDL；**真实库升级验证**（拷贝 data/cyrene.db 到临时目录跑 MigrateUp：既有 1 会话/109 消息 persona 仍为 ''，原库未被触碰）
- history：SessionMeta/Message 加 PersonaID + SetSessionPersona；ListSessions/List/scan/Append 带上
- runtime：SessionEnv.PersonaID + loop.personaFor(ctx) 替换固定 l.personaID（memory/request/turn）+ **agent 池**（Options.Agents，按角色取 instructions，未命中回落）+ **提取器不再把 persona 烤死在构造期**（多角色下记忆不再永远记在 cyrene 名下）+ **thinking 旋钮改为按会话角色解析**（否则角色 Y 会用角色 X 的预算/展示）
- gateway：会话 DTO/POST/PATCH 支持 personaId；ws/rest/voice 的 OwnerEnv 硬编改为**按会话解析角色**（actor 元数据语义不变）；admin memories 搜索去掉硬编 cyrene
- Docs/api-client v0.42；**两处有意的行为收紧**已写明：POST /sessions body 非法 JSON → 400（原先完全忽略 body；空/null/{} 仍 201）；POST /admin/memories/search 省略 personaId 且无默认角色 → 400（原先默认查 cyrene）
- 测试：design/22 §5 验收 1-6（记忆隔离/提示词首条 system/情绪按角色/提取 persona 归属/迁移兼容）+ 进程内 e2e（断言 adapter 收到的 model.Request.Messages[0] 即该角色的 systemPrompt）
- **已知未做（留批次 2 / 待确认）**：① IM 与渠道级未做 → **§7-3「owner 私聊镜像随主会话角色」尚未生效**；② 请求级覆盖未实现（本批只做会话级）；③ **§7-5 的「上下文发送角色标记」只完成存储+DTO 半边**——消息 persona 已落库且 DTO 已带，但模型上下文不从 chat_messages 重建，「第 N 条是谁说的」尚未进上下文（需确认是否属批次 2）；④「切角色重建摘要」当前是空操作（压缩未接线）；⑤ thinking 旋钮仅在流式路径生效（非流式 agent.RunStep 不带，A 批次既有性质）
- 验收：go build/vet/test 全绿 + linux/amd64 与 darwin/arm64 交叉编译（既有 flake 本轮未触发；同批 session_orchestrator.go 一度 501 行 → 抽出 session_tail.go）
83 lines
3.5 KiB
Go
83 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/memory"
|
|
modelpkg "git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
|
|
)
|
|
|
|
// testRuntimeLLM builds a provider stack from env alone (the pointed-at
|
|
// models.json does not exist), so a test controls the active group through the
|
|
// flat LLM config exactly as a keyless .env boot would.
|
|
func testRuntimeLLM(t *testing.T, env config.LLMConfig) *modelpkg.RuntimeLLM {
|
|
t.Helper()
|
|
return modelpkg.NewRuntimeLLM(filepath.Join(t.TempDir(), "models.json"), env, testLog())
|
|
}
|
|
|
|
// TestBuildMemoryExtractor_LLMWhenCredentials proves a configured provider group
|
|
// selects the LLM extractor (design/03 §1). The extractor is role-agnostic
|
|
// (design/22 §4.2): it is built without a persona and the loop stamps the turn's
|
|
// role onto each extracted item, so multi-role extraction cannot be filed under
|
|
// the default role.
|
|
func TestBuildMemoryExtractor_LLMWhenCredentials(t *testing.T) {
|
|
t.Parallel()
|
|
a := &app{
|
|
log: testLog(),
|
|
llm: testRuntimeLLM(t, config.LLMConfig{
|
|
BaseURL: "https://api.example.com/v1",
|
|
APIKey: "sk-test",
|
|
Model: "gpt-4o-mini",
|
|
}),
|
|
}
|
|
if _, ok := a.buildMemoryExtractor().(*memory.LLMExtractor); !ok {
|
|
t.Fatalf("extractor = %T, want *memory.LLMExtractor (credentials configured)", a.buildMemoryExtractor())
|
|
}
|
|
}
|
|
|
|
// TestBuildMemoryExtractor_ManualWhenNoCredentials proves the keyless fallback: a
|
|
// boot without an API key (and without a local endpoint) keeps the deterministic
|
|
// ManualExtractor instead of firing a doomed model call on every turn.
|
|
func TestBuildMemoryExtractor_ManualWhenNoCredentials(t *testing.T) {
|
|
t.Parallel()
|
|
a := &app{log: testLog(), llm: testRuntimeLLM(t, config.LLMConfig{})}
|
|
if _, ok := a.buildMemoryExtractor().(*memory.ManualExtractor); !ok {
|
|
t.Fatalf("extractor = %T, want *memory.ManualExtractor (no usable llm group)", a.buildMemoryExtractor())
|
|
}
|
|
|
|
bare := &app{log: testLog()}
|
|
if _, ok := bare.buildMemoryExtractor().(*memory.ManualExtractor); !ok {
|
|
t.Fatalf("extractor = %T, want *memory.ManualExtractor (nil provider stack)", bare.buildMemoryExtractor())
|
|
}
|
|
}
|
|
|
|
// TestLLMGroupUsable_KeylessLocalEndpoint locks the ollama-style case: a local
|
|
// OpenAI-compatible server needs no API key, so extraction stays LLM-backed.
|
|
func TestLLMGroupUsable_KeylessLocalEndpoint(t *testing.T) {
|
|
t.Parallel()
|
|
cases := []struct {
|
|
name string
|
|
env config.LLMConfig
|
|
ready bool
|
|
}{
|
|
{name: "api key", env: config.LLMConfig{BaseURL: "https://api.example.com/v1", APIKey: "sk", Model: "m"}, ready: true},
|
|
{name: "loopback keyless", env: config.LLMConfig{BaseURL: "http://127.0.0.1:11434/v1", Model: "llama3"}, ready: true},
|
|
{name: "localhost keyless", env: config.LLMConfig{BaseURL: "http://localhost:1234/v1", Model: "m"}, ready: true},
|
|
{name: "private keyless", env: config.LLMConfig{BaseURL: "http://192.168.1.20:11434/v1", Model: "m"}, ready: true},
|
|
{name: "remote keyless", env: config.LLMConfig{BaseURL: "https://api.example.com/v1", Model: "m"}, ready: false},
|
|
{name: "no model", env: config.LLMConfig{BaseURL: "https://api.example.com/v1", APIKey: "sk"}, ready: false},
|
|
{name: "no base url", env: config.LLMConfig{APIKey: "sk", Model: "m"}, ready: false},
|
|
}
|
|
for _, tc := range cases {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
if got := llmGroupUsable(testRuntimeLLM(t, tc.env)); got != tc.ready {
|
|
t.Fatalf("llmGroupUsable = %v, want %v", got, tc.ready)
|
|
}
|
|
})
|
|
}
|
|
}
|