refactor: move codes into src folder

This commit is contained in:
steven
2023-03-31 16:12:09 +08:00
parent 483e750bc5
commit 861a22f13d
57 changed files with 44 additions and 8 deletions

30
src/store/assistant.ts Normal file
View File

@ -0,0 +1,30 @@
import { first } from "lodash-es";
import { Id, User } from "@/types";
// Assistant is a special user.
export const assistantList: User[] = [
{
id: "sql-assistant",
name: "SQL Chat",
description: "🤖️ I'm an expert in SQL. I can answer your questions about databases and SQL.",
avatar: "",
},
];
export const getAssistantById = (id: Id) => {
const user = assistantList.find((user) => user.id === id);
return user || (first(assistantList) as User);
};
// getPromptOfAssistant define the special prompt for each assistant.
export const getPromptGeneratorOfAssistant = (assistant: User) => {
const basicPrompt = `Please follow the instructions to answer the questions:
1. Set the language to the markdown code block for each code block. For example, \`SELECT * FROM table\` is SQL.
2. Please be careful to return only key information, and try not to make it too long.
`;
if (assistant.id === "sql-assistant") {
return (schema: string) =>
`This is my database schema"${schema}". You will see the tables and columns in the database. And please answer the following questions about the database.\n${basicPrompt}`;
}
return () => `\n${basicPrompt}`;
};