mirjam-m commited on
Commit
e69f0ff
·
1 Parent(s): e94d38d

graph test

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -144,7 +144,11 @@ class BasicAgent:
144
 
145
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
146
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
147
- api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=100000)
 
 
 
 
148
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
149
  results = wiki_tool.run(str(state["wiki_request"]))
150
  print(f"Wiki results: {results}")
 
144
 
145
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
146
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
147
+ api_wrapper = WikipediaAPIWrapper(
148
+ top_k_results=1,
149
+ doc_content_chars_max=100000,
150
+ load_all_available_meta=True,
151
+ )
152
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
153
  results = wiki_tool.run(str(state["wiki_request"]))
154
  print(f"Wiki results: {results}")