feat(models): add new Gemini and Claude model variants to QueryInput …

…and VectorSearch components
HUBioDataLab · Feb 27, 2025 · 7217da2 · 7217da2
1 parent ed85e46
commit 7217da2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/crossbar_llm/backend/tools/langchain_llm_qa_trial.py b/crossbar_llm/backend/tools/langchain_llm_qa_trial.py
@@ -410,6 +410,10 @@ def __init__(
     def define_llm(self, model_name):
 
         google_llm_models = [
+            "gemini-2.0-flash-thinking-exp-01-21",
+            "gemini-2.0-pro-exp-02-05",
+            "gemini-2.0-flash",
+            "gemini-2.0-flash-lite",
             "gemini-pro",
             "gemini-1.5-pro-latest",
             "gemini-1.5-flash-latest",
@@ -433,7 +437,8 @@ def define_llm(self, model_name):
             "claude-3-opus-20240229",
             "claude-3-sonnet-20240229",
             "claude-3-haiku-20240307",
-            "claude-3-5-sonnet-20240620",
+            "claude-3-5-sonnet-latest",
+            "claude-3-7-sonnet-latest",
             "claude-2.1",
             "claude-2.0",
             "claude-instant-1.2",

diff --git a/crossbar_llm/frontend/src/components/QueryInput.js b/crossbar_llm/frontend/src/components/QueryInput.js
@@ -59,6 +59,8 @@ function QueryInput({
       'gpt-3.5-turbo-16k',
     ],
     Anthropic: [
+      'claude-3-5-sonnet-latest',
+      'claude-3-7-sonnet-latest',
       'claude-3-5-sonnet-20240620',
       'claude-3-opus-20240229',
       'claude-3-sonnet-20240229',
@@ -76,6 +78,10 @@ function QueryInput({
       "deepseek/deepseek-chat",
     ], 
     Google: [
+      'gemini-2.0-flash-thinking-exp-01-21',
+      'gemini-2.0-pro-exp-02-05',
+      'gemini-2.0-flash',
+      'gemini-2.0-flash-lite',
       'gemini-pro',
       'gemini-1.5-pro-latest',
       'gemini-1.5-flash-latest',

diff --git a/crossbar_llm/frontend/src/components/VectorSearch.js b/crossbar_llm/frontend/src/components/VectorSearch.js
@@ -60,6 +60,8 @@ function VectorSearch({
       'gpt-3.5-turbo-16k',
     ],
     Anthropic: [
+      'claude-3-5-sonnet-latest',
+      'claude-3-7-sonnet-latest',
       'claude-3-5-sonnet-20240620',
       'claude-3-opus-20240229',
       'claude-3-sonnet-20240229',
@@ -77,6 +79,10 @@ function VectorSearch({
       "deepseek/deepseek-chat",
     ], 
     Google: [
+      'gemini-2.0-flash-thinking-exp-01-21',
+      'gemini-2.0-pro-exp-02-05',
+      'gemini-2.0-flash',
+      'gemini-2.0-flash-lite',
       'gemini-pro',
       'gemini-1.5-pro-latest',
       'gemini-1.5-flash-latest',