1
1
mirror of https://github.com/StanGirard/quivr.git synced 2024-12-23 03:12:24 +03:00
quivr/frontend/lib/api/brain/brain.ts
Mamadou DICKO 3529222b95
Brain management 4 ()
* feat: add <ApiKeyConfig/>

* feat(SDK): add update brain

* feat: add removeUndefined helper

* feat: remove unnecessary autofocus flag

* add brain settings tab

* ui: add tab delimitor

* feat: improve ux
2023-07-25 23:12:46 +02:00

139 lines
3.8 KiB
TypeScript

/* eslint-disable max-lines */
import { AxiosInstance } from "axios";
import { BrainRoleType } from "@/lib/components/NavBar/components/NavItems/components/BrainsDropDown/components/BrainActions/types";
import {
BackendMinimalBrainForUser,
Brain,
MinimalBrainForUser,
} from "@/lib/context/BrainProvider/types";
import { Document } from "@/lib/types/Document";
import {
CreateBrainInput,
SubscriptionUpdatableProperties,
UpdateBrainInput,
} from "./types";
import { mapBackendMinimalBrainToMinimalBrain } from "./utils/mapBackendMinimalBrainToMinimalBrain";
import {
BackendSubscription,
mapSubscriptionToBackendSubscription,
} from "./utils/mapSubscriptionToBackendSubscription";
import { mapSubscriptionUpdatablePropertiesToBackendSubscriptionUpdatableProperties } from "./utils/mapSubscriptionUpdatablePropertiesToBackendSubscriptionUpdatableProperties";
export const getBrainDocuments = async (
brainId: string,
axiosInstance: AxiosInstance
): Promise<Document[]> => {
const response = await axiosInstance.get<{ documents: Document[] }>(
`/explore/?brain_id=${brainId}`
);
return response.data.documents;
};
export const createBrain = async (
brain: CreateBrainInput,
axiosInstance: AxiosInstance
): Promise<MinimalBrainForUser> => {
return mapBackendMinimalBrainToMinimalBrain(
(await axiosInstance.post<BackendMinimalBrainForUser>(`/brains/`, brain))
.data
);
};
export const getBrain = async (
brainId: string,
axiosInstance: AxiosInstance
): Promise<Brain | undefined> => {
const brain = (
await axiosInstance.get<Brain | undefined>(`/brains/${brainId}/`)
).data;
return brain;
};
export const deleteBrain = async (
brainId: string,
axiosInstance: AxiosInstance
): Promise<void> => {
await axiosInstance.delete(`/brains/${brainId}/subscription`);
};
export const getDefaultBrain = async (
axiosInstance: AxiosInstance
): Promise<MinimalBrainForUser | undefined> => {
return mapBackendMinimalBrainToMinimalBrain(
(await axiosInstance.get<BackendMinimalBrainForUser>(`/brains/default/`))
.data
);
};
export const getBrains = async (
axiosInstance: AxiosInstance
): Promise<MinimalBrainForUser[]> => {
const { brains } = (
await axiosInstance.get<{ brains: BackendMinimalBrainForUser[] }>(
`/brains/`
)
).data;
return brains.map(mapBackendMinimalBrainToMinimalBrain);
};
export type Subscription = { email: string; role: BrainRoleType };
export const addBrainSubscriptions = async (
brainId: string,
subscriptions: Subscription[],
axiosInstance: AxiosInstance
): Promise<void> => {
await axiosInstance.post(
`/brains/${brainId}/subscription`,
subscriptions.map(mapSubscriptionToBackendSubscription)
);
};
export const getBrainUsers = async (
brainId: string,
axiosInstance: AxiosInstance
): Promise<Subscription[]> => {
const brainsUsers = (
await axiosInstance.get<BackendSubscription[]>(`/brains/${brainId}/users`)
).data;
return brainsUsers.map((brainUser) => ({
email: brainUser.email,
role: brainUser.rights,
}));
};
export const updateBrainAccess = async (
brainId: string,
userEmail: string,
subscription: SubscriptionUpdatableProperties,
axiosInstance: AxiosInstance
): Promise<void> => {
await axiosInstance.put(`/brains/${brainId}/subscription`, {
...mapSubscriptionUpdatablePropertiesToBackendSubscriptionUpdatableProperties(
subscription
),
email: userEmail,
});
};
export const setAsDefaultBrain = async (
brainId: string,
axiosInstance: AxiosInstance
): Promise<void> => {
await axiosInstance.post(`/brains/${brainId}/default`);
};
export const updateBrain = async (
brainId: string,
brain: UpdateBrainInput,
axiosInstance: AxiosInstance
): Promise<void> => {
await axiosInstance.put(`/brains/${brainId}/`, brain);
};