Updated CBORG models

Knowledge-Graph-Hub · Aug 22, 2024 · 785411e · 785411e
1 parent 2fb75cd
commit 785411e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/kg_chat/utils.py b/src/kg_chat/utils.py
@@ -232,10 +232,9 @@ def get_lbl_cborg_models():
     """Get the list of LBNL-hosted models via CBORG."""
     return [
         "lbl/llama-3",  # LBNL-hosted model (free to use)
-        "openai/gpt-4o-mini",  # OpenAI-hosted model
-        "anthropic/claude-haiku",  # Anthropic-hosted model
-        "anthropic/claude-sonnet",  # Anthropic-hosted model
-        "anthropic/claude-opus",  # Anthropic-hosted model
+        "openai/chatgpt:latest",  # OpenAI-hosted model
+        "anthropic/claude:latest",  # Anthropic-hosted model
+        "google/gemini:latest",  # Google-hosted model
     ]