Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 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":
|
| 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', '')
|