File size: 1,157 Bytes
22dcdfd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import asyncio
from typing import cast
from uuid import uuid4
from dotenv import load_dotenv
from langchain_core.messages import HumanMessage
from langchain_core.runnables import RunnableConfig
from langgraph.graph import MessagesState
from langgraph.graph.state import CompiledStateGraph
load_dotenv()
from agents import DEFAULT_AGENT, get_agent # noqa: E402
# The default agent uses StateGraph.compile() which returns CompiledStateGraph
agent = cast(CompiledStateGraph, get_agent(DEFAULT_AGENT))
async def main() -> None:
inputs: MessagesState = {
"messages": [HumanMessage("Find me a recipe for chocolate chip cookies")]
}
result = await agent.ainvoke(
input=inputs,
config=RunnableConfig(configurable={"thread_id": uuid4()}),
)
result["messages"][-1].pretty_print()
# Draw the agent graph as png
# requires:
# brew install graphviz
# export CFLAGS="-I $(brew --prefix graphviz)/include"
# export LDFLAGS="-L $(brew --prefix graphviz)/lib"
# pip install pygraphviz
#
# agent.get_graph().draw_png("agent_diagram.png")
if __name__ == "__main__":
asyncio.run(main())
|