wishwakankanamg commited on
Commit
0f9e6a3
·
1 Parent(s): 6c789a5
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -21,13 +21,15 @@ class BasicAgent:
21
  """A langgraph agent."""
22
  def __init__(self):
23
  print("BasicAgent initialized.")
24
- self.graph = build_graph()
25
 
26
  def __call__(self, question: str) -> str:
27
  print(f"Agent received question (first 50 chars): {question[:50]}...")
28
  # Wrap the question in a HumanMessage from langchain_core
29
  messages = [HumanMessage(content=question)]
30
- messages = self.graph.invoke({"messages": messages})
 
 
31
  answer = messages['messages'][-1].content
32
  return answer[14:]
33
 
 
21
  """A langgraph agent."""
22
  def __init__(self):
23
  print("BasicAgent initialized.")
24
+ self.graph = build_graph(provider="huggingface")
25
 
26
  def __call__(self, question: str) -> str:
27
  print(f"Agent received question (first 50 chars): {question[:50]}...")
28
  # Wrap the question in a HumanMessage from langchain_core
29
  messages = [HumanMessage(content=question)]
30
+ config = {"recursion_limit": 27}
31
+
32
+ messages = graph.invoke({"messages": messages}, config=config)
33
  answer = messages['messages'][-1].content
34
  return answer[14:]
35