Function modify_query

autoagent/tools/rag_tools.py:151–195

Modify the query based on what you know. Use this function when you need to modify the query to search for more relevant information. Args: what_you_know: The knowledge you have about the case. query_text: The original query. Returns: The modified query.

(what_you_know: str, query_text: str, **kwargs)

@register_tool("modify_query")
@register_plugin_tool("modify_query")
def modify_query(what_you_know: str, query_text: str, **kwargs) -> str:
    """
    Modify the query based on what you know. Use this function when you need to modify the query to search for more relevant information.

    Args:
        what_you_know: The knowledge you have about the case.
        query_text: The original query.
    Returns:
        The modified query.
    """
    system_prompt = \
    f"""

    Assume you are an assistant searching for information. Now that you already know some knowledge ([What you know]), what sub-questions ([Modified query]) do you need to search for to help you answer the question ([Query]) you want to explore.

    Modify the query based on what you know, here is some example:
    Example 1:
    [What you know]: Alice and Bob have lunch together at 12:00 PM.
    [Query]: What did Alice and Bob do after the lunch?
    [Modified query]: What did Alice and Bob do after 12:00 PM?

    Example 2:
    [What you know]: Alice and Bob went to the cinema yesterday.
    [Query]: What did Alice and Bob do after the cinema?
    [Modified query]: What did Alice and Bob do yesterday?
    
    Return only 1 modified query.
    """
    
    user_prompt = f"""
    What you know: {what_you_know}
    Query: {query_text}
    Modified query:
    """
    create_params = {
        "model": "gpt-4o-mini",
        "messages": [
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": user_prompt}
        ],
        "stream": False,
    }
    response = completion(**create_params)
    modified_query = response.choices[0].message.content
    return f"The modified query is: {modified_query}. Now use function `query_db` to search the related information in the DataBase."