Aasher commited on
Commit
82793b3
·
1 Parent(s): 91cf9c0

Remove unavailable models

Browse files
Files changed (1) hide show
  1. src/axiom/app.py +1 -2
src/axiom/app.py CHANGED
@@ -106,7 +106,6 @@ async def on_chat_start():
106
  values=[
107
  "gemini-2.0-flash",
108
  "gemini-2.0-flash-lite",
109
- "gemini-2.0-pro-exp-02-05",
110
  "gemini-exp-1206",
111
  "gemini-2.5-pro-exp-03-25"
112
  ],
@@ -142,7 +141,7 @@ async def on_message(message: cl.Message):
142
  config = {"configurable": {"thread_id": thread_id}}
143
 
144
  # Get model & chat profile from session
145
- model = cl.user_session.get("model", "gemini-2.0-flash")
146
  answer_mode = cl.user_session.get("chat_profile", "Agent Mode")
147
 
148
  # Prepare the content list for the current message
 
106
  values=[
107
  "gemini-2.0-flash",
108
  "gemini-2.0-flash-lite",
 
109
  "gemini-exp-1206",
110
  "gemini-2.5-pro-exp-03-25"
111
  ],
 
141
  config = {"configurable": {"thread_id": thread_id}}
142
 
143
  # Get model & chat profile from session
144
+ model = cl.user_session.get("model")
145
  answer_mode = cl.user_session.get("chat_profile", "Agent Mode")
146
 
147
  # Prepare the content list for the current message