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 |
|