From b95c60b106ae5a80caafdc25e1ecdc8638e86899 Mon Sep 17 00:00:00 2001
From: cthomas <caren@letta.com>
Date: Fri, 29 Aug 2025 13:32:31 -0700
Subject: [PATCH] feat: add new approval message types to letta message union
 [LET-4113] (#4311)

feat: add new approval message types to letta message union
---
 letta/schemas/letta_message.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/letta/schemas/letta_message.py b/letta/schemas/letta_message.py
index 16b6f23f..0e3859fd 100644
--- a/letta/schemas/letta_message.py
+++ b/letta/schemas/letta_message.py
@@ -312,7 +312,17 @@ class AssistantMessage(LettaMessage):
 
 # NOTE: use Pydantic's discriminated unions feature: https://docs.pydantic.dev/latest/concepts/unions/#discriminated-unions
 LettaMessageUnion = Annotated[
-    Union[SystemMessage, UserMessage, ReasoningMessage, HiddenReasoningMessage, ToolCallMessage, ToolReturnMessage, AssistantMessage],
+    Union[
+        SystemMessage,
+        UserMessage,
+        ReasoningMessage,
+        HiddenReasoningMessage,
+        ToolCallMessage,
+        ToolReturnMessage,
+        AssistantMessage,
+        ApprovalRequestMessage,
+        ApprovalResponseMessage,
+    ],
     Field(discriminator="message_type"),
 ]
 
@@ -327,6 +337,8 @@ def create_letta_message_union_schema():
             {"$ref": "#/components/schemas/ToolCallMessage"},
             {"$ref": "#/components/schemas/ToolReturnMessage"},
             {"$ref": "#/components/schemas/AssistantMessage"},
+            {"$ref": "#/components/schemas/ApprovalRequestMessage"},
+            {"$ref": "#/components/schemas/ApprovalResponseMessage"},
         ],
         "discriminator": {
             "propertyName": "message_type",
@@ -338,6 +350,8 @@ def create_letta_message_union_schema():
                 "tool_call_message": "#/components/schemas/ToolCallMessage",
                 "tool_return_message": "#/components/schemas/ToolReturnMessage",
                 "assistant_message": "#/components/schemas/AssistantMessage",
+                "approval_request_message": "#/components/schemas/ApprovalRequestMessage",
+                "approval_response_message": "#/components/schemas/ApprovalResponseMessage",
             },
         },
     }