added new models

neo4j-labs · Jan 15, 2025 · 31d2a10 · 31d2a10
1 parent b618a36
commit 31d2a10
Showing 1 changed file with 52 additions and 49 deletions.
diff --git a/frontend/src/utils/Constants.ts b/frontend/src/utils/Constants.ts
@@ -13,22 +13,25 @@ export const llms =
   process.env?.VITE_LLM_MODELS?.trim() != ''
     ? (process.env.VITE_LLM_MODELS?.split(',') as string[])
     : [
-        'openai_gpt_3.5',
-        'openai_gpt_4o',
-        'openai_gpt_4o_mini',
-        'gemini_1.5_pro',
-        'gemini_1.5_flash',
-        'gemini_2.0_flash',
-        'diffbot',
-        'azure_ai_gpt_35',
-        'azure_ai_gpt_4o',
-        'ollama_llama3',
-        'groq_llama3_70b',
-        'anthropic_claude_3_5_sonnet',
-        'fireworks_llama_v3p2_90b',
-        'fireworks_qwen72b_instruct',
-        'bedrock_claude_3_5_sonnet',
-      ];
+      'openai_gpt_3.5',
+      'openai_gpt_4o',
+      'openai_gpt_4o_mini',
+      'gemini_1.5_pro',
+      'gemini_1.5_flash',
+      'gemini_2.0_flash',
+      'diffbot',
+      'azure_ai_gpt_35',
+      'azure_ai_gpt_4o',
+      'ollama_llama3',
+      'groq_llama3_70b',
+      'anthropic_claude_3_5_sonnet',
+      'fireworks_llama_v3p2_90b',
+      'fireworks_qwen72b_instruct',
+      'bedrock_claude_3_5_sonnet',
+      'bedrock_nova_micro_v1',
+      'bedrock_nova_lite_v1',
+      'bedrock_nova_pro_v1'
+    ];
 
 export const supportedLLmsForRagas = [
   'openai_gpt_3.5',
@@ -86,40 +89,40 @@ export const chatModeReadableLables: Record<string, string> = {
 export const chatModes =
   process.env?.VITE_CHAT_MODES?.trim() != ''
     ? process.env.VITE_CHAT_MODES?.split(',').map((mode) => ({
-        mode: mode.trim(),
-        description: getDescriptionForChatMode(mode.trim()),
-      }))
+      mode: mode.trim(),
+      description: getDescriptionForChatMode(mode.trim()),
+    }))
     : [
-        {
-          mode: chatModeLables.vector,
-          description: 'Performs semantic similarity search on text chunks using vector indexing.',
-        },
-        {
-          mode: chatModeLables.graph,
-          description: 'Translates text to Cypher queries for precise data retrieval from a graph database.',
-        },
-        {
-          mode: chatModeLables['graph+vector'],
-          description: 'Combines vector indexing and graph connections for contextually enhanced semantic search.',
-        },
-        {
-          mode: chatModeLables.fulltext,
-          description: 'Conducts fast, keyword-based search using full-text indexing on text chunks.',
-        },
-        {
-          mode: chatModeLables['graph+vector+fulltext'],
-          description: 'Integrates vector, graph, and full-text indexing for comprehensive search results.',
-        },
-        {
-          mode: chatModeLables['entity search+vector'],
-          description: 'Uses vector indexing on entity nodes for highly relevant entity-based search.',
-        },
-        {
-          mode: chatModeLables['global search+vector+fulltext'],
-          description:
-            'Use vector and full-text indexing on community nodes to provide accurate, context-aware answers globally.',
-        },
-      ];
+      {
+        mode: chatModeLables.vector,
+        description: 'Performs semantic similarity search on text chunks using vector indexing.',
+      },
+      {
+        mode: chatModeLables.graph,
+        description: 'Translates text to Cypher queries for precise data retrieval from a graph database.',
+      },
+      {
+        mode: chatModeLables['graph+vector'],
+        description: 'Combines vector indexing and graph connections for contextually enhanced semantic search.',
+      },
+      {
+        mode: chatModeLables.fulltext,
+        description: 'Conducts fast, keyword-based search using full-text indexing on text chunks.',
+      },
+      {
+        mode: chatModeLables['graph+vector+fulltext'],
+        description: 'Integrates vector, graph, and full-text indexing for comprehensive search results.',
+      },
+      {
+        mode: chatModeLables['entity search+vector'],
+        description: 'Uses vector indexing on entity nodes for highly relevant entity-based search.',
+      },
+      {
+        mode: chatModeLables['global search+vector+fulltext'],
+        description:
+          'Use vector and full-text indexing on community nodes to provide accurate, context-aware answers globally.',
+      },
+    ];
 
 export const chunkSize = process.env.VITE_CHUNK_SIZE ? parseInt(process.env.VITE_CHUNK_SIZE) : 1 * 1024 * 1024;
 export const timeperpage = process.env.VITE_TIME_PER_PAGE ? parseInt(process.env.VITE_TIME_PER_PAGE) : 50;