mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-21 21:31:32 +03:00
808b8ffa12
* feat: refecth chats list on first message * style: change chat items colors
118 lines
3.3 KiB
TypeScript
118 lines
3.3 KiB
TypeScript
/* eslint-disable max-lines */
|
|
import { useQueryClient } from "@tanstack/react-query";
|
|
import { AxiosError } from "axios";
|
|
import { useParams, useRouter } from "next/navigation";
|
|
import { useState } from "react";
|
|
import { useTranslation } from "react-i18next";
|
|
|
|
import { getChatsConfigFromLocalStorage } from "@/lib/api/chat/chat.local";
|
|
import { CHATS_DATA_KEY } from "@/lib/api/chat/config";
|
|
import { useChatApi } from "@/lib/api/chat/useChatApi";
|
|
import { useChatContext } from "@/lib/context";
|
|
import { useBrainContext } from "@/lib/context/BrainProvider/hooks/useBrainContext";
|
|
import { useToast } from "@/lib/hooks";
|
|
import { useEventTracking } from "@/services/analytics/june/useEventTracking";
|
|
|
|
import { useQuestion } from "./useQuestion";
|
|
import { ChatQuestion } from "../types";
|
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
|
export const useChat = () => {
|
|
const { track } = useEventTracking();
|
|
const queryClient = useQueryClient();
|
|
|
|
const params = useParams();
|
|
const [chatId, setChatId] = useState<string | undefined>(
|
|
params?.chatId as string | undefined
|
|
);
|
|
const [generatingAnswer, setGeneratingAnswer] = useState(false);
|
|
const router = useRouter();
|
|
const { messages } = useChatContext();
|
|
const { currentBrain, currentPromptId, currentBrainId } = useBrainContext();
|
|
const { publish } = useToast();
|
|
const { createChat } = useChatApi();
|
|
|
|
const { addStreamQuestion } = useQuestion();
|
|
const { t } = useTranslation(["chat"]);
|
|
|
|
const addQuestion = async (question: string, callback?: () => void) => {
|
|
if (question === "") {
|
|
publish({
|
|
variant: "danger",
|
|
text: t("ask"),
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
try {
|
|
setGeneratingAnswer(true);
|
|
|
|
let currentChatId = chatId;
|
|
|
|
let shouldUpdateUrl = false;
|
|
|
|
//if chatId is not set, create a new chat. Chat name is from the first question
|
|
if (currentChatId === undefined) {
|
|
const chatName = question.split(" ").slice(0, 3).join(" ");
|
|
const chat = await createChat(chatName);
|
|
currentChatId = chat.chat_id;
|
|
setChatId(currentChatId);
|
|
shouldUpdateUrl = true;
|
|
void queryClient.invalidateQueries({
|
|
queryKey: [CHATS_DATA_KEY],
|
|
});
|
|
}
|
|
|
|
void track("QUESTION_ASKED", {
|
|
brainId: currentBrainId,
|
|
promptId: currentPromptId,
|
|
});
|
|
|
|
const chatConfig = getChatsConfigFromLocalStorage();
|
|
|
|
const chatQuestion: ChatQuestion = {
|
|
model: chatConfig?.model,
|
|
question,
|
|
temperature: chatConfig?.temperature,
|
|
max_tokens: chatConfig?.maxTokens,
|
|
brain_id: currentBrain?.id,
|
|
prompt_id: currentPromptId ?? undefined,
|
|
};
|
|
|
|
await addStreamQuestion(currentChatId, chatQuestion);
|
|
|
|
callback?.();
|
|
|
|
if (shouldUpdateUrl) {
|
|
router.replace(`/chat/${currentChatId}`);
|
|
}
|
|
} catch (error) {
|
|
console.error({ error });
|
|
|
|
if ((error as AxiosError).response?.status === 429) {
|
|
publish({
|
|
variant: "danger",
|
|
text: t("limit_reached", { ns: "chat" }),
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
publish({
|
|
variant: "danger",
|
|
text: t("error_occurred", { ns: "chat" }),
|
|
});
|
|
} finally {
|
|
setGeneratingAnswer(false);
|
|
}
|
|
};
|
|
|
|
return {
|
|
messages,
|
|
addQuestion,
|
|
generatingAnswer,
|
|
chatId,
|
|
};
|
|
};
|