Release AI runtime availability fix

addon/components/ai-header-tray-button.js

@@ -8,7 +8,7 @@ export default class AiHeaderTrayButtonComponent extends Component {
 
     constructor() {
         super(...arguments);
-        this.ai.loadConfig.perform();
+        this.ai.loadStatus.perform();
     }
 
     @action togglePrompt() {

addon/services/ai.js

@@ -69,6 +69,19 @@ export default class AiService extends Service {
         }
     }
 
+    @task *loadStatus() {
+        try {
+            const response = yield this.fetch.get('status', {}, { namespace: 'ai/int/v1' });
+            this.applyConfig({ enabled: response.enabled === true });
+
+            return this.config;
+        } catch (error) {
+            this.applyConfig({ enabled: false });
+
+            return this.config;
+        }
+    }
+
     @task *submit(prompt, attachments = []) {
         if (!this.isEnabled) {
             return null;

package.json

@@ -1,6 +1,6 @@
 {
     "name": "@fleetbase/ai-engine",
-    "version": "0.0.1",
+    "version": "0.0.2",
     "description": "Fleetbase AI operations copilot and task automation module",
     "keywords": [
         "fleetbase-extension",
@@ -44,7 +44,7 @@
     "dependencies": {
         "@babel/core": "^7.23.2",
         "@fleetbase/ember-core": "^0.3.24",
-        "@fleetbase/ember-ui": "^0.3.39",
+        "@fleetbase/ember-ui": "^0.3.41",
         "@fortawesome/ember-fontawesome": "^2.0.0",
         "@fortawesome/fontawesome-svg-core": "6.4.0",
         "@fortawesome/free-brands-svg-icons": "6.4.0",

server/src/Http/Controllers/Internal/AiConfigController.php

@@ -7,9 +7,19 @@
 use Fleetbase\Http\Controllers\Controller;
 use Fleetbase\Http\Requests\AdminRequest;
 use Fleetbase\Models\Setting;
+use Illuminate\Http\Request;
 
 class AiConfigController extends Controller
 {
+    public function status(Request $request, AiProviderManager $providers)
+    {
+        $config = $providers->normalizeConfig(Setting::system('ai', $providers->defaultConfig()));
+
+        return response()->json([
+            'enabled' => (bool) data_get($config, 'enabled', false),
+        ]);
+    }
+
     public function show(AdminRequest $request, AiProviderManager $providers)
     {
         $config = $providers->normalizeConfig(Setting::system('ai', $providers->defaultConfig()));

server/src/Http/Controllers/Internal/AiTaskController.php

@@ -5,6 +5,7 @@
 use Fleetbase\Ai\Models\AiTask;
 use Fleetbase\Ai\Services\AiTaskService;
 use Fleetbase\Http\Controllers\Controller;
+use Fleetbase\Models\Setting;
 use Illuminate\Http\Request;
 
 class AiTaskController extends Controller
@@ -22,6 +23,8 @@ public function index(Request $request)
 
     public function store(Request $request, AiTaskService $tasks)
     {
+        $this->abortIfAiDisabled();
+
         $request->validate([
             'prompt'        => ['required', 'string', 'max:12000'],
             'session_uuid'  => ['nullable', 'string'],
@@ -39,13 +42,17 @@ public function show(string $id)
 
     public function preview(string $id, Request $request, AiTaskService $tasks)
     {
+        $this->abortIfAiDisabled();
+
         $task = $this->findTask($id);
 
         return response()->json(['task' => $tasks->refreshPreview($task, $request->input('action_key'), $request->input('input', []))->load('session')]);
     }
 
     public function apply(string $id, Request $request, AiTaskService $tasks)
     {
+        $this->abortIfAiDisabled();
+
         $task = $this->findTask($id);
 
         return response()->json(['task' => $tasks->apply($task, $request->input('action_key'), $request->input('input', []))->load('session')]);
@@ -87,4 +94,9 @@ protected function findTask(string $id): AiTask
             })
             ->firstOrFail();
     }
+
+    protected function abortIfAiDisabled(): void
+    {
+        abort_unless((bool) data_get(Setting::system('ai', []), 'enabled', false), 403, 'Fleetbase AI is disabled.');
+    }
 }

server/src/routes.php

@@ -27,6 +27,7 @@ function ($router) {
                 $router->group(
                     ['prefix' => 'v1', 'middleware' => ['fleetbase.protected']],
                     function ($router) {
+                        $router->get('status', 'Internal\AiConfigController@status');
                         $router->get('config', 'Internal\AiConfigController@show');
                         $router->post('config', 'Internal\AiConfigController@store');
                         $router->post('test-provider', 'Internal\AiConfigController@testProvider');