From c697bee9e76213a4ac462805f44ea215991fd6fa Mon Sep 17 00:00:00 2001
From: "S.Go" <43789832+SwastikGorai@users.noreply.github.com>
Date: Fri, 20 Dec 2024 01:43:38 +0530
Subject: [PATCH] Add support for non pro models (#2285)

---
 letta/providers.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/letta/providers.py b/letta/providers.py
index 5721db46..83069a76 100644
--- 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: