ZeroTimo commited on
Commit
c499c9b
·
verified ·
1 Parent(s): 1f1da72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -16,12 +16,13 @@ class BasicAgent:
16
  print("LLM Tool-Enhanced Agent initialized.")
17
 
18
  def __call__(self, question: str) -> str:
19
- try:
20
- result = agent_executor.run(question)
21
- return result.strip()
22
- except Exception as e:
23
- print(f"Agent error: {e}")
24
- return "I don't know."
 
25
 
26
  def run_and_submit_all( profile: gr.OAuthProfile | None):
27
  """
 
16
  print("LLM Tool-Enhanced Agent initialized.")
17
 
18
  def __call__(self, question: str) -> str:
19
+ try:
20
+ result = agent_executor.invoke({"messages": [HumanMessage(content=question)]})
21
+ answer = result["messages"][-1].content
22
+ return answer.strip()
23
+ except Exception as e:
24
+ print(f"Agent error: {e}")
25
+ return "I don't know."
26
 
27
  def run_and_submit_all( profile: gr.OAuthProfile | None):
28
  """