mirror of
https://github.com/CopilotKit/CopilotKit.git
synced 2026-09-14 16:26:20 +08:00
Jordan Ritter
24 lines
697 B
Python
24 lines
697 B
Python
"""
|
|
This is the main entry point for the agent.
|
|
It defines the workflow graph, state, tools, nodes and edges.
|
|
"""
|
|
|
|
from typing import Annotated
|
|
from typing_extensions import TypedDict
|
|
from langgraph.graph import StateGraph, START, END
|
|
from langgraph.graph.message import add_messages
|
|
|
|
class State(TypedDict):
|
|
messages: Annotated[list, add_messages]
|
|
|
|
async def chat_node(state: State):
|
|
return {"messages": state["messages"] + [{"role": "assistant", "content": "This is a test message"}]}
|
|
|
|
# Define the workflow graph
|
|
workflow = StateGraph(State)
|
|
workflow.add_edge(START, "chat_node")
|
|
workflow.add_node("chat_node", chat_node)
|
|
workflow.add_edge("chat_node", END)
|
|
|
|
graph = workflow.compile()
|