mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-17 11:21:35 +03:00
b5e2d5ad9c
# Description - Implement Icon Component - Implement TextButton Component - Change Add Brain Button And Set it in the Search Page - Fix Errors When sending empty message - Change EsLint rules ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate):
93 lines
2.3 KiB
TypeScript
93 lines
2.3 KiB
TypeScript
import { useTranslation } from "react-i18next";
|
|
|
|
import { useChatContext } from "@/lib/context";
|
|
import { useBrainContext } from "@/lib/context/BrainProvider/hooks/useBrainContext";
|
|
import { useFetch, useToast } from "@/lib/hooks";
|
|
|
|
import { useHandleStream } from "./useHandleStream";
|
|
|
|
import { ChatQuestion } from "../types";
|
|
import { generatePlaceHolderMessage } from "../utils/generatePlaceHolderMessage";
|
|
|
|
interface UseChatService {
|
|
addStreamQuestion: (
|
|
chatId: string,
|
|
chatQuestion: ChatQuestion
|
|
) => Promise<void>;
|
|
}
|
|
|
|
export const useQuestion = (): UseChatService => {
|
|
const { fetchInstance } = useFetch();
|
|
const { currentBrain } = useBrainContext();
|
|
|
|
const { t } = useTranslation(["chat"]);
|
|
const { publish } = useToast();
|
|
const { handleStream } = useHandleStream();
|
|
const { removeMessage, updateStreamingHistory } = useChatContext();
|
|
|
|
const handleFetchError = async (response: Response) => {
|
|
if (response.status === 429) {
|
|
publish({
|
|
variant: "danger",
|
|
text: t("tooManyRequests", { ns: "chat" }),
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
const errorMessage = (await response.json()) as { detail: string };
|
|
publish({
|
|
variant: "danger",
|
|
text: errorMessage.detail,
|
|
});
|
|
};
|
|
|
|
const addStreamQuestion = async (
|
|
chatId: string,
|
|
chatQuestion: ChatQuestion
|
|
): Promise<void> => {
|
|
const headers = {
|
|
"Content-Type": "application/json",
|
|
Accept: "text/event-stream",
|
|
};
|
|
|
|
const placeHolderMessage = generatePlaceHolderMessage({
|
|
user_message: chatQuestion.question ?? "",
|
|
chat_id: chatId,
|
|
});
|
|
updateStreamingHistory(placeHolderMessage);
|
|
|
|
const body = JSON.stringify(chatQuestion);
|
|
|
|
try {
|
|
const response = await fetchInstance.post(
|
|
`/chat/${chatId}/question/stream?brain_id=${currentBrain?.id ?? ""}`,
|
|
body,
|
|
headers
|
|
);
|
|
if (!response.ok) {
|
|
void handleFetchError(response);
|
|
|
|
return;
|
|
}
|
|
|
|
if (response.body === null) {
|
|
throw new Error(t("resposeBodyNull", { ns: "chat" }));
|
|
}
|
|
|
|
await handleStream(response.body.getReader(), () =>
|
|
removeMessage(placeHolderMessage.message_id)
|
|
);
|
|
} catch (error) {
|
|
publish({
|
|
variant: "danger",
|
|
text: String(error),
|
|
});
|
|
}
|
|
};
|
|
|
|
return {
|
|
addStreamQuestion,
|
|
};
|
|
};
|