WikipediaQueryRunTool

def query_run(self, query: str) -> Union[list[str], str]:
    """
    Execute a Wikipedia search and return processed page summaries.

    Args:
        query (str): Search term(s) to look up in Wikipedia. Will be truncated to
            MAX_QUERY_LENGTH characters if too long.

    Returns:
        Union[list[str], str]:
            - List of formatted page summaries ("Page: <title>\nSummary: <content>") if successful
            - Error message string if no results found or exception occurs

    Note:
        Automatically handles API exceptions and returns error strings for robust operation
    """
    try:
        if self.verbose:
            print(f"INFO\t [{self.tool_name}] search query='{query[:MAX_QUERY_LENGTH]}' top_k={self.top_k}")
        search_results = self.wiki_cli.search(query[:MAX_QUERY_LENGTH], limit=self.top_k)
        summaries: list[str] = []
        for item in search_results:
            title = item["title"]
            page = self.wiki_cli.get_page(title)
            # Only format the summary if the page exists and has a summary.
            if page is not None and page.summary:
                summary = f"Page: {title}\nSummary: {page.summary}"
                summaries.append(summary)
        if not summaries:
            return "No good Wikipedia Search Result was found"
        return summaries
    except Exception as e:
        return f"wikipedia search failed: {str(e)}"

def register_for_llm(self, agent: "ConversableAgent") -> None:
    """Registers the tool for use with a ConversableAgent's language model (LLM).

    This method registers the tool so that it can be invoked by the agent during
    interactions with the language model.

    Args:
        agent (ConversableAgent): The agent to which the tool will be registered.
    """
    if self._func_schema:
        agent.update_tool_signature(self._func_schema, is_remove=False)
    else:
        agent.register_for_llm()(self)

def register_for_execution(self, agent: "ConversableAgent") -> None:
    """Registers the tool for direct execution by a ConversableAgent.

    This method registers the tool so that it can be executed by the agent,
    typically outside of the context of an LLM interaction.

    Args:
        agent (ConversableAgent): The agent to which the tool will be registered.
    """
    agent.register_for_execution()(self)

def register_tool(self, agent: "ConversableAgent") -> None:
    """Register a tool to be both proposed and executed by an agent.

    Equivalent to calling both `register_for_llm` and `register_for_execution` with the same agent.

    Note: This will not make the agent recommend and execute the call in the one step. If the agent
    recommends the tool, it will need to be the next agent to speak in order to execute the tool.

    Args:
        agent (ConversableAgent): The agent to which the tool will be registered.
    """
    self.register_for_llm(agent)
    self.register_for_execution(agent)