AskaEth
fa33c79e11
- 03.4.1：per-persona 路由载体选 models.json（组内新增 personaModels/personaProviders + 校验 + PersonaRoute()）， Router/RuntimeLLM 快照绑定、loop 流式路径按该 turn 的 persona 路由；主步用路由结果、降级步钉自己的模型 - 03.4.3：本地第三级降级链（LLM_LOCAL_*，Configured() 含 Enabled ⇒ **默认关**），warn 标注来源 - 14.4.4：capacity 接入 L3（分类器接口 + TTL 缓存 + 严格 JSON 解析），CAPACITY_L3_ENABLED **默认 false** - 14.4.1：语义修正 —— group.multimodal 才是「主模型能否收图」（env 路径不再臆断），LLM_VISION_MODEL 只配视觉工具； vision/OCR 可各用不同模型；runtime 不再因有图抢路由（vision.Router 是唯一分层权威） - 14.4.5：新增 image-crop 工具（工具数 5→6，策略默认 owner-only），schema 支持按区域裁切 - 组合根：wire_runtime 读 group.Multimodal；capacity 装配拆出 wire_capacity.go（守 500 行） - 复核：`go build`/`go vet`/各包 `go test -count=1` 全绿；**6 条变异反证**先红后还原（含作者自查出的 2 条假阳性变异） - 未落：按 provider 的 thinking 开关（需 openai_encode.go/openai.go 文件面）、persona awakening（需 loader 文件面）
90 lines
3.1 KiB
Go
90 lines
3.1 KiB
Go
package vision
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"sync"
|
|
"testing"
|
|
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
|
|
)
|
|
|
|
// modelRecorder is a /chat/completions endpoint that records the requested model,
|
|
// so a test can pin which model each direction (Describe / Extract) asked for.
|
|
type modelRecorder struct {
|
|
mu sync.Mutex
|
|
models []string
|
|
}
|
|
|
|
func (r *modelRecorder) seen() []string {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
return append([]string(nil), r.models...)
|
|
}
|
|
|
|
func newModelRecorderServer(t *testing.T) (*httptest.Server, *modelRecorder) {
|
|
t.Helper()
|
|
rec := &modelRecorder{}
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var body struct {
|
|
Model string `json:"model"`
|
|
}
|
|
_ = json.NewDecoder(r.Body).Decode(&body)
|
|
rec.mu.Lock()
|
|
rec.models = append(rec.models, body.Model)
|
|
rec.mu.Unlock()
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = io.WriteString(w, `{"choices":[{"message":{"content":"ok"}}]}`)
|
|
}))
|
|
t.Cleanup(srv.Close)
|
|
return srv, rec
|
|
}
|
|
|
|
// TestOpenAICompatibleAdapter_OCRUsesOwnModel pins the independent OCR model
|
|
// (裁决 14.4.1 的「独立视觉模型 + OCR 模型」): when LLM_OCR_MODEL is set the OCR
|
|
// direction sends it, while Describe keeps the vision model.
|
|
func TestOpenAICompatibleAdapter_OCRUsesOwnModel(t *testing.T) {
|
|
t.Parallel()
|
|
srv, rec := newModelRecorderServer(t)
|
|
ad := NewOpenAICompatible(config.LLMConfig{
|
|
BaseURL: srv.URL, Model: "vision-main", OCRModel: "ocr-special",
|
|
})
|
|
|
|
if _, err := ad.Describe(context.Background(), ImageRef{Source: SourceBase64, Data: "aGVsbG8="}, ""); err != nil {
|
|
t.Fatalf("Describe: %v", err)
|
|
}
|
|
if _, err := ad.Extract(context.Background(), ImageRef{Source: SourceBase64, Data: "aGVsbG8="}); err != nil {
|
|
t.Fatalf("Extract: %v", err)
|
|
}
|
|
seen := rec.seen()
|
|
if len(seen) != 2 || seen[0] != "vision-main" || seen[1] != "ocr-special" {
|
|
t.Fatalf("models seen = %v, want [vision-main ocr-special]", seen)
|
|
}
|
|
}
|
|
|
|
// TestOpenAICompatibleAdapter_OCRWithoutOwnModelFallsBack pins the compatibility
|
|
// edge: without LLM_OCR_MODEL both directions use the vision model.
|
|
func TestOpenAICompatibleAdapter_OCRWithoutOwnModelFallsBack(t *testing.T) {
|
|
t.Parallel()
|
|
srv, rec := newModelRecorderServer(t)
|
|
ad := NewOpenAICompatible(config.LLMConfig{BaseURL: srv.URL, Model: "vision-main"})
|
|
|
|
if _, err := ad.Extract(context.Background(), ImageRef{Source: SourceBase64, Data: "aGVsbG8="}); err != nil {
|
|
t.Fatalf("Extract: %v", err)
|
|
}
|
|
if seen := rec.seen(); len(seen) != 1 || seen[0] != "vision-main" {
|
|
t.Fatalf("models seen = %v, want [vision-main]", seen)
|
|
}
|
|
// WithOCRModel overrides cfg.OCRModel, so an explicit option still wins.
|
|
ad2 := NewOpenAICompatible(config.LLMConfig{BaseURL: srv.URL, Model: "vision-main", OCRModel: "cfg-ocr"}, WithOCRModel("opt-ocr"))
|
|
if _, err := ad2.Extract(context.Background(), ImageRef{Source: SourceBase64, Data: "aGVsbG8="}); err != nil {
|
|
t.Fatalf("Extract: %v", err)
|
|
}
|
|
if seen := rec.seen(); len(seen) != 2 || seen[1] != "opt-ocr" {
|
|
t.Fatalf("models seen = %v, want the explicit option opt-ocr", seen)
|
|
}
|
|
}
|