update models

kaarthik108 · Dec 25, 2024 · 7c9bdd6 · 7c9bdd6
1 parent 995829a
commit 7c9bdd6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/agent.py b/agent.py
@@ -38,23 +38,23 @@ class ModelConfig:
     "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",
-        api_key=st.secrets["OPENROUTER_API_KEY"],
-        base_url="https://openrouter.ai/api/v1",
+    "Gemini Flash 1.5": ModelConfig(
+        model_name="gemini-1.5-flash",
+        api_key=st.secrets["GEMINI_API_KEY"],
+        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
     ),
     # "Mistral 7B": ModelConfig(
     #     model_name="mistralai/mistral-7b-v0.1", api_key=st.secrets["REPLICATE_API_TOKEN"]
     # ),
-    "llama-3.3-70b": ModelConfig(
-        model_name="accounts/fireworks/models/llama-v3p3-70b-instruct",
+    "Qwen 2.5": ModelConfig(
+        model_name="accounts/fireworks/models/qwen2p5-coder-32b-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",
+    "Gemini Exp 1206": ModelConfig(
+        model_name="gemini-exp-1206",
+        api_key=st.secrets["GEMINI_API_KEY"],
+        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
     ),
 }
 sys_msg = SystemMessage(

diff --git a/main.py b/main.py
@@ -36,8 +36,8 @@
 
 model_options = {
     "gpt-4o": "GPT-4o",
-    "llama-3.3-70b": "Llama 3.3 70B",
-    "Gemini Pro 1.5": "Gemini Pro 1.5",
+    "Qwen 2.5": "Qwen 2.5",
+    "Gemini Exp 1206": "Gemini Exp 1206",
     "Gemini Flash 1.5": "Gemini Flash 1.5",
 }