feat(devops): add InputType,OutputType and GenLocalStateType in Graph… (

#26) feat(devops): add InputType,OutputType and GenLocalStateType in GraphSchema
cloudwego · Jan 6, 2025 · ed83839 · ed83839
1 parent 0c38eea
commit ed83839
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/devops/model/canvas.go b/devops/model/canvas.go
@@ -54,8 +54,15 @@ type GraphSchema struct {
 	Branches  []*Branch `json:"branches"`
 
 	// graph config option
-	NodeTriggerMode     NodeTriggerMode `json:"node_trigger_mode"`
-	GenLocalStateMethod *string         `json:"gen_local_state_method,omitempty"`
+	NodeTriggerMode NodeTriggerMode `json:"node_trigger_mode"`
+	GenLocalState   *GenLocalState  `json:"gen_local_state,omitempty"`
+	InputType       string          `json:"input_type"`
+	OutputType      string          `json:"output_type"`
+}
+
+type GenLocalState struct {
+	Method     string `json:"method"`
+	OutputType string `json:"output_type"`
 }
 
 type Node struct {
@@ -101,8 +108,9 @@ type Branch struct {
 }
 
 type Condition struct {
-	Method   string `json:"method"`
-	IsStream bool   `json:"is_stream"`
+	Method    string `json:"method"`
+	IsStream  bool   `json:"is_stream"`
+	InputType string `json:"input_type"`
 }
 
 type JsonType string