Method insertWithAutoCreateCollection

api/service/rag/embedding.go:873–906

insertWithAutoCreateCollection 插入向量，如果集合不存在则自动创建

(ctx context.Context, collection string, record vectorstore.VectorRecord, dimension int)

// insertWithAutoCreateCollection 插入向量，如果集合不存在则自动创建
func (s *EmbeddingService) insertWithAutoCreateCollection(ctx context.Context, collection string, record vectorstore.VectorRecord, dimension int) error {
	// 尝试直接插入
	err := s.vectorStore.Insert(ctx, collection, []vectorstore.VectorRecord{record})
	if err == nil {
		return nil
	}

	// 检查是否是集合不存在的错误
	if vsErr, ok := err.(*vectorstore.VectorStoreError); ok {
		if vsErr.Code == vectorstore.ErrCodeCollectionNotFound || vsErr.Code == vectorstore.ErrCodeUnknown {
			// 尝试创建集合
			collectionConfig := vectorstore.CollectionConfig{
				Name:      collection,
				Dimension: dimension,
				Metric:    "cosine",
				IndexType: "HNSW",
			}

			logger.SysLogf("集合不存在，正在创建集合: %s", collection)
			if createErr := s.vectorStore.CreateCollection(ctx, collectionConfig); createErr != nil && !vectorstore.IsExistsError(createErr) {
				return fmt.Errorf("创建集合失败: %v", createErr)
			}

			// 重新尝试插入
			if insertErr := s.vectorStore.Insert(ctx, collection, []vectorstore.VectorRecord{record}); insertErr != nil {
				return fmt.Errorf("创建集合后插入向量失败: %v", insertErr)
			}
			return nil
		}
	}

	// 其他错误，使用重试机制
	return s.insertWithRetry(ctx, collection, record)
}

// insertWithRetry 带重试机制的插入
func (s *EmbeddingService) insertWithRetry(ctx context.Context, collection string, record vectorstore.VectorRecord) error {