@@ -336,8 +336,15 @@ def TensorCreate_FromValues(record):
336336 tensor = redisAI.createTensorFromValues('DOUBLE', [2,2], [1.0, 2.0, 3.0, 4.0])
337337 redisAI.setTensorInKey('test1_res{1}', tensor)
338338 return "test1_OK"
339+
340+ def TensorCreate_FromBlob(record):
341+ tensor_blob = bytearray([5, 6, 7, 8])
342+ tensor = redisAI.createTensorFromBlob('INT8', [2,2], tensor_blob)
343+ redisAI.setTensorInKey('test2_res{1}', tensor)
344+ return "test2_OK"
339345
340346GB("CommandReader").map(TensorCreate_FromValues).register(trigger="TensorCreate_FromValues_test1")
347+ GB("CommandReader").map(TensorCreate_FromBlob).register(trigger="TensorCreate_FromBlob_test2")
341348 '''
342349
343350 con = env .getConnection ()
@@ -348,3 +355,9 @@ def TensorCreate_FromValues(record):
348355
349356 values = con .execute_command ('AI.TENSORGET' , 'test1_res{1}' , 'VALUES' )
350357 env .assertEqual (values , [b'1' , b'2' , b'3' , b'4' ])
358+
359+ ret = con .execute_command ('rg.trigger' , 'TensorCreate_FromBlob_test2' )
360+ env .assertEqual (ret [0 ], b'test2_OK' )
361+
362+ values = con .execute_command ('AI.TENSORGET' , 'test2_res{1}' , 'VALUES' )
363+ env .assertEqual (values , [5 , 6 , 7 , 8 ])
0 commit comments