Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import base64
|
|
| 6 |
|
| 7 |
from dotenv import load_dotenv
|
| 8 |
from typing import TypedDict, Annotated
|
| 9 |
-
from langchain_core.messages import AnyMessage, SystemMessage
|
| 10 |
from langchain_community.tools import DuckDuckGoSearchRun
|
| 11 |
from langchain_ollama.chat_models import ChatOllama
|
| 12 |
from langchain.tools import Tool
|
|
@@ -149,7 +149,8 @@ class BasicAgent:
|
|
| 149 |
|
| 150 |
def __call__(self, question: str) -> str:
|
| 151 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 152 |
-
|
|
|
|
| 153 |
answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
|
| 154 |
answer = answer.replace('FINAL ANSWER: ', '')
|
| 155 |
answer = answer.replace('[', '')
|
|
|
|
| 6 |
|
| 7 |
from dotenv import load_dotenv
|
| 8 |
from typing import TypedDict, Annotated
|
| 9 |
+
from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
|
| 10 |
from langchain_community.tools import DuckDuckGoSearchRun
|
| 11 |
from langchain_ollama.chat_models import ChatOllama
|
| 12 |
from langchain.tools import Tool
|
|
|
|
| 149 |
|
| 150 |
def __call__(self, question: str) -> str:
|
| 151 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 152 |
+
message = HumanMessage(content=question)
|
| 153 |
+
answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
|
| 154 |
answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
|
| 155 |
answer = answer.replace('FINAL ANSWER: ', '')
|
| 156 |
answer = answer.replace('[', '')
|