update models

agent.py

@@ -35,8 +35,8 @@ class ModelConfig:
 
 
 model_configurations = {
-    "gpt-4o-mini": ModelConfig(
-        model_name="gpt-4o-mini", api_key=st.secrets["OPENAI_API_KEY"], base_url=f"https://gateway.ai.cloudflare.com/v1/{st.secrets['CLOUDFLARE_ACCOUNT_ID']}/snowchat/openai"
+    "gpt-4o": ModelConfig(
+        model_name="gpt-4o", api_key=st.secrets["OPENAI_API_KEY"], base_url=f"https://gateway.ai.cloudflare.com/v1/{st.secrets['CLOUDFLARE_ACCOUNT_ID']}/snowchat/openai"
     ),
     "Gemini Pro 1.5": ModelConfig(
         model_name="google/gemini-pro-1.5",
@@ -46,15 +46,15 @@ class ModelConfig:
     # "Mistral 7B": ModelConfig(
     #     model_name="mistralai/mistral-7b-v0.1", api_key=st.secrets["REPLICATE_API_TOKEN"]
     # ),
-    "llama-3.2-3b": ModelConfig(
-        model_name="accounts/fireworks/models/llama-v3p2-3b-instruct",
+    "llama-3.3-70b": ModelConfig(
+        model_name="accounts/fireworks/models/llama-v3p3-70b-instruct",
         api_key=st.secrets["FIREWORKS_API_KEY"],
         base_url="https://api.fireworks.ai/inference/v1",
     ),
-    "llama-3.1-405b": ModelConfig(
-        model_name="accounts/fireworks/models/llama-v3p1-405b-instruct",
-        api_key=st.secrets["FIREWORKS_API_KEY"],
-        base_url="https://api.fireworks.ai/inference/v1",
+    "Gemini Flash 1.5": ModelConfig(
+        model_name="google/gemini-flash-1.5",
+        api_key=st.secrets["OPENROUTER_API_KEY"],
+        base_url="https://openrouter.ai/api/v1",
     ),
 }
 sys_msg = SystemMessage(

main.py

@@ -35,10 +35,10 @@
 st.caption("Talk your way through data")
 
 model_options = {
-    "gpt-4o-mini": "GPT-4o Mini",
-    "llama-3.1-405b": "Llama 3.1 405B",
-    "llama-3.2-3b": "Llama 3.2 3B",
+    "gpt-4o": "GPT-4o",
+    "llama-3.3-70b": "Llama 3.3 70B",
     "Gemini Pro 1.5": "Gemini Pro 1.5",
+    "Gemini Flash 1.5": "Gemini Flash 1.5",
 }
 
 model = st.radio(