opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-09-25 17:15:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: prompt generator

Browse Source
This commit is contained in:
Steven
2023-04-20 21:49:50 +08:00
parent 6dac9146b1
commit 22f56f07c3
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/components/ConversationView/index.tsx
View File

@ -33,7 +33,9 @@ const ConversationView = () => {
const [showHeaderShadow, setShowHeaderShadow] = useState<boolean>(false); const [showHeaderShadow, setShowHeaderShadow] = useState<boolean>(false);
const conversationViewRef = useRef<HTMLDivElement>(null); const conversationViewRef = useRef<HTMLDivElement>(null);
const currentConversation = conversationStore.getConversationById(conversationStore.currentConversationId); const currentConversation = conversationStore.getConversationById(conversationStore.currentConversationId);
const messageList = messageStore.messageList.filter((message) => message.conversationId === currentConversation?.id); const messageList = currentConversation
? messageStore.messageList.filter((message) => message.conversationId === currentConversation.id)
: [];
const lastMessage = last(messageList); const lastMessage = last(messageList);
useEffect(() => { useEffect(() => {
@ -111,7 +113,8 @@ const ConversationView = () => {
} }
const messageList = messageStore.getState().messageList.filter((message) => message.conversationId === currentConversation.id); const messageList = messageStore.getState().messageList.filter((message) => message.conversationId === currentConversation.id);
let prompt = ""; const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
let prompt = promptGenerator();
let tokens = 0; let tokens = 0;
const message: Message = { const message: Message = {
@ -138,7 +141,6 @@ const ConversationView = () => {
} catch (error: any) { } catch (error: any) {
toast.error(error.message); toast.error(error.message);
} }
const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
prompt = promptGenerator(schema); prompt = promptGenerator(schema);
} }
let formatedMessageList = []; let formatedMessageList = [];