MariaFilipkowska commited on
Commit
84742aa
·
verified ·
1 Parent(s): 9b8e04b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ chat3 = ChatMistralAI(
33
  mistral_api_key = os.getenv("Mistral")
34
  )
35
  chat2 = ChatMistralAI(
36
- model="pixtral-12b-2409",
37
  temperature=0.01,
38
  max_retries=6,
39
  mistral_api_key = os.getenv("Mistral")
@@ -211,7 +211,7 @@ class BasicAgent:
211
  alfred = build_agent(chat2)
212
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
  except:
214
- alfred = build_agent(chat3)
215
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
216
 
217
  if answer:
 
33
  mistral_api_key = os.getenv("Mistral")
34
  )
35
  chat2 = ChatMistralAI(
36
+ model="pixtral-large-2411",
37
  temperature=0.01,
38
  max_retries=6,
39
  mistral_api_key = os.getenv("Mistral")
 
211
  alfred = build_agent(chat2)
212
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
  except:
214
+ alfred = build_agent(chat1)
215
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
216
 
217
  if answer: