langgraph_sdk_async_client_thread_creation_and_streaming.py

1import asyncio
2from langgraph_sdk import get_client
3
4async def main():
5    # Initialize the client with the URL of your LangGraph deployment
6    # If running locally with the LangGraph CLI, the default is http://localhost:8123
7    client = get_client(url="http://localhost:8123")
8
9    # List available assistants (graphs)
10    assistants = await client.assistants.search()
11    if not assistants:
12        print("No assistants found. Make sure your LangGraph server is running.")
13        return
14    
15    assistant = assistants[0]
16    assistant_id = assistant["assistant_id"]
17
18    # Create a new thread for the conversation
19    thread = await client.threads.create()
20
21    # Define the input for the graph
22    input_data = {"messages": [{"role": "user", "content": "Hello!"}]}
23
24    # Stream the runs from the graph
25    async for event in client.runs.stream(
26        thread["thread_id"],
27        assistant_id,
28        input=input_data,
29    ):
30        if event.event == "values":
31            print(f"Node: {event.namespace[-1]}")
32            print(f"Values: {event.data}")
33            print("-" * 20)
34
35if __name__ == "__main__":
36    asyncio.run(main())