init push

backend/repo/mq/rag.go

@@ -0,0 +1,30 @@
+package mq
+
+import (
+	"context"
+	"encoding/json"
+
+	"github.com/chaitin/panda-wiki/domain"
+	"github.com/chaitin/panda-wiki/mq"
+)
+
+type RAGRepository struct {
+	producer mq.MQProducer
+}
+
+func NewRAGRepository(producer mq.MQProducer) *RAGRepository {
+	return &RAGRepository{producer: producer}
+}
+
+func (r *RAGRepository) AsyncUpdateNodeReleaseVector(ctx context.Context, request []*domain.NodeReleaseVectorRequest) error {
+	for _, req := range request {
+		requestBytes, err := json.Marshal(req)
+		if err != nil {
+			return err
+		}
+		if err := r.producer.Produce(ctx, domain.VectorTaskTopic, "", requestBytes); err != nil {
+			return err
+		}
+	}
+	return nil
+}