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