Add AI.INFO command

Author: lantiga

src/backends.c

@@ -21,7 +21,6 @@ RedisModuleString* RAI_GetModulePath(RedisModuleCtx *ctx) {
   return module_path;
 }
 
-
 RedisModuleString* RAI_GetBackendsPath(RedisModuleCtx *ctx) {
   Dl_info info;
   RedisModuleString* backends_path = NULL;
@@ -36,6 +35,20 @@ RedisModuleString* RAI_GetBackendsPath(RedisModuleCtx *ctx) {
   return backends_path;
 }
 
+const char* RAI_BackendName(int backend) {
+  switch (backend) {
+    case RAI_BACKEND_TENSORFLOW:
+      return "TF";
+    case RAI_BACKEND_TFLITE:
+      return "TFLITE";
+    case RAI_BACKEND_TORCH:
+      return "TORCH";
+    case RAI_BACKEND_ONNXRUNTIME:
+      return "ONNX";
+  }
+  return NULL;
+}
+
 int RAI_LoadBackend_TensorFlow(RedisModuleCtx *ctx, const char *path) {
   if (RAI_backends.tf.model_run != NULL) {
     RedisModule_Log(ctx, "warning", "Could not load TF backend: backend already loaded");

src/backends.h

@@ -35,4 +35,6 @@ char* RAI_BackendsPath;
 int RAI_LoadBackend(RedisModuleCtx *ctx, int backend, const char *path);
 int RAI_LoadDefaultBackend(RedisModuleCtx *ctx, int backend);
 
-#endif
+const char* RAI_BackendName(int backend);
+
+#endif

src/model.c

@@ -135,25 +135,11 @@ static void RAI_Model_AofRewrite(RedisModuleIO *aof, RedisModuleString *key, voi
     array_append(outputs_, RedisModule_CreateString(ctx, model->outputs[i], strlen(model->outputs[i])));
   }
 
-  char backend[256] = "";
-  switch (model->backend) {
-    case RAI_BACKEND_TENSORFLOW:
-      strcpy(backend, "TF");
-      break;
-    case RAI_BACKEND_TFLITE:
-      strcpy(backend, "TFLITE");
-      break;
-    case RAI_BACKEND_TORCH:
-      strcpy(backend, "TORCH");
-      break;
-    case RAI_BACKEND_ONNXRUNTIME:
-      strcpy(backend, "ONNX");
-      break;
-  }
+  const char* backendstr = RAI_BackendName(model->backend);
 
   RedisModule_EmitAOF(aof, "AI.MODELSET", "slccvcvb",
                       key,
-                      backend, model->devicestr,
+                      backendstr, model->devicestr,
                       "INPUTS", inputs_, model->ninputs,
                       "OUTPUTS", outputs_, model->noutputs,
                       buffer, len);

src/model_struct.h

@@ -17,8 +17,6 @@ typedef struct RAI_Model {
   char **outputs;
   size_t noutputs;
   long long refCount;
-  long long backend_calls;
-  long long backend_us;
   void* data;
 } RAI_Model;