t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
55 lines
1.8 KiB
Go
55 lines
1.8 KiB
Go
package rag
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/contract"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/ai"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/ai/rag/vectordb"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/config"
|
|
)
|
|
|
|
type dimensionTestPlatformProvider struct{}
|
|
|
|
func (dimensionTestPlatformProvider) ModelSource(context.Context) (string, error) {
|
|
return contract.ModelSourcePlatform, nil
|
|
}
|
|
|
|
func (dimensionTestPlatformProvider) Config(context.Context) (*contract.PlatformAIConfig, error) {
|
|
return &contract.PlatformAIConfig{
|
|
APIKey: "license-signed",
|
|
BaseURL: "https://platform.example/v1",
|
|
EmbeddingModel: "qwen3.7-text-embedding",
|
|
EmbeddingDimension: 4,
|
|
}, nil
|
|
}
|
|
|
|
func (dimensionTestPlatformProvider) Status(context.Context) (*contract.PlatformAIStatus, error) {
|
|
return &contract.PlatformAIStatus{Enabled: true, EmbeddingEnabled: true}, nil
|
|
}
|
|
|
|
func TestEnsureCollectionRejectsChangedEmbeddingDimension(t *testing.T) {
|
|
if err := vectordb.Init(&config.VectorDBConfig{Path: filepath.Join(t.TempDir(), "vectors.db")}); err != nil {
|
|
t.Fatalf("vectordb.Init() error = %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = vectordb.Close() })
|
|
provider := vectordb.GetProvider()
|
|
if err := provider.CreateCollection(context.Background(), knowledgeCollectionName, 3); err != nil {
|
|
t.Fatalf("CreateCollection() error = %v", err)
|
|
}
|
|
|
|
ai.SetPlatformAIProvider(dimensionTestPlatformProvider{})
|
|
t.Cleanup(func() { ai.SetPlatformAIProvider(nil) })
|
|
|
|
err := Index.EnsureCollection(context.Background())
|
|
if err == nil {
|
|
t.Fatal("expected dimension mismatch error")
|
|
}
|
|
if message := err.Error(); !strings.Contains(message, "dimension is 3") || !strings.Contains(message, "uses 4") || !strings.Contains(message, "rebuild") {
|
|
t.Fatalf("unexpected dimension mismatch error: %v", err)
|
|
}
|
|
}
|