opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-07-29 10:13:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add prompt for assistant

Browse Source
This commit is contained in:
Steven
2023-03-19 23:19:08 +08:00
parent 797eb5d824
commit d11648f2bb
6 changed files with 35 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
components/ChatView/index.tsx
View File

@ -1,6 +1,6 @@
import axios from "axios";
import { useEffect, useRef, useState } from "react";
import { defaultChat, useChatStore, useMessageStore, useUserStore } from "../../store";
import { defaultChat, getPromptOfAssistant, useChatStore, useMessageStore, useUserStore } from "../../store";
import { Chat, Message, UserRole } from "../../types";
import { generateUUID } from "../../utils";
import Icon from "../Icon";
@ -47,11 +47,18 @@ const ChatView = () => {
setIsRequesting(true);
const messageList = messageStore.getState().messageList.filter((message) => message.chatId === currentChat.id);
const prompt = getPromptOfAssistant(userStore.getAssistantById(currentChat.assistantId)!);
const { data } = await axios.post<string>("/api/chat", {
messages: messageList.map((message) => ({
role: message.creatorId === userStore.currentUser.id ? UserRole.User : UserRole.Assistant,
content: message.content,
})),
messages: [
{
role: "system",
content: prompt,
},
...messageList.map((message) => ({
role: message.creatorId === userStore.currentUser.id ? UserRole.User : UserRole.Assistant,
content: message.content,
})),
],
});
messageStore.addMessage({
id: generateUUID(),