# example_usage.py
from graph.graph_builder import graph
from langchain_core.messages import HumanMessage

# Simple example
def ask_agent(question: str):
    initial_state = {"messages": [HumanMessage(content=question)]}
    result = graph.invoke(initial_state)
    return result["messages"][-1].content

# Usage
if __name__ == "__main__":
    response = ask_agent("How many albums did Michael Jackson release in his lifetime?")
    print(f"Agent: {response}")