diff --git a/compose/graph/react_with_interrupt/main.go b/compose/graph/react_with_interrupt/main.go
index 5fd312b1..bdd99768 100644
--- a/compose/graph/react_with_interrupt/main.go
+++ b/compose/graph/react_with_interrupt/main.go
@@ -34,7 +34,7 @@ import (
 )
 
 func main() {
-	compose.RegisterSerializableType[myState]("state")
+	_ = compose.RegisterSerializableType[myState]("state")
 
 	ctx := context.Background()
 	runner, err := composeGraph[map[string]any, *schema.Message](
@@ -95,7 +95,7 @@ func newChatTemplate(_ context.Context) prompt.ChatTemplate {
 	)
 }
 
-func newChatModel(ctx context.Context) model.ChatModel {
+func newChatModel(ctx context.Context) model.ToolCallingChatModel {
 	cm, err := openai.NewChatModel(ctx, &openai.ChatModelConfig{
 		APIKey:  os.Getenv("OPENAI_API_KEY"),
 		Model:   os.Getenv("OPENAI_MODEL"),
@@ -146,7 +146,7 @@ type myState struct {
 	history []*schema.Message
 }
 
-func composeGraph[I, O any](ctx context.Context, tpl prompt.ChatTemplate, cm model.ChatModel, tn *compose.ToolsNode, store compose.CheckPointStore) (compose.Runnable[I, O], error) {
+func composeGraph[I, O any](ctx context.Context, tpl prompt.ChatTemplate, cm model.ToolCallingChatModel, tn *compose.ToolsNode, store compose.CheckPointStore) (compose.Runnable[I, O], error) {
 	g := compose.NewGraph[I, O](compose.WithGenLocalState(func(ctx context.Context) *myState {
 		return &myState{}
 	}))