James 6f50424917 allow user to query for HF models
Signed-off-by: James <james@jan.ai>
2023-10-06 16:46:47 +07:00

40 lines
1.1 KiB
TypeScript

"use client";
import React from "react";
import SecondaryButton from "../SecondaryButton";
import { useAtomValue, useSetAtom } from "jotai";
import {
MainViewState,
setMainViewStateAtom,
} from "@/_helpers/atoms/MainView.atom";
import { currentProductAtom } from "@/_helpers/atoms/Model.atom";
import useCreateConversation from "@/_hooks/useCreateConversation";
import useInitModel from "@/_hooks/useInitModel";
import { Product } from "@/_models/Product";
const NewChatButton: React.FC = () => {
const activeModel = useAtomValue(currentProductAtom);
const setMainView = useSetAtom(setMainViewStateAtom);
const { requestCreateConvo } = useCreateConversation();
const { initModel } = useInitModel();
const onClick = () => {
if (!activeModel) {
setMainView(MainViewState.ConversationEmptyModel);
} else {
createConversationAndInitModel(activeModel);
}
};
const createConversationAndInitModel = async (model: Product) => {
await requestCreateConvo(model);
await initModel(model);
};
return (
<SecondaryButton title={"New Chat"} onClick={onClick} className="my-5 mx-3" />
);
};
export default NewChatButton;