Commit
·
0f9e6a3
1
Parent(s):
6c789a5
asf
Browse files
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 |
-
|
|
|
|
|
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 |
|