mirror of https://github.com/usememos/memos
chore: remove memo chat components (#2073)
boojack
2 years ago
committed by
GitHub
parent
1ce82ba0d6
commit
aa26cc30d7
@ -1,37 +0,0 @@
|
|||||||
import Textarea from "@mui/joy/Textarea/Textarea";
|
|
||||||
import Icon from "@/components/Icon";
|
|
||||||
import { useTranslate } from "@/utils/i18n";
|
|
||||||
|
|
||||||
interface Props {
|
|
||||||
question: string;
|
|
||||||
handleQuestionTextareaChange: any;
|
|
||||||
setIsInIME: any;
|
|
||||||
handleKeyDown: any;
|
|
||||||
handleSendQuestionButtonClick: any;
|
|
||||||
}
|
|
||||||
|
|
||||||
const ChatInput = ({ question, handleQuestionTextareaChange, setIsInIME, handleKeyDown, handleSendQuestionButtonClick }: Props) => {
|
|
||||||
const t = useTranslate();
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="w-full relative mt-4">
|
|
||||||
<Textarea
|
|
||||||
className="w-full"
|
|
||||||
placeholder={t("memo-chat.placeholder")}
|
|
||||||
value={question}
|
|
||||||
minRows={1}
|
|
||||||
maxRows={5}
|
|
||||||
onChange={handleQuestionTextareaChange}
|
|
||||||
onCompositionStart={() => setIsInIME(true)}
|
|
||||||
onCompositionEnd={() => setIsInIME(false)}
|
|
||||||
onKeyDown={handleKeyDown}
|
|
||||||
/>
|
|
||||||
<Icon.Send
|
|
||||||
className="cursor-pointer w-7 p-1 h-auto rounded-md bg-gray-100 dark:bg-zinc-800 absolute right-2 bottom-1.5 shadow hover:opacity-80"
|
|
||||||
onClick={handleSendQuestionButtonClick}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
export default ChatInput;
|
|
||||||
@ -1,60 +0,0 @@
|
|||||||
import toast from "react-hot-toast";
|
|
||||||
import Icon from "@/components/Icon";
|
|
||||||
import { marked } from "@/labs/marked";
|
|
||||||
import { useMemoStore } from "@/store/module";
|
|
||||||
import { Message } from "@/store/v1/message";
|
|
||||||
import { useTranslate } from "@/utils/i18n";
|
|
||||||
import Dropdown from "../kit/Dropdown";
|
|
||||||
|
|
||||||
interface MessageProps {
|
|
||||||
index: number;
|
|
||||||
message: Message;
|
|
||||||
}
|
|
||||||
|
|
||||||
const ChatMessage = ({ index, message }: MessageProps) => {
|
|
||||||
const memoStore = useMemoStore();
|
|
||||||
const t = useTranslate();
|
|
||||||
|
|
||||||
const handleSaveAsMemos = async () => {
|
|
||||||
await memoStore.createMemo({
|
|
||||||
content: message.content,
|
|
||||||
visibility: "PRIVATE",
|
|
||||||
resourceIdList: [],
|
|
||||||
relationList: [],
|
|
||||||
});
|
|
||||||
toast.success(t("memo-chat.save-as-memo-success"));
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div key={index} className="w-full flex flex-col justify-start items-start space-y-2">
|
|
||||||
{message.role === "user" ? (
|
|
||||||
<div className="w-full flex flex-row justify-end items-start pl-6">
|
|
||||||
<span className="word-break shadow rounded-lg rounded-tr-none px-3 py-2 opacity-80 bg-white dark:bg-zinc-800">
|
|
||||||
{message.content}
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<div className="w-full flex flex-row justify-start items-start pr-8 space-x-2">
|
|
||||||
<Icon.Bot className="mt-2 shrink-0 mr-1 w-6 h-auto opacity-80" />
|
|
||||||
<div className="memo-content-wrapper !w-auto flex flex-col justify-start items-start shadow rounded-lg rounded-tl-none px-3 py-2 bg-white dark:bg-zinc-800">
|
|
||||||
<div className="memo-content-text">{marked(message.content)}</div>
|
|
||||||
</div>
|
|
||||||
<Dropdown
|
|
||||||
actions={
|
|
||||||
<>
|
|
||||||
<button
|
|
||||||
className="w-full m-auto text-left text-sm whitespace-nowrap leading-6 py-1 px-3 cursor-pointer rounded hover:bg-gray-100 dark:hover:bg-zinc-600"
|
|
||||||
onClick={() => handleSaveAsMemos()}
|
|
||||||
>
|
|
||||||
{t("memo-chat.save-as-memo")}
|
|
||||||
</button>
|
|
||||||
</>
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
export default ChatMessage;
|
|
||||||
@ -1,33 +0,0 @@
|
|||||||
import Icon from "@/components/Icon";
|
|
||||||
import { Conversation } from "@/store/v1/conversation";
|
|
||||||
|
|
||||||
interface ConversationTabProps {
|
|
||||||
item: Conversation;
|
|
||||||
selectedConversationId: string;
|
|
||||||
setSelectedConversationId: (id: string) => void;
|
|
||||||
closeConversation: (e: any) => void;
|
|
||||||
}
|
|
||||||
|
|
||||||
const ConversationTab = ({ item, selectedConversationId, setSelectedConversationId, closeConversation }: ConversationTabProps) => {
|
|
||||||
return (
|
|
||||||
<div
|
|
||||||
className={`flex rounded-lg h-8 px-3 cursor-pointer border dark:border-zinc-600 ${
|
|
||||||
selectedConversationId === item.messageStorageId ? "bg-white dark:bg-zinc-700" : "bg-gray-200 dark:bg-zinc-800 opacity-60"
|
|
||||||
}`}
|
|
||||||
key={item.messageStorageId}
|
|
||||||
onClick={() => {
|
|
||||||
setSelectedConversationId(item.messageStorageId);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<div className="truncate m-auto">{item.name}</div>
|
|
||||||
<Icon.X
|
|
||||||
className="ml-1 w-4 h-auto m-auto cursor-pointer opacity-60"
|
|
||||||
onClick={(e: any) => {
|
|
||||||
closeConversation(e);
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
export default ConversationTab;
|
|
||||||
@ -1,218 +0,0 @@
|
|||||||
import { fetchEventSource } from "@microsoft/fetch-event-source";
|
|
||||||
import { Button, Stack } from "@mui/joy";
|
|
||||||
import { head } from "lodash-es";
|
|
||||||
import React, { useEffect, useState } from "react";
|
|
||||||
import { toast } from "react-hot-toast";
|
|
||||||
import Empty from "@/components/Empty";
|
|
||||||
import Icon from "@/components/Icon";
|
|
||||||
import ChatInput from "@/components/MemoChat/ChatInput";
|
|
||||||
import ChatMessage from "@/components/MemoChat/ChatMessage";
|
|
||||||
import ConversationTab from "@/components/MemoChat/ConversationTab";
|
|
||||||
import MobileHeader from "@/components/MobileHeader";
|
|
||||||
import * as api from "@/helpers/api";
|
|
||||||
import useLoading from "@/hooks/useLoading";
|
|
||||||
import { Conversation, useConversationStore } from "@/store/v1/conversation";
|
|
||||||
import { Message, useMessageStore } from "@/store/v1/message";
|
|
||||||
import { useTranslate } from "@/utils/i18n";
|
|
||||||
import { generateUUID } from "@/utils/uuid";
|
|
||||||
|
|
||||||
const chatStreaming = async (messageList: Array<Message>, onmessage: any, onclose: any) => {
|
|
||||||
await fetchEventSource("/api/v1/openai/chat-streaming", {
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
},
|
|
||||||
body: JSON.stringify(messageList),
|
|
||||||
async onopen() {
|
|
||||||
// to do nth
|
|
||||||
},
|
|
||||||
onmessage(event: any) {
|
|
||||||
onmessage(event);
|
|
||||||
},
|
|
||||||
onclose() {
|
|
||||||
onclose();
|
|
||||||
},
|
|
||||||
onerror(error: any) {
|
|
||||||
console.log("error", error);
|
|
||||||
},
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
const MemoChat = () => {
|
|
||||||
const t = useTranslate();
|
|
||||||
const fetchingState = useLoading(false);
|
|
||||||
const [isEnabled, setIsEnabled] = useState<boolean>(true);
|
|
||||||
const [isInIME, setIsInIME] = useState(false);
|
|
||||||
const [question, setQuestion] = useState<string>("");
|
|
||||||
const conversationStore = useConversationStore();
|
|
||||||
const conversationList = conversationStore.conversationList;
|
|
||||||
const [selectedConversationId, setSelectedConversationId] = useState<string>(head(conversationList)?.messageStorageId || "");
|
|
||||||
const messageStore = useMessageStore(selectedConversationId)();
|
|
||||||
const messageList = messageStore.messageList;
|
|
||||||
// The state didn't show in component, just for trigger re-render
|
|
||||||
const [message, setMessage] = useState<string>("");
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
api.checkOpenAIEnabled().then(({ data }) => {
|
|
||||||
setIsEnabled(data);
|
|
||||||
});
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
// to new a conversation when no conversation
|
|
||||||
useEffect(() => {
|
|
||||||
if (conversationList.length === 0) {
|
|
||||||
newConversation();
|
|
||||||
}
|
|
||||||
}, [conversationList]);
|
|
||||||
|
|
||||||
// to select head message conversation(conversation) when conversation be deleted
|
|
||||||
useEffect(() => {
|
|
||||||
setSelectedConversationId(head(conversationList)?.messageStorageId || "");
|
|
||||||
}, [conversationList]);
|
|
||||||
|
|
||||||
const handleGotoSystemSetting = () => {
|
|
||||||
window.open(`/setting`);
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleQuestionTextareaChange = async (event: React.ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setQuestion(event.currentTarget.value);
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleKeyDown = (event: React.KeyboardEvent) => {
|
|
||||||
if (event.key === "Enter" && !event.shiftKey && !isInIME) {
|
|
||||||
event.preventDefault();
|
|
||||||
handleSendQuestionButtonClick().then();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleSendQuestionButtonClick = async () => {
|
|
||||||
if (!question) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
fetchingState.setLoading();
|
|
||||||
setQuestion("");
|
|
||||||
messageStore.addMessage({
|
|
||||||
id: generateUUID(),
|
|
||||||
role: "user",
|
|
||||||
content: question,
|
|
||||||
});
|
|
||||||
|
|
||||||
const messageId = generateUUID();
|
|
||||||
messageStore.addMessage({
|
|
||||||
id: messageId,
|
|
||||||
role: "assistant",
|
|
||||||
content: "",
|
|
||||||
});
|
|
||||||
try {
|
|
||||||
fetchChatStreaming(messageId);
|
|
||||||
} catch (error: any) {
|
|
||||||
console.error(error);
|
|
||||||
toast.error(error.response.data.error);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const fetchChatStreaming = async (messageId: string) => {
|
|
||||||
const messageList = messageStore.getState().messageList;
|
|
||||||
await chatStreaming(
|
|
||||||
messageList,
|
|
||||||
async (event: any) => {
|
|
||||||
messageStore.updateMessage(messageId, event.data);
|
|
||||||
// to trigger re-render
|
|
||||||
setMessage(message + event.data);
|
|
||||||
},
|
|
||||||
async () => {
|
|
||||||
fetchingState.setFinish();
|
|
||||||
}
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
const newConversation = () => {
|
|
||||||
const uuid = generateUUID();
|
|
||||||
// get the time HH:mm as the default name
|
|
||||||
const name = new Date().toLocaleTimeString("en-US", {
|
|
||||||
hour: "2-digit",
|
|
||||||
minute: "2-digit",
|
|
||||||
});
|
|
||||||
conversationStore.addConversation({
|
|
||||||
name: name,
|
|
||||||
messageStorageId: uuid,
|
|
||||||
});
|
|
||||||
setSelectedConversationId(uuid);
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
|
||||||
<section className="w-full max-w-3xl min-h-full flex flex-col justify-start items-center px-4 sm:px-2 sm:pt-4 pb-8 bg-zinc-100 dark:bg-zinc-800">
|
|
||||||
<MobileHeader showSearch={false} />
|
|
||||||
<div className="w-full flex flex-col justify-start items-start px-4 py-3 rounded-xl bg-white dark:bg-zinc-700 text-black dark:text-gray-300">
|
|
||||||
<div className="w-full flex">
|
|
||||||
<div className="w-auto flex flex-row justify-start items-center select-none shrink-0 mr-4">
|
|
||||||
<Icon.Bot className="w-5 h-auto mr-1" /> {t("memo-chat.title")}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="flex flex-row w-auto justify-start items-center overflow-y-hidden overflow-x-auto">
|
|
||||||
<div className="flex space-x-2 overflow-x-auto">
|
|
||||||
{conversationList.map((item: Conversation) => (
|
|
||||||
<ConversationTab
|
|
||||||
key={item.messageStorageId}
|
|
||||||
item={item}
|
|
||||||
selectedConversationId={selectedConversationId}
|
|
||||||
setSelectedConversationId={setSelectedConversationId}
|
|
||||||
closeConversation={(e) => {
|
|
||||||
// this is very important. otherwise, the select event also be clicked.
|
|
||||||
e.stopPropagation();
|
|
||||||
conversationStore.removeConversation(item);
|
|
||||||
toast.success("Remove successfully");
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
<button className="btn-text px-1 ml-1 shrink-0">
|
|
||||||
<Icon.Plus
|
|
||||||
className="w-4 h-auto"
|
|
||||||
onClick={() => {
|
|
||||||
newConversation();
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="dialog-content-container w-full">
|
|
||||||
<Stack spacing={2} style={{ width: "100%" }}>
|
|
||||||
{messageList.length == 0 && (
|
|
||||||
<div className="w-full mt-8 mb-8 flex flex-col justify-center items-center italic">
|
|
||||||
<Empty />
|
|
||||||
<p className="mt-4 text-gray-600 dark:text-gray-400">{t("memo-chat.no-message")}</p>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{messageList.map((message, index) => (
|
|
||||||
<ChatMessage key={index} message={message} index={index} />
|
|
||||||
))}
|
|
||||||
</Stack>
|
|
||||||
{fetchingState.isLoading && (
|
|
||||||
<p className="w-full py-2 mt-4 flex flex-row justify-center items-center">
|
|
||||||
<Icon.Loader className="w-5 h-auto animate-spin" />
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
{!isEnabled && (
|
|
||||||
<div className="w-full flex flex-col justify-center items-center mt-4 space-y-2">
|
|
||||||
<p>{t("memo-chat.not-enabled")}</p>
|
|
||||||
<Button onClick={() => handleGotoSystemSetting()}>{t("memo-chat.go-to-settings")}</Button>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<ChatInput
|
|
||||||
question={question}
|
|
||||||
handleQuestionTextareaChange={handleQuestionTextareaChange}
|
|
||||||
setIsInIME={setIsInIME}
|
|
||||||
handleKeyDown={handleKeyDown}
|
|
||||||
handleSendQuestionButtonClick={handleSendQuestionButtonClick}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</section>
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
export default MemoChat;
|
|
||||||
@ -1,43 +0,0 @@
|
|||||||
import { t } from "i18next";
|
|
||||||
import { create } from "zustand";
|
|
||||||
import { persist } from "zustand/middleware";
|
|
||||||
|
|
||||||
export interface Conversation {
|
|
||||||
name: string;
|
|
||||||
messageStorageId: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface ConversationState {
|
|
||||||
conversationList: Conversation[];
|
|
||||||
getState: () => ConversationState;
|
|
||||||
addConversation: (conversation: Conversation) => void;
|
|
||||||
removeConversation: (conversation: Conversation) => Conversation;
|
|
||||||
}
|
|
||||||
|
|
||||||
export const defaultConversation: Conversation = {
|
|
||||||
name: t("ask-ai.default-message-conversation-title"),
|
|
||||||
messageStorageId: "message-storage",
|
|
||||||
};
|
|
||||||
|
|
||||||
export const useConversationStore = create<ConversationState>()(
|
|
||||||
persist(
|
|
||||||
(set, get) => ({
|
|
||||||
conversationList: [],
|
|
||||||
getState: () => get(),
|
|
||||||
addConversation: (conversation: Conversation) => set((state) => ({ conversationList: [...state.conversationList, conversation] })),
|
|
||||||
removeConversation: (conversation: Conversation) => {
|
|
||||||
set((state) => ({
|
|
||||||
conversationList: state.conversationList.filter(
|
|
||||||
(i) => i.name != conversation.name || i.messageStorageId != conversation.messageStorageId
|
|
||||||
),
|
|
||||||
}));
|
|
||||||
localStorage.removeItem(conversation.messageStorageId);
|
|
||||||
const conversationList = get().conversationList;
|
|
||||||
return conversationList.length > 0 ? conversationList[conversationList.length - 1] : defaultConversation;
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
{
|
|
||||||
name: "message-conversation-storage",
|
|
||||||
}
|
|
||||||
)
|
|
||||||
);
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
import { create } from "zustand";
|
|
||||||
import { persist } from "zustand/middleware";
|
|
||||||
|
|
||||||
export interface Message {
|
|
||||||
id: string;
|
|
||||||
role: "user" | "assistant";
|
|
||||||
content: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface MessageState {
|
|
||||||
messageList: Message[];
|
|
||||||
getState: () => MessageState;
|
|
||||||
addMessage: (message: Message) => void;
|
|
||||||
updateMessage: (id: string, appendContent: string) => void;
|
|
||||||
}
|
|
||||||
|
|
||||||
export const useMessageStore = (messageStorageId: string) => {
|
|
||||||
return create<MessageState>()(
|
|
||||||
persist(
|
|
||||||
(set, get) => ({
|
|
||||||
messageList: [] as Message[],
|
|
||||||
getState: () => get(),
|
|
||||||
addMessage: (message: Message) => {
|
|
||||||
return set((state) => ({ messageList: [...state.messageList, message] }));
|
|
||||||
},
|
|
||||||
updateMessage: (id: string, appendContent: string) =>
|
|
||||||
set((state) => ({
|
|
||||||
...state,
|
|
||||||
messageList: state.messageList.map((item) => (item.id === id ? { ...item, content: item.content + appendContent } : item)),
|
|
||||||
})),
|
|
||||||
}),
|
|
||||||
{
|
|
||||||
name: messageStorageId,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
);
|
|
||||||
};
|
|
||||||
Loading…
Reference in New Issue