type StateGraph[S any] struct { ... }

type Graph interface {
    Invoke(ctx context.Context, input S) (S, error)
    Stream(ctx context.Context, input S) (<-chan StreamEvent, error)
}

type CheckpointSaver interface {
    Get(ctx context.Context, checkpointID string) (*Checkpoint, error)
    Put(ctx context.Context, checkpoint *Checkpoint) error
}

import "github.com/LingByte/ling-base/agentkit/graph"

type AgentState struct {
    Messages []compat.Message
    Step     int
}

sg := graph.NewStateGraph[AgentState]()

sg.AddNode("llm", llmNodeFunc)
sg.AddNode("tools", toolNodeFunc)
sg.AddEdge("llm", "tools")
sg.AddConditionalEdge("tools", shouldContinue, map[string]string{
    "continue": "llm",
    "end":      graph.END,
})

g, _ := sg.Compile()
final, err := g.Invoke(ctx, AgentState{Messages: initialMsgs})

import "github.com/LingByte/ling-base/agentkit/graph/checkpoint/sqlite"

saver, _ := sqlite.NewSaver("checkpoints.db")
g, _ := sg.Compile(graph.WithCheckpointSaver(saver))

import "github.com/LingByte/ling-base/agentkit/agent/graphagent"

ga, _ := graphagent.New("workflow", graphagent.WithGraph(g))
r := runner.NewRunner("app", ga)