MariaFilipkowska commited on
Commit
a6ff346
·
verified ·
1 Parent(s): f19bbd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -32,12 +32,12 @@ chat3 = ChatMistralAI(
32
  max_retries=6,
33
  mistral_api_key = os.getenv("Mistral")
34
  )
35
- # chat2 = ChatMistralAI(
36
- # model="mistral-ocr-2505",
37
- # temperature=0.01,
38
- # max_retries=6,
39
- # mistral_api_key = os.getenv("Mistral")
40
- #)
41
  print(f"Model {chat3.model} downloaded!")
42
  chat1 = ChatMistralAI(
43
  model="mistral-medium-2505",
@@ -208,7 +208,7 @@ class BasicAgent:
208
  wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
209
  while not answer or answer in wrong_answers or answer.lower().startswith("error"):
210
  try:
211
- alfred = build_agent(chat3)
212
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
  except:
214
  alfred = build_agent(chat1)
 
32
  max_retries=6,
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")
40
+ )
41
  print(f"Model {chat3.model} downloaded!")
42
  chat1 = ChatMistralAI(
43
  model="mistral-medium-2505",
 
208
  wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
209
  while not answer or answer in wrong_answers or answer.lower().startswith("error"):
210
  try:
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)