hanshan1988 commited on
Commit
1cde839
·
verified ·
1 Parent(s): 1c99ac0

introduce thinking mode and reduce recursion limit

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=1000)
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 + ' /nothink'
181
  )
182
  ]
183
- response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit": 10})
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