promptfoo--promptfoo/examples/integration-langgraph/provider.py

from agent import run_research_agent


# Main API function that external tools or systems will call
def call_api(prompt, options, context):
    """
    Executes the research agent with the given prompt.

    Args:
        prompt (str): The research query or question.
        options (dict): Additional options for future extension (currently unused).
        context (dict): Contextual information (currently unused).

    Returns:
        dict: A dictionary containing the agent's output or an error message.
    """
    try:
        # Run the research agent and get the result
        result = run_research_agent(prompt)
        # Wrap and return the result inside a dictionary
        return {"output": result}
    except Exception as e:
        # Handle any exceptions and return an error summary
        return {"output": {"summary": f"Error: {str(e)}"}}


# If this file is run directly, execute a simple test
if __name__ == "__main__":
    print("✅ Testing Research Agent provider...")
    test_prompt = "latest AI research trends"
    result = call_api(test_prompt, {}, {})
    print("Provider result:", result)