Create new test module that uses the LLAPI. Added a new flow test that loads this module and tests the LLAPI (currently contains a basic check only)

Author: alonre24

src/background_workers.c

@@ -178,7 +178,7 @@ void *RedisAI_Run_ThreadMain(void *arg) {
                         RedisAI_RunInfo *orig = rinfo->orig_copy;
                         long long dagRefCount = RAI_DagRunInfoFreeShallowCopy(rinfo);
                         if (dagRefCount == 0) {
-                            RedisAI_OnFinishCtx finish_ctx = (RedisAI_RunInfo *)orig;
+                            RedisAI_OnFinishCtx *finish_ctx = orig;
                             orig->OnFinish(finish_ctx, orig->private_data);
                         }
 
@@ -417,7 +417,7 @@ void *RedisAI_Run_ThreadMain(void *arg) {
                         RedisAI_RunInfo *orig = rinfo->orig_copy;
                         long long dagRefCount = RAI_DagRunInfoFreeShallowCopy(rinfo);
                         if (dagRefCount == 0) {
-                            RedisAI_OnFinishCtx finish_ctx = (RedisAI_RunInfo *)orig;
+                            RedisAI_OnFinishCtx *finish_ctx = orig;
                             orig->OnFinish(finish_ctx, orig->private_data);
                         }
                     } else {
@@ -449,7 +449,7 @@ void *RedisAI_Run_ThreadMain(void *arg) {
                 // If the reference count for the DAG is zero and the client is still around,
                 // then we actually unblock the client
                 if (dagRefCount == 0) {
-                    RedisAI_OnFinishCtx finish_ctx = (RedisAI_RunInfo *)orig;
+                    RedisAI_OnFinishCtx *finish_ctx = orig;
                     orig->OnFinish(finish_ctx, orig->private_data);
                 }
             }

src/dag.c

@@ -34,6 +34,7 @@
 
 #include "model.h"
 #include "redisai.h"
+#include "background_workers.h"
 #include "rmutil/alloc.h"
 #include "rmutil/args.h"
 #include "run_info.h"
@@ -889,7 +890,7 @@ void RedisAI_Disconnected(RedisModuleCtx *ctx, RedisModuleBlockedClient *bc) {
     RedisModule_Log(ctx, "warning", "Blocked client %p disconnected!", (void *)bc);
 }
 
-// Parse the DAG run command and return true if it is a valid command to execute.
+// Parse the DAG run command and return REDISMODULE_OK only if it is a valid command to execute.
 static int DAG_CommandParser(RedisModuleCtx *ctx, RedisModuleString **argv, int argc, int dagMode,
                              RedisAI_RunInfo **rinfo_ptr) {
 
@@ -1229,7 +1230,7 @@ static int DAG_CommandParser(RedisModuleCtx *ctx, RedisModuleString **argv, int
 
 // Add Shallow copies of the DAG run info to the devices' queues.
 // Return REDISMODULE_OK in case of success, REDISMODULE_ERR if (at least) one insert op had failed.
-static bool DAG_InsertDAGToQueue(RedisAI_RunInfo *rinfo) {
+static int DAG_InsertDAGToQueue(RedisAI_RunInfo *rinfo) {
     const char **devices = array_new(const char *, 10);
 
     for (long long i = 0; i < array_len(rinfo->dagOps); i++) {
@@ -1280,7 +1281,7 @@ static bool DAG_InsertDAGToQueue(RedisAI_RunInfo *rinfo) {
             array_free(rinfo_copies);
             array_free(run_queues_info);
             RAI_SetError(rinfo->err, RAI_EDAGRUN, "ERR Queue not initialized for device");
-            rinfo->OnFinish((RedisAI_OnFinishCtx)rinfo, rinfo->private_data);
+            rinfo->OnFinish((RedisAI_OnFinishCtx *)rinfo, rinfo->private_data);
             return REDISMODULE_ERR;
         }
         run_queues_info = array_append(run_queues_info, run_queue_info);
@@ -1302,7 +1303,7 @@ static bool DAG_InsertDAGToQueue(RedisAI_RunInfo *rinfo) {
     return REDISMODULE_OK;
 }
 
-void DAG_ReplyAndUnblock(RedisAI_OnFinishCtx ctx, void *private_data) {
+void DAG_ReplyAndUnblock(RedisAI_OnFinishCtx *ctx, void *private_data) {
 
     RedisAI_RunInfo *rinfo = (RedisAI_RunInfo *)ctx;
     if (rinfo->client)

src/dag.h

@@ -175,6 +175,6 @@ int RedisAI_ProcessDagRunCommand(RedisModuleCtx *ctx, RedisModuleString **argv,
  * @param ctx Context object that contains errors and results
  * @param private_data is a pointer to the DAG run info struct
  */
-void DAG_ReplyAndUnblock(RedisAI_OnFinishCtx ctx, void *private_data);
+void DAG_ReplyAndUnblock(RedisAI_OnFinishCtx *ctx, void *private_data);
 
 #endif /* SRC_DAG_H_ */

src/model.c

@@ -8,6 +8,7 @@
  */
 
 #include "model.h"
+#include "version.h"
 #include "backends.h"
 #include "backends/util.h"
 #include "model_struct.h"

src/redisai.h

@@ -1,14 +1,11 @@
 #ifndef SRC_REDISAI_H_
 #define SRC_REDISAI_H_
 
-#include "background_workers.h"
-#include "model_struct.h"
-#include "redismodule.h"
-#include "util/dict.h"
-#include "version.h"
 #include <stdbool.h>
+#include "redismodule.h"
 
-#define MODULE_API_FUNC(x) (*x)
+#define REDISAI_LLAPI_VERSION 1
+#define MODULE_API_FUNC(x)    (*x)
 
 #ifndef REDISAI_H_INCLUDE
 typedef struct RAI_Tensor RAI_Tensor;
@@ -18,6 +15,8 @@ typedef struct RAI_Script RAI_Script;
 typedef struct RAI_ModelRunCtx RAI_ModelRunCtx;
 typedef struct RAI_ScriptRunCtx RAI_ScriptRunCtx;
 typedef struct RAI_Error RAI_Error;
+typedef struct RAI_ModelOpts RAI_ModelOpts;
+typedef struct RAI_OnFinishCtx RAI_OnFinishCtx;
 #endif
 
 #define REDISAI_BACKEND_TENSORFLOW  0