Update agent.py
Browse files
agent.py
CHANGED
|
@@ -70,9 +70,9 @@ class BasicAgent():
|
|
| 70 |
|
| 71 |
if file_name is not None and file_name!='':
|
| 72 |
print(f"Provided file: {file_name}.")
|
| 73 |
-
messages=[
|
| 74 |
else:
|
| 75 |
-
messages=[
|
| 76 |
|
| 77 |
response = self.agent.invoke({"messages":messages}, config={"callbacks": [self.langfuse_handler]})
|
| 78 |
answer = response['messages'][-1].content
|
|
|
|
| 70 |
|
| 71 |
if file_name is not None and file_name!='':
|
| 72 |
print(f"Provided file: {file_name}.")
|
| 73 |
+
messages=[sys_msg] + [HumanMessage(content=f"{question}. The file you have access to is {file_name}.")]
|
| 74 |
else:
|
| 75 |
+
messages=[sys_msg] + [HumanMessage(content=question)]
|
| 76 |
|
| 77 |
response = self.agent.invoke({"messages":messages}, config={"callbacks": [self.langfuse_handler]})
|
| 78 |
answer = response['messages'][-1].content
|