From c72534614fb1f5b95c484ad3db123841dea86d0f Mon Sep 17 00:00:00 2001
From: Hadi Khazaee Asl <hadi_khazaee_asl@yahoo.com>
Date: Fri, 17 Apr 2026 03:04:28 +0330
Subject: [PATCH] last ...

---
 Controllers/XAiServiceControllerBase.cs | 4 ++--
 Interfaces/IXAiService.cs               | 4 ++--
 Interfaces/IXAiServiceControllerBase.cs | 2 +-
 Services/XAiService.cs                  | 5 +++--
 xAiService.csproj                       | 3 ++-
 5 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/Controllers/XAiServiceControllerBase.cs b/Controllers/XAiServiceControllerBase.cs
index e6acfc3..feea1ec 100644
--- a/Controllers/XAiServiceControllerBase.cs
+++ b/Controllers/XAiServiceControllerBase.cs
@@ -67,7 +67,7 @@ namespace xAiService.Controllers
         }
         
         [HttpGet("AskStream")]
-        public async Task<ActionResult> AskAIStream(
+        public virtual async Task<ActionResult> AskStream(
             [FromQuery] string prompt,
             CancellationToken cancellationToken = default
         )
@@ -78,7 +78,7 @@ namespace xAiService.Controllers
             {
                 //
                 var userInfo = await GetUserInfo();
-                var result = aiService
+                var result = await aiService
                     .GetTextReponseStreamAsync(
                         prompt: prompt,
                         userInfo: userInfo,
diff --git a/Interfaces/IXAiService.cs b/Interfaces/IXAiService.cs
index 3e4b314..389c693 100644
--- a/Interfaces/IXAiService.cs
+++ b/Interfaces/IXAiService.cs
@@ -1,4 +1,4 @@
-using System.Collections.Generic;
+using System.IO;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.Extensions.AI;
@@ -51,7 +51,7 @@ namespace xAiService.Interfaces
         /// <param name="userInfo"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        IAsyncEnumerable<string> GetTextReponseStreamAsync(
+        Task<Stream> GetTextReponseStreamAsync(
             string prompt,
             XUserClaimsInfoDto userInfo,
             CancellationToken cancellationToken = default
diff --git a/Interfaces/IXAiServiceControllerBase.cs b/Interfaces/IXAiServiceControllerBase.cs
index bfe126c..9d1fe38 100644
--- a/Interfaces/IXAiServiceControllerBase.cs
+++ b/Interfaces/IXAiServiceControllerBase.cs
@@ -11,7 +11,7 @@ namespace xAiService.Interfaces
             CancellationToken cancellationToken = default
         );
         
-        Task<ActionResult> AskAIStream(
+        Task<ActionResult> AskStream(
             [FromQuery] string prompt,
             CancellationToken cancellationToken = default
         );        
diff --git a/Services/XAiService.cs b/Services/XAiService.cs
index 79728f1..f3843fc 100644
--- a/Services/XAiService.cs
+++ b/Services/XAiService.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.IO;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.Extensions.AI;
@@ -104,7 +105,7 @@ namespace xAiService.Services
         /// <param name="userInfo"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        public IAsyncEnumerable<string> GetTextReponseStreamAsync(
+        public async Task<Stream> GetTextReponseStreamAsync(
             string prompt,
             XUserClaimsInfoDto userInfo,
             CancellationToken cancellationToken = default
@@ -123,7 +124,7 @@ namespace xAiService.Services
 
             //
             var tokens = ToTokenResponse(userInfo);
-            var result = StreamData<string>(
+            var result = await StreamData(
                 tokens: tokens,
                 addXPoweredValue: true,
                 endpoint: XAiApiEndpoint.Stream,
diff --git a/xAiService.csproj b/xAiService.csproj
index 848fedf..9c6f89f 100644
--- a/xAiService.csproj
+++ b/xAiService.csproj
@@ -22,12 +22,13 @@
 
   <!-- Local Projects -->
    <ItemGroup>
+    <ProjectReference Include="..\xCommons\xCommons.csproj" />
     <ProjectReference Include="..\xIdentityHelper\xIdentityHelper.csproj" />
    </ItemGroup>
 
   <!-- Local Modules -->
   <ItemGroup>
-    <PackageReference Include="xDashboard.xCommons" Version="1.0.0" />
+    <!-- <PackageReference Include="xDashboard.xCommons" Version="1.0.0" /> -->
     <!-- <PackageReference Include="xDashboard.xIdentityService" Version="1.0.0" /> -->
   </ItemGroup>