SearchAgent

class SearchAgent(ChatAgent):

An agent that summarizes text based on a query and evaluates the relevance of an answer.

Parameters:

  • model (BaseModelBackend, optional): The model backend to use for generating responses. (default: :obj:OpenAIModel with GPT_4O_MINI)

init

def __init__(self, model: Optional[BaseModelBackend] = None):

summarize_text

def summarize_text(self, text: str, query: str):

Summarize the information from the text, base on the query.

Parameters:

  • text (str): Text to summarize.
  • query (str): What information you want.

Returns:

str: Strings with information.

def continue_search(self, query: str, answer: str):

Ask whether to continue search or not based on the provided answer.

Parameters:

  • query (str): The question.
  • answer (str): The answer to the question.

Returns:

bool: True if the user want to continue search, False otherwise.