Function FinishAsyncModelRun

plugins/python/redisgears_python.c:3978–4020

}

static void FinishAsyncModelRun(RAI_OnFinishCtx *onFinishCtx, void *private_data) {

	RAI_Error *error;
	RedisAI_InitError(&error);
	RAI_ModelRunCtx* mctx = RedisAI_GetAsModelRunCtx(onFinishCtx, error);

	RedisGearsPy_LOCK
	PyObject* future = private_data;
	PyObject* pArgs = PyTuple_New(2);
	PyTuple_SetItem(pArgs, 0, future);

	if (RedisAI_GetErrorCode(error) != RedisAI_ErrorCode_OK) {
		const char *errStr = RedisAI_GetError(error);
		PyObject* pyErr = PyUnicode_FromStringAndSize(errStr, strlen(errStr));
		PyTuple_SetItem(pArgs, 1, pyErr);
		PyObject_CallObject(setFutureExceptionFunction, pArgs);
		goto finish;
	}

    PyObject* tensorList = PyList_New(0);
	for(size_t i = 0 ; i < RedisAI_ModelRunCtxNumOutputs(mctx) ; ++i){
		PyTensor* pyt = PyObject_New(PyTensor, &PyTensorType);
		pyt->t = RedisAI_TensorGetShallowCopy(RedisAI_ModelRunCtxOutputTensor(mctx, i));
		PyList_Append(tensorList, (PyObject*)pyt);
		GearsPyDecRef((PyObject*)pyt);
	}

	PyTuple_SetItem(pArgs, 1, tensorList);
	PyObject* r = PyObject_CallObject(setFutureResultsFunction, pArgs);
	if(!r){
		char* err = getPyError();
		RedisModule_Log(NULL, "warning", "Error happened when releasing execution future, error='%s'", err);
		RG_FREE(err);
	}else{
		GearsPyDecRef(r);
	}

	finish:
	RedisAI_FreeError(error);
	RedisAI_ModelRunCtxFree(mctx);
	GearsPyDecRef(pArgs);
	RedisGearsPy_UNLOCK
}

static PyObject* modelRunnerRunAsync(PyObject *cls, PyObject *args){
	verifyRedisAILoaded();