tf version

Author: DvirDukhan

src/backends.c

@@ -132,6 +132,16 @@ int RAI_LoadBackend_TensorFlow(RedisModuleCtx *ctx, const char *path) {
         return REDISMODULE_ERR;
     }
 
+    backend.get_version = (const char*(*)(void))(unsigned long)dlsym(handle, "RAI_GetBackendVersionTF");
+    if (backend.get_version == NULL) {
+        dlclose(handle);
+        RedisModule_Log(ctx, "warning",
+                        "Backend does not export RAI_GetBackendVersionTF. TF backend "
+                        "not loaded from %s",
+                        path);
+        return REDISMODULE_ERR;
+    }
+
     RAI_backends.tf = backend;
 
     RedisModule_Log(ctx, "notice", "TF backend loaded from %s", path);

src/backends.h

@@ -74,6 +74,9 @@ typedef struct RAI_LoadedBackend {
     // RAI_ScriptRunCtx pointer
     int (*script_run)(RAI_ScriptRunCtx *, RAI_Error *);
 
+    // Returns the backend version.
+    const char* (*get_version)(void);
+
 } RAI_LoadedBackend;
 
 typedef struct RAI_LoadedBackends {

src/backends/tensorflow.c

@@ -591,3 +591,7 @@ int RAI_ModelSerializeTF(RAI_Model *model, char **buffer, size_t *len, RAI_Error
 
     return 0;
 }
+
+const char* RAI_GetBackendVersionTF(void) {
+    return TF_Version();
+}

src/backends/tensorflow.h

@@ -19,4 +19,6 @@ int RAI_ModelRunTF(RAI_ModelRunCtx **mctxs, RAI_Error *error);
 
 int RAI_ModelSerializeTF(RAI_Model *model, char **buffer, size_t *len, RAI_Error *error);
 
+const char* RAI_GetBackendVersionTF(void);
+
 #endif /* SRC_BACKENDS_TENSORFLOW_H_ */

src/redisai.c

@@ -805,12 +805,73 @@ int RedisAI_ScriptScan_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **arg
     return REDISMODULE_OK;
 }
 
+void _RedisAI_Info(RedisModuleCtx* ctx) {
+    RedisModuleString* rai_version = RedisModule_CreateStringPrintf(ctx, "%d.%d.%d", REDISAI_VERSION_MAJOR, REDISAI_VERSION_MINOR, REDISAI_VERSION_PATCH);
+    RedisModuleString* llapi_version = RedisModule_CreateStringPrintf(ctx, "%d", REDISAI_LLAPI_VERSION);
+    RedisModuleString* rdb_version = RedisModule_CreateStringPrintf(ctx, "%llu", REDISAI_ENC_VER);
+
+    int reponse_len = 6;
+
+    if(RAI_backends.tf.get_version){
+        reponse_len+=2;   
+    }
+
+    if(RAI_backends.torch.get_version){
+        reponse_len+=2;   
+    }
+
+    if(RAI_backends.tflite.get_version){
+        reponse_len+=2;   
+    }
+
+    if(RAI_backends.onnx.get_version){
+        reponse_len+=2;   
+    }
+
+    RedisModule_ReplyWithArray(ctx, reponse_len);
+
+    RedisModule_ReplyWithSimpleString(ctx, "Version");
+    RedisModule_ReplyWithString(ctx, rai_version);
+
+    // TODO: Add Git SHA
+
+    RedisModule_ReplyWithSimpleString(ctx, "Low Level API Version");
+    RedisModule_ReplyWithString(ctx, llapi_version);
+
+
+    RedisModule_ReplyWithSimpleString(ctx, "RDB Encoding version");
+    RedisModule_ReplyWithString(ctx, llapi_version);
+
+    if(RAI_backends.tf.get_version){
+        RedisModule_ReplyWithSimpleString(ctx, "TensorFlow version"); 
+        RedisModule_ReplyWithSimpleString(ctx, RAI_backends.tf.get_version()); 
+    }
+
+    if(RAI_backends.torch.get_version){
+        RedisModule_ReplyWithSimpleString(ctx, "Torch version"); 
+        RedisModule_ReplyWithSimpleString(ctx, RAI_backends.torch.get_version()); 
+    }
+
+
+    RedisModule_FreeString(ctx, rai_version);
+    RedisModule_FreeString(ctx, llapi_version);
+    RedisModule_FreeString(ctx, rdb_version);
+}
+
 /**
  * AI.INFO <model_or_script_key> [RESETSTAT]
  */
 int RedisAI_Info_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
-    if (argc != 2 && argc != 3)
+    if (argc > 3)
         return RedisModule_WrongArity(ctx);
+
+
+    if(argc == 1) {
+        _RedisAI_Info(ctx);
+        return REDISMODULE_OK;
+    }
+
+
     RedisModuleString *runkey = argv[1];
     struct RedisAI_RunStats *rstats = NULL;
     if (RAI_GetRunStats(runkey, &rstats) == REDISMODULE_ERR) {

src/version.h

@@ -1,5 +1,4 @@
-#ifndef SRC_VERSION_H_
-#define SRC_VERSION_H_
+#pragma once
 
 #define REDISAI_VERSION_MAJOR 99
 #define REDISAI_VERSION_MINOR 99
@@ -13,4 +12,3 @@
 
 static const long long REDISAI_ENC_VER = 1;
 
-#endif /* SRC_VERSION_H_ */