Fix leaks

Author: alonre24

src/DAG/dag_builder.c

@@ -36,6 +36,7 @@ int RAI_DAGLoadTensor(RAI_DAGRunCtx *run_info, const char *t_name, RAI_Tensor *t
     AI_dictAdd(rinfo->tensorsNamesToIndices, (void *)key_name, (void *)instance);
     RAI_TensorGetShallowCopy(tensor);
     rinfo->dagSharedTensors = array_append(rinfo->dagSharedTensors, (void *)tensor);
+    RedisModule_FreeString(NULL, key_name);
 
     return REDISMODULE_OK;
 }

src/DAG/dag_execute.c

@@ -14,6 +14,7 @@ int ValidatePersistKeys(RedisAI_RunInfo *rinfo, AI_dict *tensorsNamesToInd,
             AI_dictEntry *entry = AI_dictFind(tensorsNamesToInd, persist_key);
             if (!entry) {
                 RAI_SetError(rinfo->err, RAI_EDAGRUN, "ERR PERSIST key cannot be found in DAG");
+                AI_dictReleaseIterator(iter);
                 return REDISMODULE_ERR;
             }
         }
@@ -52,6 +53,11 @@ int MapTensorsKeysToIndices(RedisAI_RunInfo *rinfo, AI_dict *tensorsNamesToInd)
                 rinfo->dagSharedTensors = array_append(rinfo->dagSharedTensors, NULL);
             }
             currentOp->outkeys_indices = array_append(currentOp->outkeys_indices, *ind);
+            AI_dictEntry *entry = AI_dictFind(tensorsNamesToInd, (void *)key);
+            // If this key was already in the dict, remove and free the previous index
+            if (entry) {
+                RedisModule_Free(AI_dictGetVal(entry));
+            }
             AI_dictReplace(tensorsNamesToInd, (void *)key, (void *)ind);
         }
     }