mirjam-m commited on
Commit
a30a657
·
1 Parent(s): 9f197ec

graph test

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -92,6 +92,7 @@ class BasicAgent:
92
  self.graph.add_edge(START, "log_question")
93
  self.graph.add_edge("log_question", "invoke_model")
94
  self.graph.add_edge("web_search", "invoke_model")
 
95
 
96
  # Add conditional edges
97
  self.graph.add_conditional_edges(
@@ -143,7 +144,7 @@ class BasicAgent:
143
 
144
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
145
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
146
- api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=5000)
147
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
148
  results = wiki_tool.run(str(state["wiki_request"]))
149
  print(f"Wiki results: {results}")
 
92
  self.graph.add_edge(START, "log_question")
93
  self.graph.add_edge("log_question", "invoke_model")
94
  self.graph.add_edge("web_search", "invoke_model")
95
+ self.graph.add_edge("wiki_search", "invoke_model")
96
 
97
  # Add conditional edges
98
  self.graph.add_conditional_edges(
 
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=10000)
148
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
149
  results = wiki_tool.run(str(state["wiki_request"]))
150
  print(f"Wiki results: {results}")