wip: refactoring tui

2 files changed, 71 insertions, 1 deletions

diff --git a/packages/tui/pkg/client/gen/openapi.json b/packages/tui/pkg/client/gen/openapi.json
index bdad3ca20..a9f19d4b3 100644
--- a/packages/tui/pkg/client/gen/openapi.json
+++ b/packages/tui/pkg/client/gen/openapi.json
@@ -427,6 +427,9 @@
             "$ref": "#/components/schemas/Event.message.updated"
           },
           {
+            "$ref": "#/components/schemas/Event.message.part.updated"
+          },
+          {
             "$ref": "#/components/schemas/Event.lsp.client.diagnostics"
           },
           {
@@ -441,6 +444,7 @@
           "mapping": {
             "storage.write": "#/components/schemas/Event.storage.write",
             "message.updated": "#/components/schemas/Event.message.updated",
+            "message.part.updated": "#/components/schemas/Event.message.part.updated",
             "lsp.client.diagnostics": "#/components/schemas/Event.lsp.client.diagnostics",
             "permission.updated": "#/components/schemas/Event.permission.updated",
             "session.updated": "#/components/schemas/Event.session.updated"
@@ -533,6 +537,9 @@
                   "created"
                 ]
               },
+              "error": {
+                "type": "string"
+              },
               "sessionID": {
                 "type": "string"
               },
@@ -841,6 +848,30 @@
           "type"
         ]
       },
+      "Event.message.part.updated": {
+        "type": "object",
+        "properties": {
+          "type": {
+            "type": "string",
+            "const": "message.part.updated"
+          },
+          "properties": {
+            "type": "object",
+            "properties": {
+              "part": {
+                "$ref": "#/components/schemas/Message.Part"
+              }
+            },
+            "required": [
+              "part"
+            ]
+          }
+        },
+        "required": [
+          "type",
+          "properties"
+        ]
+      },
       "Event.lsp.client.diagnostics": {
         "type": "object",
         "properties": {
diff --git a/packages/tui/pkg/client/generated-client.go b/packages/tui/pkg/client/generated-client.go
index cd0de355c..1237685a9 100644
--- a/packages/tui/pkg/client/generated-client.go
+++ b/packages/tui/pkg/client/generated-client.go
@@ -53,6 +53,14 @@ type EventLspClientDiagnostics struct {
 	Type string `json:"type"`
 }
 
+// EventMessagePartUpdated defines model for Event.message.part.updated.
+type EventMessagePartUpdated struct {
+	Properties struct {
+		Part MessagePart `json:"part"`
+	} `json:"properties"`
+	Type string `json:"type"`
+}
+
 // EventMessageUpdated defines model for Event.message.updated.
 type EventMessageUpdated struct {
 	Properties struct {
@@ -99,7 +107,8 @@ type MessageInfo struct {
 				Reasoning float32 `json:"reasoning"`
 			} `json:"tokens"`
 		} `json:"assistant,omitempty"`
-		SessionID string `json:"sessionID"`
+		Error     *string `json:"error,omitempty"`
+		SessionID string  `json:"sessionID"`
 		Time      struct {
 			Completed *float32 `json:"completed,omitempty"`
 			Created   float32  `json:"created"`
@@ -352,6 +361,34 @@ func (t *Event) MergeEventMessageUpdated(v EventMessageUpdated) error {
 	return err
 }
 
+// AsEventMessagePartUpdated returns the union data inside the Event as a EventMessagePartUpdated
+func (t Event) AsEventMessagePartUpdated() (EventMessagePartUpdated, error) {
+	var body EventMessagePartUpdated
+	err := json.Unmarshal(t.union, &body)
+	return body, err
+}
+
+// FromEventMessagePartUpdated overwrites any union data inside the Event as the provided EventMessagePartUpdated
+func (t *Event) FromEventMessagePartUpdated(v EventMessagePartUpdated) error {
+	v.Type = "message.part.updated"
+	b, err := json.Marshal(v)
+	t.union = b
+	return err
+}
+
+// MergeEventMessagePartUpdated performs a merge with any union data inside the Event, using the provided EventMessagePartUpdated
+func (t *Event) MergeEventMessagePartUpdated(v EventMessagePartUpdated) error {
+	v.Type = "message.part.updated"
+	b, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+
+	merged, err := runtime.JSONMerge(t.union, b)
+	t.union = merged
+	return err
+}
+
 // AsEventLspClientDiagnostics returns the union data inside the Event as a EventLspClientDiagnostics
 func (t Event) AsEventLspClientDiagnostics() (EventLspClientDiagnostics, error) {
 	var body EventLspClientDiagnostics
@@ -452,6 +489,8 @@ func (t Event) ValueByDiscriminator() (interface{}, error) {
 	switch discriminator {
 	case "lsp.client.diagnostics":
 		return t.AsEventLspClientDiagnostics()
+	case "message.part.updated":
+		return t.AsEventMessagePartUpdated()
 	case "message.updated":
 		return t.AsEventMessageUpdated()
 	case "permission.updated":