mchinea commited on
Commit
ed6bc85
·
1 Parent(s): 6063bbe

update agent

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -18,13 +18,12 @@ class MyAgent:
18
  print("MyAgent initialized.")
19
  self.graph = build_agent_graph()
20
  def __call__(self, question: str) -> str:
21
- print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  # Wrap the question in a HumanMessage from langchain_core
24
  messages = [HumanMessage(content=question)]
25
  messages = self.graph.invoke({"messages": messages})
26
  answer = messages['messages'][-1].content
27
- print(f"Agent returning fixed answer: {answer}")
28
  return answer
29
 
30
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
18
  print("MyAgent initialized.")
19
  self.graph = build_agent_graph()
20
  def __call__(self, question: str) -> str:
 
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  # Wrap the question in a HumanMessage from langchain_core
23
  messages = [HumanMessage(content=question)]
24
  messages = self.graph.invoke({"messages": messages})
25
  answer = messages['messages'][-1].content
26
+ print(f"Agent returning answer: {answer}")
27
  return answer
28
 
29
  def run_and_submit_all( profile: gr.OAuthProfile | None):