Add support for non pro models (#2285)

2024-12-20 01:43:38 +05:30
parent 7d5be32a59
commit c697bee9e7
1 changed files with 2 additions and 1 deletions
--- a/letta/providers.py
+++ b/letta/providers.py
@@ -482,7 +482,8 @@ class GoogleAIProvider(Provider):
        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: