dennis111 commited on
Commit
3a2c53a
·
1 Parent(s): 356f548
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -33,6 +33,7 @@ def get_graph(llm):
33
  def call_model(state: State):
34
  print("\n-------------------- Agent has been called -----------------------------------\n")
35
  prompt = prompt_template.invoke(state["messages"])
 
36
  response = llm.invoke(prompt)
37
  print("Agent has made a decision: ",response.content)
38
  return {"messages": [response], "aggregate": ["Agent"]}
 
33
  def call_model(state: State):
34
  print("\n-------------------- Agent has been called -----------------------------------\n")
35
  prompt = prompt_template.invoke(state["messages"])
36
+ print("\nThe Prompt is: ", prompt, "\n")
37
  response = llm.invoke(prompt)
38
  print("Agent has made a decision: ",response.content)
39
  return {"messages": [response], "aggregate": ["Agent"]}