@truedat/ai 8.14.2 → 8.14.3

package/src/hooks/__tests__/useIndices.spec.js

@@ -0,0 +1,75 @@
+import useSWR from "swr";
+import useSWRMutations from "swr/mutation";
+import { renderHook } from "@testing-library/react";
+import { apiJsonPost, apiJsonDelete } from "@truedat/core/services/api";
+import {
+  useIndices,
+  useIndexCreate,
+  useIndexDelete,
+  useIndexEnable,
+  useIndexDisable,
+} from "../useIndices";
+
+jest.mock("swr", () => ({
+  __esModule: true,
+  ...jest.requireActual("swr"),
+  default: jest.fn(() => ({
+    data: { data: { data: [{ id: 1 }] } },
+    error: null,
+    mutate: jest.fn(),
+  })),
+}));
+
+jest.mock("swr/mutation", () => ({
+  __esModule: true,
+  ...jest.requireActual("swr/mutation"),
+  default: jest.fn(),
+}));
+
+jest.mock("@truedat/core/services/api", () => ({
+  __esModule: true,
+  ...jest.requireActual("@truedat/core/services/api"),
+  apiJsonPost: jest.fn(),
+  apiJsonDelete: jest.fn(),
+}));
+
+const fetcherOf = (hook) => {
+  renderHook(() => hook());
+  const [key, fetcher] = useSWRMutations.mock.calls.at(-1);
+  return [key, fetcher];
+};
+
+describe("useIndices", () => {
+  beforeEach(() => jest.clearAllMocks());
+
+  it("lists indices from /api/indices", () => {
+    const { result } = renderHook(() => useIndices());
+    expect(useSWR.mock.calls[0][0]).toBe("/api/indices");
+    expect(result.current.data).toEqual([{ id: 1 }]);
+    expect(result.current.loading).toBe(false);
+  });
+
+  it("creates through POST /api/indices", () => {
+    const [key, fetcher] = fetcherOf(useIndexCreate);
+    fetcher(key, { arg: { index: { embedding: "m" } } });
+    expect(key).toBe("/api/indices");
+    expect(apiJsonPost).toHaveBeenCalledWith("/api/indices", {
+      index: { embedding: "m" },
+    });
+  });
+
+  it("deletes through DELETE /api/indices/:id", () => {
+    const [key, fetcher] = fetcherOf(useIndexDelete);
+    fetcher(key, { arg: { id: 3 } });
+    expect(apiJsonDelete).toHaveBeenCalledWith("/api/indices/3");
+  });
+
+  it("enables and disables through the action endpoints", () => {
+    const [ek, ef] = fetcherOf(useIndexEnable);
+    ef(ek, { arg: { id: 3 } });
+    expect(apiJsonPost).toHaveBeenCalledWith("/api/indices/3/enable");
+    const [dk, df] = fetcherOf(useIndexDisable);
+    df(dk, { arg: { id: 4 } });
+    expect(apiJsonPost).toHaveBeenCalledWith("/api/indices/4/disable");
+  });
+});

package/src/hooks/__tests__/useProviders.spec.js

