t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package vectordb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/config"
|
|
)
|
|
|
|
var defaultProvider Provider
|
|
|
|
func Init(cfg *config.VectorDBConfig) error {
|
|
if cfg == nil {
|
|
return fmt.Errorf("libsql vector database config is required")
|
|
}
|
|
provider, err := NewLibSQLProvider(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if defaultProvider != nil {
|
|
_ = defaultProvider.Close()
|
|
}
|
|
defaultProvider = provider
|
|
return nil
|
|
}
|
|
|
|
func GetProvider() Provider {
|
|
return defaultProvider
|
|
}
|
|
|
|
func Close() error {
|
|
if defaultProvider != nil {
|
|
err := defaultProvider.Close()
|
|
defaultProvider = nil
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func CreateCollection(ctx context.Context, name string, dimension int) error {
|
|
if defaultProvider == nil {
|
|
return fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.CreateCollection(ctx, name, dimension)
|
|
}
|
|
|
|
func DeleteCollection(ctx context.Context, name string) error {
|
|
if defaultProvider == nil {
|
|
return fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.DeleteCollection(ctx, name)
|
|
}
|
|
|
|
func GetCollection(ctx context.Context, name string) (*CollectionInfo, error) {
|
|
if defaultProvider == nil {
|
|
return nil, fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.GetCollection(ctx, name)
|
|
}
|
|
|
|
func ListCollections(ctx context.Context) ([]string, error) {
|
|
if defaultProvider == nil {
|
|
return nil, fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.ListCollections(ctx)
|
|
}
|
|
|
|
func UpsertVectors(ctx context.Context, collectionName string, vectors []Vector) error {
|
|
if defaultProvider == nil {
|
|
return fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.UpsertVectors(ctx, collectionName, vectors)
|
|
}
|
|
|
|
func DeleteVectors(ctx context.Context, collectionName string, ids []string) error {
|
|
if defaultProvider == nil {
|
|
return fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.DeleteVectors(ctx, collectionName, ids)
|
|
}
|
|
|
|
func Search(ctx context.Context, req *SearchRequest) ([]SearchResult, error) {
|
|
if defaultProvider == nil {
|
|
return nil, fmt.Errorf("vectordb provider not initialized")
|
|
}
|
|
return defaultProvider.Search(ctx, req)
|
|
}
|