MariaFilipkowska commited on
Commit
b31dee3
·
verified ·
1 Parent(s): a2e98e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ print("Loading LLM...")
29
  chat = ChatMistralAI(
30
  model="mistral-medium-2505",
31
  temperature=0.0,
32
- max_retries=1,
33
  mistral_api_key = os.getenv("Mistral")
34
  )
35
  print(f"Model {chat.model} downloaded!")
@@ -187,7 +187,7 @@ class BasicAgent:
187
  print(f"Agent received question (first 50 chars): {question[:50]}...")
188
  message = HumanMessage(content=question)
189
  print(message)
190
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 1})['messages'][-1].content
191
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
192
  answer = answer.replace('FINAL ANSWER:', '')
193
  answer = answer.replace('FINAL ANSWER', '')
 
29
  chat = ChatMistralAI(
30
  model="mistral-medium-2505",
31
  temperature=0.0,
32
+ max_retries=6,
33
  mistral_api_key = os.getenv("Mistral")
34
  )
35
  print(f"Model {chat.model} downloaded!")
 
187
  print(f"Agent received question (first 50 chars): {question[:50]}...")
188
  message = HumanMessage(content=question)
189
  print(message)
190
+ answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
191
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
192
  answer = answer.replace('FINAL ANSWER:', '')
193
  answer = answer.replace('FINAL ANSWER', '')