wehub-resource-sync
30 lines
722 B
Python
30 lines
722 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()
|