Method insertWithAutoCreate

api/service/rag/entity_vector.go:266–289

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

}

func (s *EntityVectorService) insertWithAutoCreate(ctx context.Context, collection string, record vectorstore.VectorRecord, dimension int) error {
	err := s.vectorDB.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",
			}
			if createErr := s.vectorDB.CreateCollection(ctx, collectionConfig); createErr != nil && !vectorstore.IsExistsError(createErr) {
				return createErr
			}
			if insertErr := s.vectorDB.Insert(ctx, collection, []vectorstore.VectorRecord{record}); insertErr != nil {
				return insertErr
			}
			return nil
		}
	}
	return err
}

func (s *EntityVectorService) updateWithAutoCreate(ctx context.Context, collection string, record vectorstore.VectorRecord, dimension int) error {
	err := s.vectorDB.Update(ctx, collection, []vectorstore.VectorRecord{record})