@truedat/ai 8.8.5 → 8.8.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@truedat/ai",
-  "version": "8.8.5",
+  "version": "8.8.7",
   "description": "Truedat Web Artificial Intelligence package",
   "sideEffects": false,
   "module": "src/index.js",
@@ -50,7 +50,7 @@
     "@testing-library/jest-dom": "^6.6.3",
     "@testing-library/react": "^16.3.0",
     "@testing-library/user-event": "^14.6.1",
-    "@truedat/test": "8.8.5",
+    "@truedat/test": "8.8.7",
     "identity-obj-proxy": "^3.0.0",
     "jest": "^29.7.0",
     "redux-saga-test-plan": "^4.0.6"
@@ -80,5 +80,5 @@
     "semantic-ui-react": "^3.0.0-beta.2",
     "swr": "^2.3.3"
   },
-  "gitHead": "d951a03df40f742b8604ae48f0b2fff197c3f5ee"
+  "gitHead": "ece3f2e166776ff7827198a134bad5a28b029dc7"
 }

package/src/components/AiRoutes.js

@@ -1,6 +1,7 @@
 import React from "react";
-import { Route, Routes } from "react-router";
+import { Navigate, Route, Routes } from "react-router";
 import { ProtectedRoute } from "@truedat/core/router";
+import { useCanAssistant } from "./assistant/hooks/useAgentConversation";
 
 import ResourceMappings from "./resourceMappings/ResourceMappings";
 import Prompts from "./prompts/Prompts";
@@ -12,6 +13,12 @@ import ActionEdit from "./actions/ActionEdit";
 import ActionNew from "./actions/ActionNew";
 import AssistantPage from "./assistant/AssistantPage";
 
+const AssistantPageGuard = () => {
+  const canAssistant = useCanAssistant();
+  if (!canAssistant) return <Navigate to="/" replace />;
+  return <AssistantPage />;
+};
+
 export default function AiRoutes() {
   return (
     <Routes>
@@ -53,7 +60,7 @@ export default function AiRoutes() {
         <Route path={":id"} element={<Action />} />
       </Route>
 
-      <Route path={"ai-assistant"} element={<AssistantPage />} />
+      <Route path={"ai-assistant"} element={<AssistantPageGuard />} />
 
       <Route path="*" element={null} />
     </Routes>

package/src/components/assistant/hooks/__tests__/useAgentConversation.spec.js

@@ -0,0 +1,48 @@
+import useSWR from "swr";
+import { renderHook } from "@testing-library/react";
+
+import { API_AGENT_LAYER_CONVERSATIONS } from "../../../../api";
+import { useCanAssistant } from "../useAgentConversation";
+
+jest.mock("swr", () => ({
+  __esModule: true,
+  default: jest.fn(),
+}));
+
+describe("useCanAssistant", () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
+  it("calls useSWR with the correct endpoint and fetcher", () => {
+    useSWR.mockReturnValue({ data: undefined, error: null });
+    renderHook(() => useCanAssistant());
+
+    expect(useSWR).toHaveBeenCalledWith(
+      API_AGENT_LAYER_CONVERSATIONS,
+      expect.any(Function),
+      expect.any(Object),
+    );
+  });
+
+  it("returns true when OPTIONS returns Allow header with GET", () => {
+    useSWR.mockReturnValue({ data: true, error: null });
+
+    const { result } = renderHook(() => useCanAssistant());
+    expect(result.current).toBe(true);
+  });
+
+  it("returns false when OPTIONS returns Allow header without GET", () => {
+    useSWR.mockReturnValue({ data: false, error: null });
+
+    const { result } = renderHook(() => useCanAssistant());
+    expect(result.current).toBe(false);
+  });
+
+  it("returns false when data is undefined (loading state)", () => {
+    useSWR.mockReturnValue({ data: undefined, error: null });
+
+    const { result } = renderHook(() => useCanAssistant());
+    expect(result.current).toBe(false);
+  });
+});

package/src/components/assistant/hooks/useAgentConversation.js

@@ -1,8 +1,24 @@
 import { useCallback, useEffect, useState } from "react";
-
-import { apiJsonPost } from "@truedat/core/services/api";
-
-import { API_AGENT_LAYER_CONVERSATION } from "../../../api";
+import useSWR from "swr";
+import { apiJsonOptions, apiJsonPost } from "@truedat/core/services/api";
+import {
+  API_AGENT_LAYER_CONVERSATIONS,
+  API_AGENT_LAYER_CONVERSATION,
+} from "../../../api";
+
+const optionsFetcher = (url) =>
+  apiJsonOptions(url).then(({ headers }) => {
+    const allow = headers?.allow || "";
+    return allow.includes("GET");
+  });
+
+export const useCanAssistant = () => {
+  const { data } = useSWR(API_AGENT_LAYER_CONVERSATIONS, optionsFetcher, {
+    revalidateOnFocus: false,
+    shouldRetryOnError: false,
+  });
+  return data ?? false;
+};
 
 const useAgentConversation = (isOpen, initialId = null) => {
   const [conversationId, setConversationId] = useState(