MariaFilipkowska commited on
Commit
9e30913
·
verified ·
1 Parent(s): 3ab47c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -91,7 +91,7 @@ extract_text_tool = Tool(
91
  chat = ChatOllama(model=model_id, verbose=True)
92
  print(f"Model {chat.model} downloaded!")
93
 
94
- tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool]
95
  chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
96
 
97
 
@@ -150,7 +150,9 @@ class BasicAgent:
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('[', '')
 
91
  chat = ChatOllama(model=model_id, verbose=True)
92
  print(f"Model {chat.model} downloaded!")
93
 
94
+ tools = [get_file_path_tool, get_content_tool]
95
  chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
96
 
97
 
 
150
  def __call__(self, question: str) -> str:
151
  print(f"Agent received question (first 50 chars): {question[:50]}...")
152
  message = HumanMessage(content=question)
153
+ print(message)
154
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
155
+ print(answer)
156
  answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
157
  answer = answer.replace('FINAL ANSWER: ', '')
158
  answer = answer.replace('[', '')