errchh commited on
Commit
2cb2810
·
1 Parent(s): 72b8c5f

fix basic agent

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -16,11 +16,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
 
 
19
  def __call__(self, question: str) -> str:
20
  print(f"Agent received question (first 50 chars): {question[:50]}...")
21
- fixed_answer = "This is a default answer."
22
- print(f"Agent returning fixed answer: {fixed_answer}")
23
- return fixed_answer
 
24
 
25
  def run_and_submit_all( profile: gr.OAuthProfile | None):
26
  """
 
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
19
+ self.graph = build_graph()
20
+
21
  def __call__(self, question: str) -> str:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
+ messages = [HumanMessage(content=question)]
24
+ result = self.graph.invoke({"messages": messages})
25
+ answer = result['messages'][-1].content
26
+ return answer
27
 
28
  def run_and_submit_all( profile: gr.OAuthProfile | None):
29
  """