@@ -30,12 +30,38 @@ typedef struct RAI_Error RAI_Error;
3030#define REDISAI_DEVICE_CPU 0
3131#define REDISAI_DEVICE_GPU 1
3232
33+ #define RedisAI_ErrorCode int
34+ #define RedisAI_ErrorCode_OK 0
35+ #define RedisAI_ErrorCode_EMODELIMPORT 1
36+ #define RedisAI_ErrorCode_EMODELCONFIGURE 2
37+ #define RedisAI_ErrorCode_EMODELCREATE 3
38+ #define RedisAI_ErrorCode_EMODELRUN 4
39+ #define RedisAI_ErrorCode_EMODELSERIALIZE 5
40+ #define RedisAI_ErrorCode_EMODELFREE 6
41+ #define RedisAI_ErrorCode_ESCRIPTIMPORT 7
42+ #define RedisAI_ErrorCode_ESCRIPTCONFIGURE 8
43+ #define RedisAI_ErrorCode_ESCRIPTCREATE 9
44+ #define RedisAI_ErrorCode_ESCRIPTRUN 10
45+ #define RedisAI_ErrorCode_EUNSUPPORTEDBACKEND 11
46+ #define RedisAI_ErrorCode_EBACKENDNOTLOADED 12
47+ #define RedisAI_ErrorCode_ESCRIPTFREE 13
48+ #define RedisAI_ErrorCode_ETENSORSET 14
49+ #define RedisAI_ErrorCode_ETENSORGET 15
50+ #define RedisAI_ErrorCode_EDAGRUN 17
51+
3352enum RedisAI_DataFmt {
3453 REDISAI_DATA_BLOB = 0 ,
3554 REDISAI_DATA_VALUES ,
3655 REDISAI_DATA_NONE
3756};
3857
58+ int MODULE_API_FUNC (RedisAI_InitError )(RAI_Error * * err );
59+ void MODULE_API_FUNC (RedisAI_ClearError )(RAI_Error * err );
60+ void MODULE_API_FUNC (RedisAI_FreeError )(RAI_Error * err );
61+ const char * MODULE_API_FUNC (RedisAI_GetError )(RAI_Error * err );
62+ const char * MODULE_API_FUNC (RedisAI_GetErrorOneLine )(RAI_Error * err );
63+ RedisAI_ErrorCode MODULE_API_FUNC (RedisAI_GetErrorCode )(RAI_Error * err );
64+
3965RAI_Tensor * MODULE_API_FUNC (RedisAI_TensorCreate )(const char * dataTypeStr , long long * dims , int ndims );
4066RAI_Tensor * MODULE_API_FUNC (RedisAI_TensorCreateByConcatenatingTensors )(RAI_Tensor * * ts , long long n );
4167RAI_Tensor * MODULE_API_FUNC (RedisAI_TensorCreateBySlicingTensor )(RAI_Tensor * t , long long offset , long long len );
@@ -96,6 +122,13 @@ static int RedisAI_Initialize(RedisModuleCtx* ctx){
96122 return REDISMODULE_ERR ;
97123 }
98124
125+ REDISAI_MODULE_INIT_FUNCTION (ctx , InitError );
126+ REDISAI_MODULE_INIT_FUNCTION (ctx , ClearError );
127+ REDISAI_MODULE_INIT_FUNCTION (ctx , FreeError );
128+ REDISAI_MODULE_INIT_FUNCTION (ctx , GetError );
129+ REDISAI_MODULE_INIT_FUNCTION (ctx , GetErrorOneLine );
130+ REDISAI_MODULE_INIT_FUNCTION (ctx , GetErrorCode );
131+
99132 REDISAI_MODULE_INIT_FUNCTION (ctx , GetLLAPIVersion );
100133
101134 REDISAI_MODULE_INIT_FUNCTION (ctx , TensorCreate );
0 commit comments