@@ -15,6 +15,8 @@ import {
   useProviderCreate,
   useProviderUpdate,
   useProviderDelete,
+  useProviderEnable,
+  useProviderDisable,
 } from "../useProviders";
 
 jest.mock("swr", () => ({
@@ -98,4 +100,16 @@ describe("useProviders", () => {
     func(url, { arg });
     expect(apiJsonDelete).toHaveBeenCalledWith(url, arg);
   });
+  it("enables and disables through the action endpoints (12-6)", () => {
+    renderHook(() => useProviderEnable());
+    const [ek, ef] = _.last(useSWRMutations.mock.calls);
+    expect(ek).toBe("/api/providers/:id/enable");
+    ef(ek, { arg: { id: 3 } });
+    expect(apiJsonPost).toHaveBeenCalledWith("/api/providers/3/enable");
+    renderHook(() => useProviderDisable());
+    const [dk, df] = _.last(useSWRMutations.mock.calls);
+    expect(dk).toBe("/api/providers/:id/disable");
+    df(dk, { arg: { id: 4 } });
+    expect(apiJsonPost).toHaveBeenCalledWith("/api/providers/4/disable");
+  });
 });

package/src/hooks/useCustomInstructions.js

@@ -0,0 +1,22 @@
+import useSWR from "swr";
+import useSWRMutations from "swr/mutation";
+import { apiJson, apiJsonDelete, apiJsonPut } from "@truedat/core/services/api";
+import { API_AGENT_LAYER_CUSTOM_INSTRUCTIONS } from "../api";
+
+export const useCustomInstructions = () => {
+  const { data, error, mutate } = useSWR(
+    API_AGENT_LAYER_CUSTOM_INSTRUCTIONS,
+    apiJson,
+  );
+  return { data: data?.data?.data, error, loading: !error && !data, mutate };
+};
+
+export const useCustomInstructionsUpdate = () =>
+  useSWRMutations(API_AGENT_LAYER_CUSTOM_INSTRUCTIONS, (url, { arg }) =>
+    apiJsonPut(url, arg),
+  );
+
+export const useCustomInstructionsDelete = () =>
+  useSWRMutations(API_AGENT_LAYER_CUSTOM_INSTRUCTIONS, (url) =>
+    apiJsonDelete(url),
+  );

package/src/hooks/useIndices.js

@@ -0,0 +1,37 @@
+import { compile } from "path-to-regexp";
+import useSWR from "swr";
+import useSWRMutations from "swr/mutation";
+import {
+  apiJson,
+  apiJsonPost,
+  apiJsonDelete,
+} from "@truedat/core/services/api";
+import {
+  API_INDICES,
+  API_INDEX,
+  API_INDEX_ENABLE,
+  API_INDEX_DISABLE,
+} from "../api";
+
+export const useIndices = () => {
+  const { data, error, mutate } = useSWR(API_INDICES, apiJson);
+  return { data: data?.data?.data, error, loading: !error && !data, mutate };
+};
+
+export const useIndexCreate = () =>
+  useSWRMutations(API_INDICES, (url, { arg }) => apiJsonPost(url, arg));
+
+export const useIndexDelete = () =>
+  useSWRMutations(API_INDEX, (url, { arg: { id } }) =>
+    apiJsonDelete(compile(url)({ id: `${id}` })),
+  );
+
+export const useIndexEnable = () =>
+  useSWRMutations(API_INDEX_ENABLE, (url, { arg: { id } }) =>
+    apiJsonPost(compile(url)({ id: `${id}` })),
+  );
+
+export const useIndexDisable = () =>
+  useSWRMutations(API_INDEX_DISABLE, (url, { arg: { id } }) =>
+    apiJsonPost(compile(url)({ id: `${id}` })),
+  );

package/src/hooks/useProviders.js

@@ -7,7 +7,13 @@ import {
   apiJsonPatch,
   apiJsonDelete,
 } from "@truedat/core/services/api";
-import { API_PROVIDERS, API_PROVIDER, API_PROVIDER_CHAT } from "../api";
+import {
+  API_PROVIDERS,
+  API_PROVIDER,
+  API_PROVIDER_CHAT,
+  API_PROVIDER_ENABLE,
+  API_PROVIDER_DISABLE,
+} from "../api";
 
 export const useProviders = () => {
   const { data, error, mutate } = useSWR(API_PROVIDERS, apiJson);
@@ -36,3 +42,13 @@ export const useProviderDelete = (payload) => {
   const url = compile(API_PROVIDER)({ id: `${id}` });
   return useSWRMutations(url, (url, { arg }) => apiJsonDelete(url, arg));
 };
+
+export const useProviderEnable = () =>
+  useSWRMutations(API_PROVIDER_ENABLE, (url, { arg: { id } }) =>
+    apiJsonPost(compile(url)({ id: `${id}` })),
+  );
+
+export const useProviderDisable = () =>
+  useSWRMutations(API_PROVIDER_DISABLE, (url, { arg: { id } }) =>
+    apiJsonPost(compile(url)({ id: `${id}` })),
+  );

package/src/serverErrors.js

@@ -0,0 +1,2 @@
+// The `errors` map of a failed request's response body (`{field: [message]}`), or `{}`.
+export const serverErrors = (error) => error?.response?.data?.errors || {};