From 755eafa7d2b8c1dfcd2d0da3da9f34d736c211c3 Mon Sep 17 00:00:00 2001
From: Hadi Khazaee Asl <hadi_khazaee_asl@yahoo.com>
Date: Wed, 30 Sep 2026 23:12:39 +0330
Subject: [PATCH] Add Support for Attaching Additional Contents to Specified
 Prompt Message when Converting to Chat Message, Specified For File Support
 ...

---
 Extensions/XAiModelsExtensions.cs | 29 ++++++++++++++++++++++++-----
 1 file changed, 24 insertions(+), 5 deletions(-)

diff --git a/Extensions/XAiModelsExtensions.cs b/Extensions/XAiModelsExtensions.cs
index 0be37c5..ab16409 100644
--- a/Extensions/XAiModelsExtensions.cs
+++ b/Extensions/XAiModelsExtensions.cs
@@ -352,22 +352,42 @@ namespace xAiModels.Extensions
             return result;
         }
 
+        /// <summary>
+        /// Converts a Message Dto to ChatMessage ...
+        /// </summary>
+        /// <param name="source"></param>
+        /// <param name="additionalContents"></param>
+        /// <returns></returns>
         public static ChatMessage ToChatMessages(
-            this XAiMessageDto source
+            this XAiMessageDto source,
+             IList<AIContent> additionalContents = null
         )
         {
-            //
-            // TODO: Handle Files Attached here ...
-
             //
             ChatMessage result = null;
 
             //
             if (!source.IsNullOrDefault())
             {
+                //
+                var contents = new List<AIContent>
+                {
+                    //
+                    // Add Source Content ...
+                    new TextContent(source.Content)
+                };
+
+                //
+                // Add Additional Content ...
+                if (additionalContents.HasChild())
+                {
+                    contents.AddRange(additionalContents);
+                }
+
                 //
                 result = new ChatMessage
                 {
+                    Contents = contents,
                     AuthorName =
                     source.Role == XAiChatRole.User &&
                     !source.Owner.IsNullOrDefault()
@@ -375,7 +395,6 @@ namespace xAiModels.Extensions
                         : string.Empty,
                     Role = source.Role.ToChatRole(),
                     MessageId = source.Id.ToString(),
-                    Contents = [new TextContent(source.Content)]
                 };
             }