From 509acc1488743362751d6c086052d6626c4e7a94 Mon Sep 17 00:00:00 2001
From: Hadi Khazaee Asl <hadi_khazaee_asl@yahoo.com>
Date: Thu, 16 Apr 2026 16:02:26 +0330
Subject: [PATCH] last ...

---
 Controllers/TestIdentity.cs        | 4 ++--
 Controllers/V1/AIController.cs     | 5 +++--
 Controllers/V1/OllamaController.cs | 4 ++--
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/Controllers/TestIdentity.cs b/Controllers/TestIdentity.cs
index fde04f3..df13611 100644
--- a/Controllers/TestIdentity.cs
+++ b/Controllers/TestIdentity.cs
@@ -138,7 +138,7 @@ namespace xAiApi.Controllers
         public ActionResult<string> HiAgent()
         {
             //
-            var result = $"Hi Admin: {User.Identity.Name} ...";
+            var result = $"Hi Agent: {User.Identity.Name} ...";
 
             //
             return Ok(result);
@@ -153,7 +153,7 @@ namespace xAiApi.Controllers
         public ActionResult<string> HiEnabledAgent()
         {
             //
-            var result = $"Hi Admin: {User.Identity.Name} is Enabled ...";
+            var result = $"Hi Agent: {User.Identity.Name} is Enabled ...";
 
             //
             return Ok(result);
diff --git a/Controllers/V1/AIController.cs b/Controllers/V1/AIController.cs
index 2d7f480..73c9377 100644
--- a/Controllers/V1/AIController.cs
+++ b/Controllers/V1/AIController.cs
@@ -10,6 +10,7 @@ using xAiApi.Base;
 using xCommons.Configurations;
 using xCommons.Extensions;
 using xCommons.Providers;
+using xIdentityHelper;
 using xIdentityService.Interfaces;
 
 namespace xAiApi.Controllers.V1
@@ -36,8 +37,8 @@ namespace xAiApi.Controllers.V1
 
         //
         #region Actions ...
-        [AllowAnonymous]
         [HttpGet("AskAI")]
+        [Authorize(Policy = XPolicies.EnabledUser)]
         public async Task<ActionResult<string>> AskAI(
             [FromQuery] string prompt,
             CancellationToken cancellationToken = default
@@ -65,8 +66,8 @@ namespace xAiApi.Controllers.V1
             }
         }
 
-        [AllowAnonymous]
         [HttpGet("AskAIStream")]
+        [Authorize(Policy = XPolicies.EnabledUser)]
         public async Task AskAIStream(
             [FromQuery] string prompt,
             CancellationToken cancellationToken = default
diff --git a/Controllers/V1/OllamaController.cs b/Controllers/V1/OllamaController.cs
index 0810ab8..0d1a32e 100644
--- a/Controllers/V1/OllamaController.cs
+++ b/Controllers/V1/OllamaController.cs
@@ -35,7 +35,7 @@ namespace xAiApi.Controllers.V1
         /// </summary>
         /// <returns></returns>
         [HttpGet("Version")]
-        [Authorize(Policy = XPolicies.Admin)]
+        [Authorize(Policy = XPolicies.EnabledAdmin)]
         public async Task<ActionResult<string>> GetVersion()
         {
             //
@@ -61,7 +61,7 @@ namespace xAiApi.Controllers.V1
         /// </summary>
         /// <returns></returns>
         [HttpGet("Models")]
-        [Authorize(Policy = XPolicies.Admin)]
+        [Authorize(Policy = XPolicies.EnabledAdmin)]
         public async Task<ActionResult<IEnumerable<string>>> GetModels()
         {
             //