latest updates

src/openrouter/components/outputmessage.py

@@ -7,8 +7,10 @@ from .openairesponsesrefusalcontent import (
 )
 from .responseoutputtext import ResponseOutputText, ResponseOutputTextTypedDict
 from openrouter.types import BaseModel
+from openrouter.utils import get_discriminator
+from pydantic import Discriminator, Tag
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypeAliasType, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 OutputMessageRole = Literal["assistant",]
@@ -52,9 +54,13 @@ OutputMessageContentTypedDict = TypeAliasType(
 )
 
 
-OutputMessageContent = TypeAliasType(
-    "OutputMessageContent", Union[OpenAIResponsesRefusalContent, ResponseOutputText]
-)
+OutputMessageContent = Annotated[
+    Union[
+        Annotated[ResponseOutputText, Tag("output_text")],
+        Annotated[OpenAIResponsesRefusalContent, Tag("refusal")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
 
 
 class OutputMessageTypedDict(TypedDict):