@@ -20,8 +20,11 @@
 from __future__ import annotations
 from .. import BaseModel
 from ...utils.unions import parse_open_union
+from .audiocontent import AudioContent, AudioContentParam
+from .documentcontent import DocumentContent, DocumentContentParam
 from .imagecontent import ImageContent, ImageContentParam
 from .textcontent import TextContent, TextContentParam
+from .videocontent import VideoContent, VideoContentParam
 from functools import partial
 from pydantic import ConfigDict
 from pydantic.functional_validators import BeforeValidator
@@ -30,7 +33,14 @@
 
 
 FunctionResultSubcontentParam = TypeAliasType(
-    "FunctionResultSubcontentParam", Union[TextContentParam, ImageContentParam]
+    "FunctionResultSubcontentParam",
+    Union[
+        TextContentParam,
+        DocumentContentParam,
+        ImageContentParam,
+        AudioContentParam,
+        VideoContentParam,
+    ],
 )
 
 
@@ -45,13 +55,23 @@ class UnknownFunctionResultSubcontent(BaseModel):
 
 
 _FUNCTION_RESULT_SUBCONTENT_VARIANTS: dict[str, Any] = {
+    "audio": AudioContent,
+    "document": DocumentContent,
     "image": ImageContent,
     "text": TextContent,
+    "video": VideoContent,
 }
 
 
 FunctionResultSubcontent = Annotated[
-    Union[ImageContent, TextContent, UnknownFunctionResultSubcontent],
+    Union[
+        AudioContent,
+        DocumentContent,
+        ImageContent,
+        TextContent,
+        VideoContent,
+        UnknownFunctionResultSubcontent,
+    ],
     BeforeValidator(
         partial(
             parse_open_union,