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

feat: implement general assistant

Browse Source
This commit is contained in:
steven
2023-04-17 16:14:39 +08:00
parent 13b23181b0
commit d03ce9fac1
10 changed files with 74 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/store/assistant.ts
View File

@ -1,30 +1,18 @@
import { first } from "lodash-es";
import { Id, User } from "@/types";
import { Assistant, Id } from "@/types";
import * as customAssistantList from "../../assistants";
// 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: "",
},
];
const assistantList: Assistant[] = Object.keys(customAssistantList).map((name) => {
return {
...((customAssistantList as any)[name].default as Assistant),
};
});
export const getAssistantById = (id: Id) => {
const user = assistantList.find((user) => user.id === id);
return user || (first(assistantList) as User);
const assistant = assistantList.find((assistant) => assistant.id === id);
return assistant || (first(assistantList) as Assistant);
};
// 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}`;
export const getPromptGeneratorOfAssistant = (assistant: Assistant) => {
return assistant.getPrompt;
};