IamRulo commited on
Commit
4342da8
·
verified ·
1 Parent(s): d50d797

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,7 +22,7 @@ class BasicAgent:
22
  def __init__(self):
23
  print("BasicAgent initialized.")
24
  #self.graph = build_graph("huggingface")
25
- self.graph = build_graph("google")
26
 
27
  def __call__(self, question: str) -> str:
28
  print(f"Agent received question (first 50 chars): {question[:50]}...")
@@ -31,7 +31,6 @@ class BasicAgent:
31
  messages = self.graph.invoke({"messages": messages})
32
  answer = messages['messages'][-1].content
33
 
34
- #print(f"answer: {answer}") #rulo
35
  return answer[14:]
36
 
37
 
 
22
  def __init__(self):
23
  print("BasicAgent initialized.")
24
  #self.graph = build_graph("huggingface")
25
+ self.graph = build_graph()
26
 
27
  def __call__(self, question: str) -> str:
28
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
31
  messages = self.graph.invoke({"messages": messages})
32
  answer = messages['messages'][-1].content
33
 
 
34
  return answer[14:]
35
 
36