Function DAGAddModelRun

plugins/python/redisgears_python.c:2381–2467

}

static PyObject* DAGAddModelRun(PyObject *self, PyObject *args, PyObject *kargs) {
    verifyRedisAILoaded();
    PyDAGRunner *pyDag = (PyDAGRunner *)self;
    if(!_IsDagAPISupported(RedisAI_DAGCreateModelRunOp) || !_IsValidDag(pyDag)) {
        return NULL;
    }
    PyObject* modelName = GearsPyDict_GetItemString(kargs, "name");
    if (!modelName) {
        PyErr_SetString(GearsError, "Model key name was not given");
        return NULL;
    }
    if(!PyUnicode_Check(modelName)){
        PyErr_SetString(GearsError, "Model name argument must be a string");
        return NULL;
    }

    RAI_Error *err;
    RedisAI_InitError(&err);
    const char* modelNameStr = PyUnicode_AsUTF8AndSize(modelName, NULL);

    // Create MODELRUN op after bringing model from keyspace (raise an exception if it does not exist)
    RAI_DAGRunOp *modelRunOp = _createModelRunOp(modelNameStr, err);
    if (modelRunOp == NULL) {
        RedisAI_FreeError(err);
        return NULL;
    }
    PyObject *ret = NULL;

    // Add to the modelRun op with its inputs and output keys and insert it to the DAG
    PyObject *inputsIter = NULL;
    PyObject *outputsIter = NULL;
    PyObject *modelOutputs = NULL;
    PyObject *modelInputs = GearsPyDict_GetItemString(kargs, "inputs");
    if (!modelInputs) {
        PyErr_SetString(GearsError, "Must specify model inputs");
        goto cleanup;
    }
    inputsIter = PyObject_GetIter(modelInputs);
    if (!inputsIter) {
        PyErr_SetString(GearsError, "Model inputs must be iterable");
        goto cleanup;
    }
    PyObject* input;
    while((input = (PyObject*)PyIter_Next(inputsIter)) != NULL) {
        if(!PyUnicode_Check(input)){
            PyErr_SetString(GearsError, "Input name must be a string");
            goto cleanup;
        }
        const char* inputStr = PyUnicode_AsUTF8AndSize(input, NULL);
        RedisAI_DAGRunOpAddInput(modelRunOp, inputStr);
    }

    modelOutputs = GearsPyDict_GetItemString(kargs, "outputs");
    if (!modelOutputs) {
        PyErr_SetString(GearsError, "Must specify model outputs");
        goto cleanup;
    }
    outputsIter = PyObject_GetIter(modelOutputs);