From 7f9b6b000d1ff1ad7382b57f068d2675adf42495 Mon Sep 17 00:00:00 2001
From: jassad095 <joaopfassad@gmail.com>
Date: Wed, 27 May 2026 17:51:38 -0300
Subject: [PATCH] Export remaining Literal type aliases

---
 src/claude_agent_sdk/__init__.py |  8 +++++++
 tests/test_types.py              | 36 ++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+)

diff --git a/src/claude_agent_sdk/__init__.py b/src/claude_agent_sdk/__init__.py
index 555f49fc..5fb49310 100644
--- a/src/claude_agent_sdk/__init__.py
+++ b/src/claude_agent_sdk/__init__.py
@@ -58,6 +58,7 @@
 from .types import (
     AgentDefinition,
     AssistantMessage,
+    AssistantMessageError,
     BaseHookInput,
     CanUseTool,
     ClaudeAgentOptions,
@@ -85,6 +86,7 @@
     MirrorErrorMessage,
     NotificationHookInput,
     NotificationHookSpecificOutput,
+    PermissionBehavior,
     PermissionMode,
     PermissionRequestHookInput,
     PermissionRequestHookSpecificOutput,
@@ -92,6 +94,7 @@
     PermissionResultAllow,
     PermissionResultDeny,
     PermissionUpdate,
+    PermissionUpdateDestination,
     PostToolUseFailureHookInput,
     PostToolUseFailureHookSpecificOutput,
     PostToolUseHookInput,
@@ -138,6 +141,7 @@
     ThinkingConfigAdaptive,
     ThinkingConfigDisabled,
     ThinkingConfigEnabled,
+    ThinkingDisplay,
     ToolPermissionContext,
     ToolResultBlock,
     ToolUseBlock,
@@ -541,6 +545,7 @@ async def call_tool(name: str, arguments: dict[str, Any]) -> Any:
     "McpToolInfo",
     "UserMessage",
     "AssistantMessage",
+    "AssistantMessageError",
     "SystemMessage",
     "TaskStartedMessage",
     "TaskProgressMessage",
@@ -563,6 +568,7 @@ async def call_tool(name: str, arguments: dict[str, Any]) -> Any:
     "ThinkingConfigAdaptive",
     "ThinkingConfigEnabled",
     "ThinkingConfigDisabled",
+    "ThinkingDisplay",
     "ToolUseBlock",
     "ToolResultBlock",
     "ServerToolName",
@@ -578,6 +584,8 @@ async def call_tool(name: str, arguments: dict[str, Any]) -> Any:
     "PermissionResultAllow",
     "PermissionResultDeny",
     "PermissionUpdate",
+    "PermissionBehavior",
+    "PermissionUpdateDestination",
     # Hook support
     "HookCallback",
     "HookContext",
diff --git a/tests/test_types.py b/tests/test_types.py
index 44d21550..6691c121 100644
--- a/tests/test_types.py
+++ b/tests/test_types.py
@@ -4,15 +4,19 @@
 
 from claude_agent_sdk import (
     AssistantMessage,
+    AssistantMessageError,
     ClaudeAgentOptions,
     EffortLevel,
     NotificationHookInput,
     NotificationHookSpecificOutput,
+    PermissionBehavior,
     PermissionRequestHookInput,
     PermissionRequestHookSpecificOutput,
+    PermissionUpdateDestination,
     ResultMessage,
     SubagentStartHookInput,
     SubagentStartHookSpecificOutput,
+    ThinkingDisplay,
 )
 from claude_agent_sdk.types import (
     PermissionRuleValue,
@@ -32,6 +36,38 @@ def test_effort_level_is_exported():
     assert set(get_args(EffortLevel)) == {"low", "medium", "high", "xhigh", "max"}
 
 
+def test_assistant_message_error_is_exported():
+    """AssistantMessageError is part of the public package API for downstream wrappers."""
+    assert set(get_args(AssistantMessageError)) == {
+        "authentication_failed",
+        "billing_error",
+        "rate_limit",
+        "invalid_request",
+        "server_error",
+        "unknown",
+    }
+
+
+def test_permission_behavior_is_exported():
+    """PermissionBehavior is part of the public package API for downstream wrappers."""
+    assert set(get_args(PermissionBehavior)) == {"allow", "deny", "ask"}
+
+
+def test_permission_update_destination_is_exported():
+    """PermissionUpdateDestination is part of the public package API for downstream wrappers."""
+    assert set(get_args(PermissionUpdateDestination)) == {
+        "userSettings",
+        "projectSettings",
+        "localSettings",
+        "session",
+    }
+
+
+def test_thinking_display_is_exported():
+    """ThinkingDisplay is part of the public package API for downstream wrappers."""
+    assert set(get_args(ThinkingDisplay)) == {"summarized", "omitted"}
+
+
 class TestPermissionUpdate:
     """Test PermissionUpdate wire-format conversion."""