Spaces:
Sleeping
Sleeping
graph test
Browse files
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=
|
| 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}")
|