refactor: 将客服后端重构为宿主可嵌入模块
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力,补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现,将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储,并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议,补齐模块初始化、业务动作和公共载荷等回归测试。
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user