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

graph test

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -144,12 +144,12 @@ 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(
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}")
155
  state["messages"].append(
 
144
 
145
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
146
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
147
+ wiki_tool = 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}")
155
  state["messages"].append(