code7-leia 0.2.20 → 0.2.22

package/dist/code7-leia.esm.js

@@ -469,7 +469,7 @@ var enTranslation = {
       deleteFileTitle: 'Delete a file',
       deleteFile: 'Important! By deleting this file, you will lose all its information and will not be able to recover it.',
       trainingTitle: 'Presset Training',
-      trainingDescription: 'Select one or more pressets to be trained. Note: If none are selected, Lira will train all pressets.'
+      trainingDescription: 'Select one or more pressets to be trained. Note: If none are selected, Le-IA will train all pressets.'
     }
   },
   buttons: {
@@ -484,6 +484,7 @@ var enTranslation = {
   testArea: {
     description: 'Type a sentence to test artificial intelligence',
     typeSentence: 'Type a sentence',
+    typePrompt: 'Insert a custom prompt here',
     selectPersona: 'Select a Persona',
     selectPresset: 'Select a presset',
     emptyState: {
@@ -515,17 +516,17 @@ var enTranslation = {
     Test: {
       fail: {
         title: 'Failure',
-        description: 'Error testing Lira'
+        description: 'Error testing Le-IA'
       }
     },
     Train: {
       fail: {
         title: 'Failure',
-        description: 'Error training Lira'
+        description: 'Error training Le-IA'
       },
       success: {
         title: 'Success',
-        description: 'Lira is in the training process!'
+        description: 'Le-IA is in the training process!'
       }
     }
   }
@@ -554,7 +555,7 @@ var ptTranslation = {
       deleteFileTitle: 'Excluir um arquivo',
       deleteFile: 'Importante! Ao excluir este arquivo, você perderá todas as suas informações e não poderá recuperá-las.',
       trainingTitle: 'Treinamento de predefinição',
-      trainingDescription: 'Selecione um ou mais predefinições para serem treinadas. Obs: Caso não selecionada nenhuma a Lira irá treinar todas as predefinições'
+      trainingDescription: 'Selecione um ou mais predefinições para serem treinadas. Obs: Caso não selecionada nenhuma a Le-IA irá treinar todas as predefinições'
     }
   },
   buttons: {
@@ -569,6 +570,7 @@ var ptTranslation = {
   testArea: {
     description: 'Digite uma frase para testar a inteligência artificial.',
     typeSentence: 'Digite uma frase',
+    typePrompt: 'Insira aqui um prompt personalizado',
     selectPersona: 'Selecione uma Persona',
     selectPresset: 'Selecione uma predefinição',
     emptyState: {
@@ -600,17 +602,17 @@ var ptTranslation = {
     Test: {
       fail: {
         title: 'Falha',
-        description: 'Erro ao testar Lira'
+        description: 'Erro ao testar Le-IA'
       }
     },
     Train: {
       fail: {
         title: 'Falha',
-        description: 'Erro ao treinar a Lira'
+        description: 'Erro ao treinar a Le-IA'
       },
       success: {
         title: 'Sucesso',
-        description: 'A Lira está em processo de treinamento!'
+        description: 'A Le-IA está em processo de treinamento!'
       }
     }
   }
@@ -639,7 +641,7 @@ var esTranslation = {
       deleteFileTitle: 'Eliminar un archivo',
       deleteFile: '¡Importante! Al eliminar este archivo, perderá toda su información y no podrá recuperarla.',
       trainingTitle: 'Entrenamiento de preconfiguración',
-      trainingDescription: 'Seleccione una o más preconfiguracións para ser entrenadas. Nota: Si no se selecciona ninguna, Lira entrenará todas las preconfiguracións.'
+      trainingDescription: 'Seleccione una o más preconfiguracións para ser entrenadas. Nota: Si no se selecciona ninguna, Le-IA entrenará todas las preconfiguracións.'
     }
   },
   buttons: {
@@ -654,6 +656,7 @@ var esTranslation = {
   testArea: {
     description: 'Escribe una frase para probar la inteligencia artificial.',
     typeSentence: 'Escribe una oración',
+    typePrompt: 'Inserte un prompt personalizado aquí',
     selectPersona: 'Seleccionar una Persona',
     selectPresset: 'Seleccionar una preconfiguración',
     emptyState: {
@@ -685,17 +688,17 @@ var esTranslation = {
     Test: {
       fail: {
         title: 'Error',
-        description: 'Error al probar Lira'
+        description: 'Error al probar Le-IA'
       }
     },
     Train: {
       fail: {
         title: 'Error',
-        description: 'Error al entrenar Lira'
+        description: 'Error al entrenar Le-IA'
       },
       success: {
         title: 'Éxito',
-        description: '¡Lira está en proceso de entrenamiento!'
+        description: '¡Le-IA está en proceso de entrenamiento!'
       }
     }
   }
@@ -1338,12 +1341,13 @@ var Types = {
   NETWORK_OFFLINE: 'NETWORK_OFFLINE'
 };
 
-var getFilesAction = function getFilesAction(id, token) {
+var getFilesAction = function getFilesAction(id, token, env) {
   return {
     type: Types.GET_FILES_REQUEST,
     payload: {
       id: id,
-      token: token
+      token: token,
+      env: env
     }
   };
 };
@@ -1356,14 +1360,15 @@ var getFilesActionSuccess = function getFilesActionSuccess(_ref) {
     }
   };
 };
-var deleteFilesAction = function deleteFilesAction(name, id, language, token) {
+var deleteFilesAction = function deleteFilesAction(name, id, language, token, env) {
   return {
     type: Types.DELETE_FILES_REQUEST,
     payload: {
       name: name,
       id: id,
       language: language,
-      token: token
+      token: token,
+      env: env
     }
   };
 };
@@ -1376,7 +1381,7 @@ var deleteFilesActionSuccess = function deleteFilesActionSuccess(_ref2) {
     }
   };
 };
-var uploadFilesAction = function uploadFilesAction(file, id, pressets, language, token) {
+var uploadFilesAction = function uploadFilesAction(file, id, pressets, language, token, env) {
   return {
     type: Types.UPLOAD_FILES_REQUEST,
     payload: {
@@ -1384,7 +1389,8 @@ var uploadFilesAction = function uploadFilesAction(file, id, pressets, language,
       id: id,
       pressets: pressets,
       language: language,
-      token: token
+      token: token,
+      env: env
     }
   };
 };
@@ -1397,26 +1403,29 @@ var uploadFilesActionSuccess = function uploadFilesActionSuccess(_ref3) {
     }
   };
 };
-var trainingAction = function trainingAction(id, pressets, language, token) {
+var trainingAction = function trainingAction(id, pressets, language, token, env) {
   return {
     type: Types.TRAINING_REQUEST,
     payload: {
       id: id,
       pressets: pressets,
       language: language,
-      token: token
+      token: token,
+      env: env
     }
   };
 };
-var testAction = function testAction(question, profile, presset, files_directory, token) {
+var testAction = function testAction(question, prompt, profile, presset, files_directory, token, env) {
   return {
     type: Types.TEST_REQUEST,
     payload: {
       question: question,
+      prompt: prompt,
       profile: profile,
       presset: presset,
       files_directory: files_directory,
-      token: token
+      token: token,
+      env: env
     }
   };
 };
@@ -1445,7 +1454,8 @@ var FileArea = function FileArea() {
     id = _useSharedProps.id,
     language = _useSharedProps.language,
     propTags = _useSharedProps.propTags,
-    token = _useSharedProps.token;
+    token = _useSharedProps.token,
+    env = _useSharedProps.env;
   var _useState = useState([]),
     files = _useState[0],
     setFiles = _useState[1];
@@ -1483,7 +1493,7 @@ var FileArea = function FileArea() {
   });
   var tags = propTags ? propTags.tags : [];
   useEffect(function () {
-    dispatch(getFilesAction(id, token));
+    dispatch(getFilesAction(id, token, env));
   }, [id]);
   useEffect(function () {
     if (tags.length > 0) {
@@ -1548,17 +1558,17 @@ var FileArea = function FileArea() {
     setModalTraining(!modalTraining);
   };
   var deleteFile = function deleteFile() {
-    dispatch(deleteFilesAction(fileDelete, id, language, token));
+    dispatch(deleteFilesAction(fileDelete, id, language, token, env));
     setModalDelete(false);
   };
   var handleUploadFile = function handleUploadFile() {
-    dispatch(uploadFilesAction(uploadFile, id, presset, language, token));
+    dispatch(uploadFilesAction(uploadFile, id, presset, language, token, env));
     setModal(false);
     setUploadFile(undefined);
     setPresset([]);
   };
   var handleTrain = function handleTrain() {
-    dispatch(trainingAction(id, presset, language, token));
+    dispatch(trainingAction(id, presset, language, token, env));
     setModalTraining(false);
     setPresset([]);
   };
@@ -1741,6 +1751,7 @@ var FileArea = function FileArea() {
     id: id,
     language: language,
     token: token,
+    env: env,
     propTags: {
       tags: []
     },
@@ -1820,13 +1831,83 @@ var InputTest = function InputTest(_ref) {
   }));
 };
 
+var _templateObject$c, _templateObject2$8, _templateObject3$6, _templateObject4$3, _templateObject5$3, _templateObject6$2;
+var Container$6 = /*#__PURE__*/styled.div(_templateObject$c || (_templateObject$c = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  display: flex;\n  flex-flow: column;\n  margin-bottom: ", ";\n"])), function (_ref) {
+  var marginBottom = _ref.marginBottom;
+  return marginBottom;
+});
+var InputWrapper = /*#__PURE__*/styled.div(_templateObject2$8 || (_templateObject2$8 = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  display: flex;\n  align-items: center;\n  position: relative;\n\n  .icon {\n    position: absolute;\n    color: #5a5d68;\n\n    &.start-icon {\n      left: 12px;\n    }\n    &.end-icon {\n      right: 12px;\n    }\n  }\n\n  & .pass-view {\n    position: absolute;\n    right: 0;\n    top: 0;\n    bottom: 0;\n    padding: 14px;\n    font-size: 1rem;\n    color: #5a5d68;\n  }\n"])));
+var LabelWrapper = /*#__PURE__*/styled.label(_templateObject3$6 || (_templateObject3$6 = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  margin-bottom: 8px;\n\n  .label {\n    display: block;\n    color: #5a5d68;\n\n    font-style: normal;\n    font-weight: 600;\n    font-size: 0.875rem;\n    line-height: 114%;\n    letter-spacing: -0.02em;\n  }\n\n  .description {\n    font-style: normal;\n    font-weight: 500;\n    font-size: 0.625rem;\n    line-height: 120%;\n    color: #5a5d68;\n    display: block;\n  }\n"])));
+var CustomTextArea = /*#__PURE__*/styled.textarea(_templateObject4$3 || (_templateObject4$3 = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  font-family: 'Monospace', -apple-system, BlinkMacSystemFont, sans-serif;\n  font-size: 0.875rem;\n  font-style: normal;\n  font-weight: 500;\n  line-height: 20px;\n  text-align: left;\n  letter-spacing: -0.02em;\n  resize: vertical;\n  padding: 8px 12px;\n\n  border-radius: 4px;\n  border: 1px solid #979aa5;\n  color: #5a5d68 !important;\n  width: 100%;\n\n  background: #ffffff;\n  transition: background-color 0.2s ease-in-out, box-shadow 0.2s ease-in-out,\n    border 0.2s ease-in-out;\n\n  &:focus {\n    outline: none;\n    box-shadow: 0 0 0 3px #6690ff;\n  }\n\n  &:hover {\n    border: 1px solid #102693;\n  }\n  &:active {\n    border: 1px solid #102693;\n  }\n  &:disabled {\n    border: 1px solid #dadce3;\n    background: #f3f5f9;\n    color: #dadce3;\n  }\n\n  &.small {\n    min-height: 32px;\n  }\n\n  &.medium {\n    min-height: 40px;\n  }\n\n  &.large {\n    min-height: 48px;\n    font-size: 1rem;\n  }\n\n  &.success {\n    background: #e3f8cc;\n    border: 1px solid #19660a;\n    color: #19660a;\n  }\n\n  &.danger {\n    background: #fae0d2;\n    border: 1px solid #871821;\n    color: #871821;\n  }\n\n  &.error {\n    background: #fae0d2;\n    border: 1px solid #871821;\n    color: #871821;\n  }\n\n  &.has-start-icon {\n    padding-left: 40px;\n  }\n  &.has-end-icon {\n    padding-right: 40px;\n  }\n"])));
+var Helper = /*#__PURE__*/styled.div(_templateObject5$3 || (_templateObject5$3 = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  &.helper {\n    font-style: normal;\n    font-weight: 500;\n    font-size: 0.625rem;\n    line-height: 120%;\n    color: #5a5d68;\n    display: block;\n    margin-top: 4px;\n\n    &.success {\n      color: #19660a;\n    }\n\n    &.danger {\n      color: #871821;\n    }\n  }\n"])));
+var Error$1 = /*#__PURE__*/styled.div(_templateObject6$2 || (_templateObject6$2 = /*#__PURE__*/_taggedTemplateLiteralLoose(["\n  font-style: normal;\n  font-weight: 500;\n  font-size: 0.625rem;\n  line-height: 120%;\n  display: block;\n  margin-top: 4px;\n  color: #871821;\n"])));
+
+var _excluded$6 = ["id", "label", "description", "helper", "error", "placeholder", "className", "size", "color", "disabled", "startIcon", "endIcon", "marginBottom", "rows"];
+var TextArea = /*#__PURE__*/forwardRef(function (_ref, ref) {
+  var id = _ref.id,
+    label = _ref.label,
+    description = _ref.description,
+    helper = _ref.helper,
+    error = _ref.error,
+    placeholder = _ref.placeholder,
+    className = _ref.className,
+    _ref$size = _ref.size,
+    size = _ref$size === void 0 ? 'medium' : _ref$size,
+    _ref$color = _ref.color,
+    color = _ref$color === void 0 ? '' : _ref$color,
+    _ref$disabled = _ref.disabled,
+    disabled = _ref$disabled === void 0 ? false : _ref$disabled,
+    startIcon = _ref.startIcon,
+    endIcon = _ref.endIcon,
+    _ref$marginBottom = _ref.marginBottom,
+    marginBottom = _ref$marginBottom === void 0 ? '3px' : _ref$marginBottom,
+    _ref$rows = _ref.rows,
+    rows = _ref$rows === void 0 ? 4 : _ref$rows,
+    props = _objectWithoutPropertiesLoose(_ref, _excluded$6);
+  var selfId = id || "input__" + Math.random().toString(36).substr(2, 9);
+  return React.createElement(Container$6, {
+    className: className,
+    marginBottom: marginBottom
+  }, label && React.createElement(LabelWrapper, {
+    htmlFor: selfId
+  }, React.createElement("span", {
+    className: "label"
+  }, label), description && React.createElement("span", {
+    className: "description"
+  }, description)), React.createElement(InputWrapper, {
+    className: cc({
+      hasLabel: Boolean(label)
+    })
+  }, React.createElement(CustomTextArea, Object.assign({
+    className: cc(['input', size, color, {
+      error: Boolean(error),
+      'has-start-icon': Boolean(startIcon),
+      'has-end-icon': Boolean(endIcon)
+    }]),
+    id: selfId,
+    placeholder: placeholder,
+    disabled: disabled,
+    rows: rows,
+    ref: ref
+  }, props)), startIcon && React.createElement("span", {
+    className: cc(['icon', 'start-icon'])
+  }, startIcon), endIcon && React.createElement("span", {
+    className: cc(['icon', 'end-icon'])
+  }, endIcon)), helper && React.createElement(Helper, {
+    className: cc(['helper', color])
+  }, helper), error && React.createElement(Error$1, {
+    className: cc(['error'])
+  }, error));
+});
+
 var TestArea = function TestArea() {
   var _useSharedProps = useSharedProps(),
     id = _useSharedProps.id,
     language = _useSharedProps.language,
     personas = _useSharedProps.personas,
     propTags = _useSharedProps.propTags,
-    token = _useSharedProps.token;
+    token = _useSharedProps.token,
+    env = _useSharedProps.env;
   var t = getLanguage(language);
   var results = useSelector(function (state) {
     return state.message;
@@ -1838,11 +1919,14 @@ var TestArea = function TestArea() {
     search = _useState[0],
     setSearch = _useState[1];
   var _useState2 = useState(''),
-    profile = _useState2[0],
-    setProfile = _useState2[1];
+    prompt = _useState2[0],
+    setPrompt = _useState2[1];
   var _useState3 = useState(''),
-    presset = _useState3[0],
-    setPresset = _useState3[1];
+    profile = _useState3[0],
+    setProfile = _useState3[1];
+  var _useState4 = useState(''),
+    presset = _useState4[0],
+    setPresset = _useState4[1];
   var dispatch = useDispatch();
   var tags = propTags ? propTags.tags : [];
   var optionsPersona = personas && personas.map(function (persona) {
@@ -1866,8 +1950,11 @@ var TestArea = function TestArea() {
   var handleSearch = function handleSearch(value) {
     setSearch(value);
   };
+  var handlePrompt = function handlePrompt(value) {
+    setPrompt(value);
+  };
   var handleTest = function handleTest() {
-    dispatch(testAction(search, profile, presset, id, token));
+    dispatch(testAction(search, prompt, profile, presset, id, token, env));
   };
   var returnTest = function returnTest() {
     return React.createElement(Container$3, {
@@ -1883,6 +1970,14 @@ var TestArea = function TestArea() {
       placeholder: t.testArea.selectPersona,
       options: optionsPersona,
       onSelect: handleSelectPersona
+    }), profile === "Custom" && React.createElement(TextArea, {
+      className: "textarea-message",
+      label: "Prompt",
+      placeholder: t.testArea.typePrompt,
+      value: prompt,
+      onChange: function onChange(e) {
+        handlePrompt(e.target.value);
+      }
     }), React.createElement(CustomSelect, {
       placeholder: t.testArea.selectPresset,
       options: optionsPresset,
@@ -1904,6 +1999,7 @@ var TestArea = function TestArea() {
     id: id,
     token: token,
     language: language,
+    env: env,
     propTags: {
       tags: []
     },
@@ -1927,6 +2023,7 @@ var Leia = function Leia(props) {
     propTags: {
       tags: []
     },
+    env: '',
     personas: [{
       name: '',
       description: ''
@@ -2001,6 +2098,10 @@ function filesReducer(state, action) {
   });
 }
 
+var url$1 = 'http://localhost:5000';
+axios.defaults.baseURL = "" + url$1;
+var apiHml = /*#__PURE__*/axios.create();
+
 var CustomToastContent = function CustomToastContent(_ref) {
   var title = _ref.title,
     description = _ref.description;
@@ -2054,7 +2155,7 @@ var _marked = /*#__PURE__*/_regeneratorRuntime().mark(getFilesSaga),
   _marked5 = /*#__PURE__*/_regeneratorRuntime().mark(TrainingSaga),
   _marked6 = /*#__PURE__*/_regeneratorRuntime().mark(filesSagas);
 function getFilesSaga(action) {
-  var _action$payload, id, token, _yield$call, data;
+  var _action$payload, id, token, env, _yield$call, data;
   return _regeneratorRuntime().wrap(function getFilesSaga$(_context) {
     while (1) switch (_context.prev = _context.next) {
       case 0:
@@ -2062,9 +2163,9 @@ function getFilesSaga(action) {
         _context.next = 3;
         return put(commonLoadingStart());
       case 3:
-        _action$payload = action.payload, id = _action$payload.id, token = _action$payload.token;
+        _action$payload = action.payload, id = _action$payload.id, token = _action$payload.token, env = _action$payload.env;
         _context.next = 6;
-        return call(api.get, "/training/" + id + "?token=" + token);
+        return call(env.toLowerCase() === 'prod' ? api.get : apiHml.get, "/training/" + id + "?token=" + token);
       case 6:
         _yield$call = _context.sent;
         data = _yield$call.data;
@@ -2092,7 +2193,7 @@ function getFilesSaga(action) {
   }, _marked, null, [[0, 12, 15, 19]]);
 }
 function deleteFilesSaga(action) {
-  var t, _action$payload2, id, name, token, _yield$call2, data;
+  var t, _action$payload2, id, name, token, env, _yield$call2, data;
   return _regeneratorRuntime().wrap(function deleteFilesSaga$(_context2) {
     while (1) switch (_context2.prev = _context2.next) {
       case 0:
@@ -2101,12 +2202,12 @@ function deleteFilesSaga(action) {
         _context2.next = 4;
         return put(commonLoadingStart());
       case 4:
-        _action$payload2 = action.payload, id = _action$payload2.id, name = _action$payload2.name, token = _action$payload2.token;
+        _action$payload2 = action.payload, id = _action$payload2.id, name = _action$payload2.name, token = _action$payload2.token, env = _action$payload2.env;
         _context2.next = 7;
-        return call(api["delete"], "/delete/" + id + "/" + name + "?token=" + token);
+        return call(env.toLowerCase() === 'prod' ? api["delete"] : apiHml["delete"], "/delete/" + id + "/" + name + "?token=" + token);
       case 7:
         _context2.next = 9;
-        return call(api.get, "/training/" + id + "?token=" + token);
+        return call(env.toLowerCase() === 'prod' ? api.get : apiHml.get, "/training/" + id + "?token=" + token);
       case 9:
         _yield$call2 = _context2.sent;
         data = _yield$call2.data;
@@ -2139,7 +2240,7 @@ function deleteFilesSaga(action) {
   }, _marked2, null, [[1, 15, 19, 23]]);
 }
 function testSaga(action) {
-  var _action$payload3, question, profile, presset, files_directory, token, _yield$call3, data;
+  var _action$payload3, question, prompt, profile, presset, files_directory, token, env, _yield$call3, data;
   return _regeneratorRuntime().wrap(function testSaga$(_context3) {
     while (1) switch (_context3.prev = _context3.next) {
       case 0:
@@ -2147,10 +2248,11 @@ function testSaga(action) {
         _context3.next = 3;
         return put(commonLoadingStart());
       case 3:
-        _action$payload3 = action.payload, question = _action$payload3.question, profile = _action$payload3.profile, presset = _action$payload3.presset, files_directory = _action$payload3.files_directory, token = _action$payload3.token;
+        _action$payload3 = action.payload, question = _action$payload3.question, prompt = _action$payload3.prompt, profile = _action$payload3.profile, presset = _action$payload3.presset, files_directory = _action$payload3.files_directory, token = _action$payload3.token, env = _action$payload3.env;
         _context3.next = 6;
-        return call(api.post, "/ask", {
+        return call(env.toLowerCase() === 'prod' ? api.post : apiHml.post, "/ask", {
           question: question,
+          description: prompt,
           profile: profile,
           tag: presset,
           files_directory: files_directory,
@@ -2183,7 +2285,7 @@ function testSaga(action) {
   }, _marked3, null, [[0, 12, 15, 19]]);
 }
 function uploadFilesSaga(action) {
-  var t, _action$payload4, id, file, pressets, token, blob, formData, sanitizedPath, queryParams, url, _yield$call4, data;
+  var t, _action$payload4, id, file, pressets, token, env, blob, formData, sanitizedPath, queryParams, url, _yield$call4, data;
   return _regeneratorRuntime().wrap(function uploadFilesSaga$(_context4) {
     while (1) switch (_context4.prev = _context4.next) {
       case 0:
@@ -2192,24 +2294,24 @@ function uploadFilesSaga(action) {
         _context4.next = 4;
         return put(commonLoadingStart());
       case 4:
-        _action$payload4 = action.payload, id = _action$payload4.id, file = _action$payload4.file, pressets = _action$payload4.pressets, token = _action$payload4.token;
+        _action$payload4 = action.payload, id = _action$payload4.id, file = _action$payload4.file, pressets = _action$payload4.pressets, token = _action$payload4.token, env = _action$payload4.env;
         blob = new Blob([file.content]);
         formData = new FormData();
-        sanitizedPath = file.properties.path.replace(/^(\.\/|\/+)|\/+$/g, '');
+        sanitizedPath = file.properties.path.replace(/^\/+|\/+$/g, '');
         formData.append('file', blob, sanitizedPath || 'default_filename');
         queryParams = pressets.length ? new URLSearchParams({
           tags: pressets.join(',')
         }).toString() : '';
         url = queryParams ? "/upload/" + id + "?token=" + token + "&" + queryParams : "/upload/" + id + "?token=" + token;
         _context4.next = 13;
-        return call(api.post, url, formData, {
+        return call(env.toLowerCase() === 'prod' ? api.post : apiHml.post, url, formData, {
           headers: {
             'Content-Type': 'multipart/form-data'
           }
         });
       case 13:
         _context4.next = 15;
-        return call(api.get, "/training/" + id + "?token=" + token);
+        return call(env.toLowerCase() === 'prod' ? api.get : apiHml.get, "/training/" + id + "?token=" + token);
       case 15:
         _yield$call4 = _context4.sent;
         data = _yield$call4.data;
@@ -2241,7 +2343,7 @@ function uploadFilesSaga(action) {
   }, _marked4, null, [[1, 21, 25, 29]]);
 }
 function TrainingSaga(action) {
-  var t, _action$payload5, id, pressets, token, tag, queryParams, _yield$call5, data;
+  var t, _action$payload5, id, pressets, token, env, tag, queryParams, _yield$call5, data;
   return _regeneratorRuntime().wrap(function TrainingSaga$(_context5) {
     while (1) switch (_context5.prev = _context5.next) {
       case 0:
@@ -2250,19 +2352,19 @@ function TrainingSaga(action) {
         _context5.next = 4;
         return put(commonLoadingStart());
       case 4:
-        _action$payload5 = action.payload, id = _action$payload5.id, pressets = _action$payload5.pressets, token = _action$payload5.token;
+        _action$payload5 = action.payload, id = _action$payload5.id, pressets = _action$payload5.pressets, token = _action$payload5.token, env = _action$payload5.env;
         tag = pressets.join(',');
         queryParams = new URLSearchParams({
           tag: tag
         }).toString();
         _context5.next = 9;
-        return call(api.post, pressets.length > 0 ? "/training?" + queryParams : "/training", {
+        return call(env.toLowerCase() === 'prod' ? api.post : apiHml.post, pressets.length > 0 ? "/training?" + queryParams : "/training", {
           files_directory: id,
           token: token
         });
       case 9:
         _context5.next = 11;
-        return call(api.get, "/training/" + id + "?token=" + token);
+        return call(env.toLowerCase() === 'prod' ? api.get : apiHml.get, "/training/" + id + "?token=" + token);
       case 11:
         _yield$call5 = _context5.sent;
         data = _yield$call5.data;