Function answer_query

autoagent/tools/rag_tools.py:201–236

Answer the user query based on the supporting documents. Args: original_user_query: The original user query. supporting_docs: The supporting documents. Returns: The answer to the user query.

(original_user_query: str, supporting_docs: str, **kwargs)

@register_tool("answer_query")
@register_plugin_tool("answer_query")
def answer_query(original_user_query: str, supporting_docs: str, **kwargs) -> str:
    """
    Answer the user query based on the supporting documents.

    Args:
        original_user_query: The original user query.
        supporting_docs: The supporting documents.
    Returns:
        The answer to the user query.
    """
    system_prompt = \
    f"""
    You are a helpful assistant. Answer the user query based on the supporting documents.  
    If you have not found the answer, say "Insufficient information."
    """
    
    user_prompt = f"""
    
    Here is the original user query and the supporting documents:
    Original user query: {original_user_query}
    Supporting documents: {supporting_docs}
    Answer:
    """
    create_params = {
        "model": "gpt-4o-mini",
        "messages": [
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": user_prompt}
        ],
        "stream": False,
    }
    response = completion(**create_params)
    answer = response.choices[0].message.content


    return answer


@register_tool("can_answer")