Add support for non pro models (#2285)

letta-ai · Dec 19, 2024 · a336092 · a336092
1 parent 87820d9
commit a336092
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/letta/providers.py b/letta/providers.py
@@ -482,7 +482,8 @@ def list_llm_models(self):
         model_options = [mo[len("models/") :] if mo.startswith("models/") else mo for mo in model_options]
 
         # TODO remove manual filtering for gemini-pro
-        model_options = [mo for mo in model_options if str(mo).startswith("gemini") and "-pro" in str(mo)]
+        # Add support for all gemini models
+        model_options = [mo for mo in model_options if str(mo).startswith("gemini-")]
 
         configs = []
         for model in model_options: