from typing import TypedDict
from agnflow import Node, Flow

# Define state
class State(TypedDict):
  message: str

# Define nodes
start = Node(
  name="start",
  exec=lambda s: ("process", {"message": "Hello"}) # Instantiate Node, specify exec
)
class ProcessNode(Node): # Inherit Node, override exec
  def exec(state: State):
    # Route to end node and update message state
    return "end", {"message": s["message"] + " World"}
process = ProcessNode() # No name specified, automatically gets variable name "process"
end = Node(exec=print) # Print state

# Create workflow
flow = Flow(start >> process >> end)

# Run workflow
state: State = {"message": ""}
flow.run(state)  # Output: {'message': 'Hello World'}

from agnflow.chatbot.server import Server

server = Server()
server.run()

flow.render_mermaid(saved_file="flow.png")  # Directly generate image

# Add/Remove nodes at runtime
flow += new_node
flow -= old_node

# Symmetric connect/disconnect syntax
a >> b >> c    # Connect nodes
a - b - c      # Symmetrically disconnect

from typing import TypedDict
from agnflow import Node, Flow

# 定义状�
class State(TypedDict):
  message: str

# 定义节点
start = Node(
  name="start",
  exec=lambda s: ("process", {"message": "Hello"}) # 实例化 Node，指定 exec
)
class ProcessNode(Node): # 继承 Node，�写 exec
  def exec(state: State):
    # 路由到 end 节点，并且更新 message 状�，
    return "end", {"message": s["message"] + " World"}
process = ProcessNode() #  没有指定 name 属性，自动获�实例化��� "process"
end = Node(exec=print) # 打� state

# 创建工作�
flow = Flow(start >> process >> end)

# �行工作�
state: State = {"message": ""}
flow.run(state)  # 输出: {'message': 'Hello World'}

from agnflow.chatbot.server import Server

server = Server()
server.run()

flow.render_mermaid(saved_file="flow.png")  # 直接生�图片

# �行期增删节点
flow += new_node
flow -= old_node

# 对称的连接/断开语法
a >> b >> c    # 建立连接
a - b - c      # 对称断开