Method updateWithAutoCreate

api/service/rag/entity_vector.go:291–314

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

}

func (s *EntityVectorService) updateWithAutoCreate(ctx context.Context, collection string, record vectorstore.VectorRecord, dimension int) error {
	err := s.vectorDB.Update(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 updateErr := s.vectorDB.Update(ctx, collection, []vectorstore.VectorRecord{record}); updateErr != nil {
				return updateErr
			}
			return nil
		}
	}
	return err
}

func (s *EntityVectorService) DeleteCollection(ctx context.Context, collection string) error {
	if s.vectorDB == nil {