dennis111 commited on
Commit
445603a
·
1 Parent(s): 788c167
Files changed (2) hide show
  1. agent.py +1 -1
  2. app.py +1 -1
agent.py CHANGED
@@ -34,7 +34,7 @@ def get_graph(llm):
34
  print("\n-------------------- Agent has been called -----------------------------------\n")
35
  prompt = prompt_template.invoke(state["messages"])
36
  response = llm.invoke(prompt)
37
-
38
  return {"messages": [response], "aggregate": ["Agent"]}
39
 
40
  # Build graph
 
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["messages"][-1].content)
38
  return {"messages": [response], "aggregate": ["Agent"]}
39
 
40
  # Build graph
app.py CHANGED
@@ -23,7 +23,7 @@ class BasicAgent:
23
  def __call__(self, question: str) -> str:
24
  print(f"Agent received question (first 50 chars): {question[:50]}...")
25
  response = self.graph.invoke({"messages": [HumanMessage(content="question"),]})
26
-
27
  return response["messages"][-1].content
28
 
29
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
23
  def __call__(self, question: str) -> str:
24
  print(f"Agent received question (first 50 chars): {question[:50]}...")
25
  response = self.graph.invoke({"messages": [HumanMessage(content="question"),]})
26
+ print("Agents answert to the question: ",response["messages"][-1].content)
27
  return response["messages"][-1].content
28
 
29
  def run_and_submit_all( profile: gr.OAuthProfile | None):