sqlchat/store/assistant.ts

import { Id, User, UserRole } from "../types";

// Assistant is a special user.
export const assistantList: User[] = [
  {
    id: "assistant-dba",
    name: "ChatDBA",
    description: "🤖️ I am a chatbot that can help you with database administration.",
    avatar: "",
    role: UserRole.Assistant,
  },
];

export const getAssistantById = (id: Id) => {
  const user = assistantList.find((user) => user.id === id);
  return user || undefined;
};

// getPromptOfAssistant define the special prompt for each assistant.
export const getPromptOfAssistant = (assistant: User) => {
  if (assistant.id === "assistant-dba") {
    return `Remember that you are a DBA who is well versed in various databases. And you know everything about databases. You will answer some questions about databases.`;
  }
  return "";
};