introduce thinking mode and reduce recursion limit
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def ask_wiki(query: str) -> str:
|
|
64 |
if not query.strip():
|
65 |
return "Please provide a valid query."
|
66 |
try:
|
67 |
-
wiki_toolapi_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=
|
68 |
wiki_tool = WikipediaQueryRun(api_wrapper=wiki_toolapi_wrapper)
|
69 |
result = wiki_tool.run(query)
|
70 |
return result
|
@@ -177,10 +177,10 @@ class BasicAgent:
|
|
177 |
HumanMessage(
|
178 |
# content="Who is Barack Obama?"
|
179 |
# content="Divide 6790 by 5"
|
180 |
-
content=question + '
|
181 |
)
|
182 |
]
|
183 |
-
response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit":
|
184 |
response_text = response['messages'][-1].content
|
185 |
return response_text.split('</think>')[-1]
|
186 |
|
|
|
64 |
if not query.strip():
|
65 |
return "Please provide a valid query."
|
66 |
try:
|
67 |
+
wiki_toolapi_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=8000)
|
68 |
wiki_tool = WikipediaQueryRun(api_wrapper=wiki_toolapi_wrapper)
|
69 |
result = wiki_tool.run(query)
|
70 |
return result
|
|
|
177 |
HumanMessage(
|
178 |
# content="Who is Barack Obama?"
|
179 |
# content="Divide 6790 by 5"
|
180 |
+
content=question # + '/nothink'
|
181 |
)
|
182 |
]
|
183 |
+
response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit": 5})
|
184 |
response_text = response['messages'][-1].content
|
185 |
return response_text.split('</think>')[-1]
|
186 |
|