Enhancement default model selection

frontend/src/components/Content.tsx

@@ -13,7 +13,6 @@ import {
   batchSize,
   buttonCaptions,
   chatModeLables,
-  defaultLLM,
   largeFileSize,
   llms,
   RETRY_OPIONS,
@@ -90,6 +89,7 @@ const Content: React.FC<ContentProps> = ({
     processedCount,
     setProcessedCount,
     setchatModes,
+    model
   } = useFileContext();
   const [viewPoint, setViewPoint] = useState<'tableView' | 'showGraphView' | 'chatInfoView' | 'neighborView'>(
     'tableView'
@@ -537,6 +537,7 @@ const Content: React.FC<ContentProps> = ({
     setProcessedCount(0);
     setConnectionStatus(false);
     localStorage.removeItem('password');
+    localStorage.removeItem('selectedModel')
     setUserCredentials({ uri: '', password: '', userName: '', database: '' });
     setSelectedNodes([]);
     setSelectedRels([]);
@@ -881,7 +882,7 @@ const Content: React.FC<ContentProps> = ({
             onSelect={handleDropdownChange}
             options={llms ?? ['']}
             placeholder='Select LLM Model'
-            defaultValue={defaultLLM}
+            defaultValue={model}
             view='ContentView'
             isDisabled={false}
           />

frontend/src/components/Dropdown.tsx

@@ -17,6 +17,10 @@ const DropdownComponent: React.FC<ReusableDropdownProps> = ({
   const isLargeDesktop = useMediaQuery(`(min-width:1440px )`);
   const handleChange = (selectedOption: OptionType | null | void) => {
     onSelect(selectedOption);
+    const existingModel = localStorage.getItem('selectedModel');
+    if (existingModel != selectedOption?.value) {
+      localStorage.setItem('selectedModel', selectedOption?.value ?? '');
+    }
   };
   const allOptions = useMemo(() => options, [options]);
   return (

frontend/src/context/UsersFiles.tsx

@@ -6,7 +6,7 @@ import {
   OptionType,
   showTextFromSchemaDialogType,
 } from '../types';
-import { chatModeLables, defaultLLM, getStoredSchema } from '../utils/Constants';
+import { chatModeLables, getStoredSchema, llms } from '../utils/Constants';
 import { useCredentials } from './UserCredentials';
 import Queue from '../utils/Queue';
 
@@ -16,13 +16,14 @@ const FileContextProvider: FC<FileContextProviderProps> = ({ children }) => {
   const selectedNodeLabelstr = localStorage.getItem('selectedNodeLabels');
   const selectedNodeRelsstr = localStorage.getItem('selectedRelationshipLabels');
   const persistedQueue = localStorage.getItem('waitingQueue');
+  const selectedModel = localStorage.getItem('selectedModel');
   const { userCredentials } = useCredentials();
   const [files, setFiles] = useState<(File | null)[] | []>([]);
   const [filesData, setFilesData] = useState<CustomFile[] | []>([]);
   const [queue, setQueue] = useState<Queue>(
     new Queue(JSON.parse(persistedQueue ?? JSON.stringify({ queue: [] })).queue)
   );
-  const [model, setModel] = useState<string>(defaultLLM);
+  const [model, setModel] = useState<string>(selectedModel ?? llms[0]);
   const [graphType, setGraphType] = useState<string>('Knowledge Graph Entities');
   const [selectedNodes, setSelectedNodes] = useState<readonly OptionType[]>([]);
   const [selectedRels, setSelectedRels] = useState<readonly OptionType[]>([]);

frontend/src/utils/Constants.ts

@@ -28,11 +28,6 @@ export const llms =
         'bedrock_claude_3_5_sonnet',
       ];
 
-export const defaultLLM = llms?.includes('openai_gpt_4o')
-  ? 'openai_gpt_4o'
-  : llms?.includes('gemini_1.5_pro')
-  ? 'gemini_1.5_pro'
-  : 'diffbot';
 export const supportedLLmsForRagas = [
   'openai_gpt_3.5',
   'openai_gpt_4',