feat: yaml conversation trees more robust

1 files changed, 3 insertions, 2 deletions

diff --git a/internal/behavior/behavior.go b/internal/behavior/behavior.go
index 57b2a7b..f81c73c 100644
--- a/internal/behavior/behavior.go
+++ b/internal/behavior/behavior.go
@@ -30,16 +30,17 @@ type TalkConfig struct {
 }
 
 type TalkNode struct {
-	Message string       `yaml:"message"`
+	Message MessageList  `yaml:"message"`
 	Options []TalkOption `yaml:"options"`
 	Action  *NodeAction  `yaml:"action"`
+	Next    string       `yaml:"next,omitempty"`
 }
 
 type TalkOption struct {
 	Text      string     `yaml:"text"`
 	Goto      string     `yaml:"goto"`
-	End       bool       `yaml:"end"`
 	Condition *Condition `yaml:"condition"`
+	Action    *NodeAction `yaml:"action,omitempty"`
 }
 
 type NodeAction struct {