giulia-fontanella commited on
Commit
bf08250
·
verified ·
1 Parent(s): cf0d636

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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=[[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
 
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