redis缓存替换+pgvector向量替换

ai-chat-service/chat-server/main.go

@@ -10,7 +10,6 @@ import (
 	"ai-chat-service/pkg/config"
 	"ai-chat-service/pkg/db/mysql"
 	"ai-chat-service/pkg/db/redis"
-	"ai-chat-service/pkg/db/vector"
 	"ai-chat-service/pkg/log"
 	"ai-chat-service/proto"
 	"flag"
@@ -57,17 +56,19 @@ func main() {
 	mysql.InitMysql(cnf)
 	// 初始化redis
 	redis.InitRedisPool(cnf)
-	// 初始化向量数据库
-	vector.InitDB(cnf)
 
 	recordsData := data.NewChatRecordsData(mysql.GetDB())
+	vectorRecordsData, err := vector_data.NewChatRecordsData(cnf)
+	if err != nil {
+		log.Fatal(err)
+	}
 
 	lis, err := net.Listen("tcp", fmt.Sprintf("%s:%d", cnf.Server.IP, cnf.Server.Port))
 	if err != nil {
 		log.Fatal(err)
 	}
 	s := grpc.NewServer(grpc.UnaryInterceptor(interceptor.UnaryAuthInterceptor), grpc.StreamInterceptor(metrics_app.NewStreamMiddleware(registry).WrapHandler()))
-	service := server.NewChatService(recordsData, vector_data.NewChatRecordsData(cnf, vector.GetVdb()), cnf, logger, busMetrics)
+	service := server.NewChatService(recordsData, vectorRecordsData, cnf, logger, busMetrics)
 	proto.RegisterChatServer(s, service)
 
 	healthCheckSrv := health.NewServer()