diff --git a/ui/src/components/modals/add-new-model/index.jsx b/ui/src/components/modals/add-new-model/index.jsx
index cecea5c7..4e0109d7 100644
--- a/ui/src/components/modals/add-new-model/index.jsx
+++ b/ui/src/components/modals/add-new-model/index.jsx
@@ -3,7 +3,6 @@ import {
 } from '@radicalbit/radicalbit-design-system';
 import useModals from '@Src/hooks/use-modals';
 import { ModelTypeEnum } from '@Src/store/state/models/constants';
-import { LlmActionButton, LlmBody, LlmHeader } from './llm-components';
 import ModalContextProvider, { useModalContext } from './modal-context-provider';
 import ActionsStepFourth from './step-four/actions';
 import BodyStepFour from './step-four/new-body';
@@ -13,6 +12,10 @@ import ActionsStepThree from './step-three/actions';
 import BodyStepThree from './step-three/body';
 import ActionsStepTwo from './step-two/actions';
 import BodyStepTwo from './step-two/body';
+import {
+  TextGenerationActionButton, TextGenerationBody,
+  TextGenerationHeader,
+} from './text-generation';
 
 const { Step } = Steps;
 
@@ -77,7 +80,7 @@ function Header() {
 
   switch (modelType) {
     case ModelTypeEnum.TEXT_GENERATION:
-      return (<LlmHeader />);
+      return (<TextGenerationHeader />);
 
     default:
       return (
@@ -165,7 +168,7 @@ function Body() {
 
   switch (modelType) {
     case ModelTypeEnum.TEXT_GENERATION:
-      return <LlmBody />;
+      return <TextGenerationBody />;
 
     default:
       return <BodyInner />;
@@ -199,7 +202,7 @@ function Actions() {
 
   switch (modelType) {
     case ModelTypeEnum.TEXT_GENERATION:
-      return (<LlmActionButton />);
+      return (<TextGenerationActionButton />);
 
     default:
       return <ActionsInner />;
diff --git a/ui/src/components/modals/add-new-model/llm-components/form-fields.jsx b/ui/src/components/modals/add-new-model/text-generation/form-fields.jsx
similarity index 100%
rename from ui/src/components/modals/add-new-model/llm-components/form-fields.jsx
rename to ui/src/components/modals/add-new-model/text-generation/form-fields.jsx
diff --git a/ui/src/components/modals/add-new-model/llm-components/index.jsx b/ui/src/components/modals/add-new-model/text-generation/index.jsx
similarity index 86%
rename from ui/src/components/modals/add-new-model/llm-components/index.jsx
rename to ui/src/components/modals/add-new-model/text-generation/index.jsx
index 748ef9ba..4fd95b9b 100644
--- a/ui/src/components/modals/add-new-model/llm-components/index.jsx
+++ b/ui/src/components/modals/add-new-model/text-generation/index.jsx
@@ -5,11 +5,11 @@ import {
 import { useModalContext } from '../modal-context-provider';
 import useHandleOnSubmit from './use-handle-on-submit';
 
-function LlmHeader() {
+function TextGenerationHeader() {
   return <SectionTitle title="New Model" />;
 }
 
-function LlmBody() {
+function TextGenerationBody() {
   const { useFormbit } = useModalContext();
   const { error } = useFormbit;
 
@@ -37,7 +37,7 @@ function LlmBody() {
   );
 }
 
-function LlmActionButton() {
+function TextGenerationActionButton() {
   const { handleOnSubmit, args, isSubmitDisabled } = useHandleOnSubmit();
 
   return (
@@ -56,4 +56,4 @@ function LlmActionButton() {
   );
 }
 
-export { LlmHeader, LlmBody, LlmActionButton };
+export { TextGenerationHeader, TextGenerationBody, TextGenerationActionButton };
diff --git a/ui/src/components/modals/add-new-model/llm-components/use-handle-on-submit.js b/ui/src/components/modals/add-new-model/text-generation/use-handle-on-submit.js
similarity index 100%
rename from ui/src/components/modals/add-new-model/llm-components/use-handle-on-submit.js
rename to ui/src/components/modals/add-new-model/text-generation/use-handle-on-submit.js