MariaFilipkowska commited on
Commit
3ab47c7
·
verified ·
1 Parent(s): 5a225f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- answer = alfred.invoke({"messages": [system_prompt, question]},{"recursion_limit": 3})['messages'][-1].content
 
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('[', '')