sampsong commited on
Commit
21bdd55
·
1 Parent(s): 0b95002

parse additional kwargs into the message

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,8 +62,8 @@ class BasicAgent:
62
  messages = self.graph.invoke({"messages": formattedMessage})
63
  else:
64
  print("active langfuse")
65
- formattedMessage = [{"role": "system", "content": systemPrompt}] + \
66
- [{"role":"user","content":msg.content} for msg in messages]
67
  #formattedMessage.append({"role": "system", "content": systemPrompt})
68
  #formattedMessage.append(SystemMessage(content="systemPrompt"))
69
  print(f"formattedMessage : {formattedMessage}")
 
62
  messages = self.graph.invoke({"messages": formattedMessage})
63
  else:
64
  print("active langfuse")
65
+ formattedMessage = [{"role": "system", "content": systemPrompt,"additional_kwargs": {"taskID": taskID}}] + \
66
+ [{"role":"user","content":msg.content,"additional_kwargs": {"taskID": taskID}} for msg in messages]
67
  #formattedMessage.append({"role": "system", "content": systemPrompt})
68
  #formattedMessage.append(SystemMessage(content="systemPrompt"))
69
  print(f"formattedMessage : {formattedMessage}")