-
+
+ {session?.user ? (
+ session.user.image ? (
+
+ ) : (
+
+ {session.user.name ? session.user.name.charAt(0) : session.user.email?.charAt(0)}
+
+ )
+ ) : (
+
+ )}
) : (
<>
{connection ? (
-
+
) : (
-
+
)}
{message.status === "LOADING" && message.content === "" ? (
@@ -113,29 +113,21 @@ const MessageView = (props: Props) => {
+
@@ -148,10 +140,8 @@ const MessageView = (props: Props) => {
>
{message.content}
-
- {dayjs(message.createdAt)
- .locale(settingStore.setting.locale)
- .format("lll")}
+
+ {dayjs(message.createdAt).locale(settingStore.setting.locale).format("lll")}
diff --git a/src/components/ConversationView/ThreeDotsLoader.tsx b/src/components/ConversationView/ThreeDotsLoader.tsx
index 2f4fb236..a87980cd 100644
--- a/src/components/ConversationView/ThreeDotsLoader.tsx
+++ b/src/components/ConversationView/ThreeDotsLoader.tsx
@@ -1,6 +1,6 @@
-import { useSettingStore } from "@/store";
import { useEffect, useState } from "react";
import { ThreeDots } from "react-loader-spinner";
+import { useSettingStore } from "@/store";
const ThreeDotsLoader = () => {
const settingStore = useSettingStore();
@@ -10,10 +10,7 @@ const ThreeDotsLoader = () => {
const theme = settingStore.setting.theme;
let appearance = theme;
if (theme === "system") {
- if (
- window.matchMedia &&
- window.matchMedia("(prefers-color-scheme: dark)").matches
- ) {
+ if (window.matchMedia && window.matchMedia("(prefers-color-scheme: dark)").matches) {
appearance = "dark";
} else {
appearance = "light";
@@ -27,14 +24,7 @@ const ThreeDotsLoader = () => {
}
}, [settingStore.setting.theme]);
- return (
-
- );
+ return
;
};
export default ThreeDotsLoader;
diff --git a/src/components/ConversationView/index.tsx b/src/components/ConversationView/index.tsx
index af6502ec..585fbb6b 100644
--- a/src/components/ConversationView/index.tsx
+++ b/src/components/ConversationView/index.tsx
@@ -1,8 +1,8 @@
import axios from "axios";
-import { first, head, last } from "lodash-es";
+import { head, last } from "lodash-es";
+import { useSession } from "next-auth/react";
import { useEffect, useRef, useState } from "react";
import { toast } from "react-hot-toast";
-import { API_KEY } from "@/env";
import {
getAssistantById,
getPromptGeneratorOfAssistant,
@@ -14,20 +14,22 @@ import {
useUserStore,
} from "@/store";
import { Conversation, CreatorRole, Message } from "@/types";
-import { countTextTokens, generateUUID } from "@/utils";
+import { countTextTokens, generateUUID, getModel, hasFeature, generateDbPromptFromContext } from "@/utils";
+import getEventEmitter from "@/utils/event-emitter";
import Header from "./Header";
import EmptyView from "../EmptyView";
import MessageView from "./MessageView";
import ClearConversationButton from "../ClearConversationButton";
import MessageTextarea from "./MessageTextarea";
import DataStorageBanner from "../DataStorageBanner";
-import QuotaOverflowBanner from "../QuotaOverflowBanner";
-
-// The maximum number of tokens that can be sent to the OpenAI API.
-// reference: https://platform.openai.com/docs/api-reference/completions/create#completions/create-max_tokens
-const MAX_TOKENS = 4000;
+import DBHubBanner from "../DBHubBanner";
+import SchemaDrawer from "../SchemaDrawer";
+import Icon from "../Icon";
+import { useTranslation } from "react-i18next";
const ConversationView = () => {
+ const { data: session } = useSession();
+ const { t } = useTranslation();
const settingStore = useSettingStore();
const layoutStore = useLayoutStore();
const connectionStore = useConnectionStore();
@@ -37,15 +39,12 @@ const ConversationView = () => {
const [isStickyAtBottom, setIsStickyAtBottom] = useState
(true);
const [showHeaderShadow, setShowHeaderShadow] = useState(false);
const conversationViewRef = useRef(null);
- const currentConversation = conversationStore.getConversationById(
- conversationStore.currentConversationId
- );
+ const currentConversation = conversationStore.getConversationById(conversationStore.currentConversationId);
const messageList = currentConversation
- ? messageStore.messageList.filter(
- (message: Message) => message.conversationId === currentConversation.id
- )
+ ? messageStore.messageList.filter((message: Message) => message.conversationId === currentConversation.id)
: [];
const lastMessage = last(messageList);
+ const [showSchemaDrawer, setShowSchemaDrawer] = useState(false);
useEffect(() => {
messageStore.messageList.map((message: Message) => {
@@ -69,21 +68,13 @@ const ConversationView = () => {
}
setShowHeaderShadow((conversationViewRef.current?.scrollTop || 0) > 0);
setIsStickyAtBottom(
- conversationViewRef.current.scrollTop +
- conversationViewRef.current.clientHeight >=
- conversationViewRef.current.scrollHeight
+ conversationViewRef.current.scrollTop + conversationViewRef.current.clientHeight >= conversationViewRef.current.scrollHeight
);
};
- conversationViewRef.current?.addEventListener(
- "scroll",
- handleConversationViewScroll
- );
+ conversationViewRef.current?.addEventListener("scroll", handleConversationViewScroll);
return () => {
- conversationViewRef.current?.removeEventListener(
- "scroll",
- handleConversationViewScroll
- );
+ conversationViewRef.current?.removeEventListener("scroll", handleConversationViewScroll);
};
}, []);
@@ -91,8 +82,7 @@ const ConversationView = () => {
if (!conversationViewRef.current) {
return;
}
- conversationViewRef.current.scrollTop =
- conversationViewRef.current.scrollHeight;
+ conversationViewRef.current.scrollTop = conversationViewRef.current.scrollHeight;
}, [currentConversation, lastMessage?.id]);
useEffect(() => {
@@ -101,17 +91,14 @@ const ConversationView = () => {
}
if (lastMessage?.status === "LOADING" && isStickyAtBottom) {
- conversationViewRef.current.scrollTop =
- conversationViewRef.current.scrollHeight;
+ conversationViewRef.current.scrollTop = conversationViewRef.current.scrollHeight;
}
}, [lastMessage?.status, lastMessage?.content, isStickyAtBottom]);
useEffect(() => {
if (
- currentConversation?.connectionId ===
- connectionStore.currentConnectionCtx?.connection.id &&
- currentConversation?.databaseName ===
- connectionStore.currentConnectionCtx?.database?.name
+ currentConversation?.connectionId === connectionStore.currentConnectionCtx?.connection.id &&
+ currentConversation?.databaseName === connectionStore.currentConnectionCtx?.database?.name
) {
return;
}
@@ -119,18 +106,14 @@ const ConversationView = () => {
// Auto select the first conversation when the current connection changes.
const conversationList = conversationStore.conversationList.filter(
(conversation: Conversation) =>
- conversation.connectionId ===
- connectionStore.currentConnectionCtx?.connection.id &&
- conversation.databaseName ===
- connectionStore.currentConnectionCtx?.database?.name
+ conversation.connectionId === connectionStore.currentConnectionCtx?.connection.id &&
+ conversation.databaseName === connectionStore.currentConnectionCtx?.database?.name
);
conversationStore.setCurrentConversationId(head(conversationList)?.id);
}, [currentConversation, connectionStore.currentConnectionCtx]);
const sendMessageToCurrentConversation = async (userPrompt: string) => {
- const currentConversation = conversationStore.getConversationById(
- conversationStore.getState().currentConversationId
- );
+ const currentConversation = conversationStore.getConversationById(conversationStore.getState().currentConversationId);
if (!currentConversation) {
return;
}
@@ -150,28 +133,11 @@ const ConversationView = () => {
};
messageStore.addMessage(userMessage);
- // Add PENDING assistant message to the store.
- const assistantMessage: Message = {
- id: generateUUID(),
- conversationId: currentConversation.id,
- creatorId: currentConversation.assistantId,
- creatorRole: CreatorRole.Assistant,
- createdAt: Date.now(),
- content: "",
- status: "LOADING",
- };
- messageStore.addMessage(assistantMessage);
-
// Construct the system prompt
- const messageList = messageStore
- .getState()
- .messageList.filter(
- (message: Message) => message.conversationId === currentConversation.id
- );
- const promptGenerator = getPromptGeneratorOfAssistant(
- getAssistantById(currentConversation.assistantId)!
- );
+ const messageList = messageStore.getState().messageList.filter((message: Message) => message.conversationId === currentConversation.id);
+ const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
let dbPrompt = promptGenerator();
+ const maxToken = getModel(settingStore.setting.openAIApiConfig?.model || "").max_token;
// Squeeze as much prompt as possible under the token limit, the prompt is in the order of:
// 1. Assistant specific prompt with database schema if applicable.
// 2. A list of previous exchanges.
@@ -185,32 +151,20 @@ const ConversationView = () => {
// Augument with database schema if available
if (connectionStore.currentConnectionCtx?.database) {
- let schema = "";
+ const schemaList = await connectionStore.getOrFetchDatabaseSchema(connectionStore.currentConnectionCtx?.database);
try {
- const tables = await connectionStore.getOrFetchDatabaseSchema(
- connectionStore.currentConnectionCtx?.database
+ dbPrompt = generateDbPromptFromContext(
+ promptGenerator,
+ connectionStore.currentConnectionCtx.connection.engineType,
+ schemaList,
+ currentConversation.selectedSchemaName || "",
+ currentConversation.selectedTableNameList || [],
+ maxToken,
+ userPrompt
);
- // Empty table name(such as "") denote all table. "" and `undefined` both are false in `if`
- if (currentConversation.tableName) {
- const table = tables.find((table) => {
- return table.name === currentConversation.tableName;
- });
- if (table) {
- tokens += countTextTokens(schema + table.structure);
- schema += table.structure;
- }
- } else {
- for (const table of tables) {
- if (tokens < MAX_TOKENS / 2) {
- tokens += countTextTokens(schema + table.structure);
- schema += table.structure;
- }
- }
- }
} catch (error: any) {
toast.error(error.message);
}
- dbPrompt = promptGenerator(schema);
}
// Sliding window to add messages with DONE status all the way back up until we reach the token
@@ -220,7 +174,7 @@ const ConversationView = () => {
for (let i = messageList.length - 1; i >= 0; i--) {
const message = messageList[i];
if (message.status === "DONE") {
- if (tokens < MAX_TOKENS) {
+ if (tokens < maxToken) {
tokens += countTextTokens(message.content);
formatedMessageList.unshift({
role: message.creatorRole,
@@ -243,34 +197,59 @@ const ConversationView = () => {
});
const requestHeaders: any = {};
- if (API_KEY) {
- requestHeaders["Authorization"] = `Bearer ${API_KEY}`;
+ if (session?.user.id) {
+ requestHeaders["Authorization"] = `Bearer ${session?.user.id}`;
+ }
+ if (settingStore.setting.openAIApiConfig?.key) {
+ requestHeaders["x-openai-key"] = settingStore.setting.openAIApiConfig?.key;
+ }
+ if (settingStore.setting.openAIApiConfig?.endpoint) {
+ requestHeaders["x-openai-endpoint"] = settingStore.setting.openAIApiConfig?.endpoint;
+ }
+ if (settingStore.setting.openAIApiConfig?.model) {
+ requestHeaders["x-openai-model"] = settingStore.setting.openAIApiConfig?.model;
}
const rawRes = await fetch("/api/chat", {
method: "POST",
body: JSON.stringify({
messages: formatedMessageList,
- openAIApiConfig: settingStore.setting.openAIApiConfig,
}),
headers: requestHeaders,
});
if (!rawRes.ok) {
console.error(rawRes);
- let errorMessage =
- "Failed to request message, please check your network.";
+ let errorMessage = "Failed to request message, please check your network.";
try {
- errorMessage = await rawRes.statusText;
+ const res = await rawRes.json();
+ errorMessage = res.error.message;
} catch (error) {
// do nth
}
- messageStore.updateMessage(assistantMessage.id, {
+ messageStore.addMessage({
+ id: generateUUID(),
+ conversationId: currentConversation.id,
+ creatorId: currentConversation.assistantId,
+ creatorRole: CreatorRole.Assistant,
+ createdAt: Date.now(),
content: errorMessage,
status: "FAILED",
});
return;
}
+ // Add PENDING assistant message to the store.
+ const assistantMessage: Message = {
+ id: generateUUID(),
+ conversationId: currentConversation.id,
+ creatorId: currentConversation.assistantId,
+ creatorRole: CreatorRole.Assistant,
+ createdAt: Date.now(),
+ content: "",
+ status: "LOADING",
+ };
+ messageStore.addMessage(assistantMessage);
+
const data = rawRes.body;
if (!data) {
toast.error("No data return");
@@ -297,30 +276,41 @@ const ConversationView = () => {
status: "DONE",
});
- // Collect system prompt
- // We only collect the db prompt for the system prompt. We do not collect the intermediate
- // exchange to save space since those can be derived from the previous record.
- usageMessageList.push({
- id: generateUUID(),
- createdAt: Date.now(),
- creatorRole: CreatorRole.System,
- content: dbPrompt,
- } as Message);
+ // Emit usage update event so quota widget can update.
+ getEventEmitter().emit("usage.update");
- // Collect user message
- usageMessageList.push(userMessage);
+ if (hasFeature("collect")) {
+ // Collect system prompt
+ // We only collect the db prompt for the system prompt. We do not collect the intermediate
+ // exchange to save space since those can be derived from the previous record.
+ usageMessageList.push({
+ id: generateUUID(),
+ createdAt: Date.now(),
+ creatorRole: CreatorRole.System,
+ content: dbPrompt,
+ } as Message);
- // Collect assistant response
- usageMessageList.push(assistantMessage);
+ // Collect user message
+ usageMessageList.push(userMessage);
- axios
- .post("/api/usage", {
- conversation: currentConversation,
- messages: usageMessageList,
- })
- .catch(() => {
- // do nth
- });
+ // Collect assistant response
+ usageMessageList.push(assistantMessage);
+
+ axios
+ .post(
+ "/api/collect",
+ {
+ conversation: currentConversation,
+ messages: usageMessageList,
+ },
+ {
+ headers: requestHeaders,
+ }
+ )
+ .catch(() => {
+ // do nth
+ });
+ }
};
return (
@@ -331,28 +321,29 @@ const ConversationView = () => {
} relative w-full h-full max-h-full flex flex-col justify-start items-start overflow-y-auto bg-white dark:bg-zinc-800`}
>
-
{messageList.length === 0 ? (
-
-
+
+
setShowSchemaDrawer(true)}>
+ {
+
+ {t("prompt.self")}
+
+ }
+
+ {showSchemaDrawer &&
setShowSchemaDrawer(false)} />}
);
diff --git a/src/components/CreateConnectionModal.tsx b/src/components/CreateConnectionModal.tsx
index 03608d66..e0844ab2 100644
--- a/src/components/CreateConnectionModal.tsx
+++ b/src/components/CreateConnectionModal.tsx
@@ -1,30 +1,32 @@
import { cloneDeep, head } from "lodash-es";
import { ChangeEvent, useEffect, useState } from "react";
import { toast } from "react-hot-toast";
+import { useTranslation } from "react-i18next";
import TextareaAutosize from "react-textarea-autosize";
import { useConnectionStore } from "@/store";
import { Connection, Engine, ResponseObject, SSLOptions } from "@/types";
-import Select from "./kit/Select";
+import Radio from "./kit/Radio";
import TextField from "./kit/TextField";
import Modal from "./kit/Modal";
import Icon from "./Icon";
+import EngineIcon from "./EngineIcon";
import DataStorageBanner from "./DataStorageBanner";
import ActionConfirmModal from "./ActionConfirmModal";
-import { useTranslation } from "react-i18next";
+import RequiredStar from "./RequiredStar";
interface Props {
connection?: Connection;
close: () => void;
}
-type SSLType = "none" | "ca-only" | "full";
+type SSLType = "preferred" | "ca-only" | "full";
type SSLFieldType = "ca" | "cert" | "key";
const SSLTypeOptions = [
{
- label: "None",
- value: "none",
+ label: "Preferred",
+ value: "preferred",
},
{
label: "CA Only",
@@ -36,12 +38,50 @@ const SSLTypeOptions = [
},
];
+const engines = [
+ {
+ type: Engine.MySQL,
+ name: "MySQL",
+ defaultPort: "3306",
+ },
+ {
+ type: Engine.PostgreSQL,
+ name: "PostgreSQL",
+ defaultPort: "5432",
+ },
+ {
+ type: Engine.MSSQL,
+ name: "SQL Server",
+ defaultPort: "1433",
+ },
+ {
+ type: Engine.TiDB,
+ name: "TiDB Serverless",
+ defaultPort: "4000",
+ },
+ {
+ type: Engine.OceanBase,
+ name: "OceanBase",
+ defaultPort: "2881",
+ },
+ // {
+ // type: Engine.Snowflake,
+ // name: "Snowflake",
+ // defaultPort: "443",
+ // },
+ // {
+ // type: Engine.Hive,
+ // name: "Hive",
+ // defaultPort: "4000",
+ // },
+];
+
const defaultConnection: Connection = {
id: "",
title: "",
engineType: Engine.MySQL,
host: "",
- port: "",
+ port: "3306",
username: "",
password: "",
};
@@ -51,14 +91,13 @@ const CreateConnectionModal = (props: Props) => {
const { t } = useTranslation();
const connectionStore = useConnectionStore();
const [connection, setConnection] = useState
(defaultConnection);
- const [showDeleteConnectionModal, setShowDeleteConnectionModal] =
- useState(false);
- const [sslType, setSSLType] = useState("none");
+ const [showDeleteConnectionModal, setShowDeleteConnectionModal] = useState(false);
+ const [sslType, setSSLType] = useState("preferred");
const [selectedSSLField, setSelectedSSLField] = useState("ca");
const [isRequesting, setIsRequesting] = useState(false);
const showDatabaseField = connection.engineType === Engine.PostgreSQL;
const isEditing = editConnection !== undefined;
- const allowSave = connection.host !== "" && connection.username !== "";
+ const allowSave = connection.title !== "" && connection.host !== "" && connection.username !== "";
useEffect(() => {
const connection = isEditing ? editConnection : defaultConnection;
@@ -106,11 +145,7 @@ const CreateConnectionModal = (props: Props) => {
}
const file = files[0];
- if (
- file.type.startsWith("audio/") ||
- file.type.startsWith("video/") ||
- file.type.startsWith("image/")
- ) {
+ if (file.type.startsWith("audio/") || file.type.startsWith("video/") || file.type.startsWith("image/")) {
toast.error(`Invalid file type:${file.type}`);
return;
}
@@ -139,7 +174,7 @@ const CreateConnectionModal = (props: Props) => {
});
};
- const handleCreateConnection = async () => {
+ const handleUpsertConnection = async () => {
if (isRequesting) {
return;
}
@@ -182,10 +217,7 @@ const CreateConnectionModal = (props: Props) => {
}
// Set the created connection as the current connection.
- const databaseList = await connectionStore.getOrFetchDatabaseList(
- connection,
- true
- );
+ const databaseList = await connectionStore.getOrFetchDatabaseList(connection, true);
connectionStore.setCurrentConnectionCtx({
connection: connection,
database: head(databaseList),
@@ -211,67 +243,51 @@ const CreateConnectionModal = (props: Props) => {
return (
<>
-
+
-
+
{t("connection.database-type")}
+
-
- setPartialConnection({ engineType: value as Engine })
- }
- />
+
+ {engines.map((engine) => (
+
setPartialConnection({ engineType: engine.type, port: engine.defaultPort })}
+ >
+
+ {engine.name}
+
+
+ ))}
+
{t("connection.title")}
+
- setPartialConnection({ title: value })}
- />
+ setPartialConnection({ title: value })} />
{t("connection.host")}
+
- setPartialConnection({ host: value })}
- />
+ setPartialConnection({ host: value })} />
-
- {t("connection.port")}
-
- setPartialConnection({ port: value })}
- />
+ {t("connection.port")}
+ setPartialConnection({ port: value })} />
{showDatabaseField && (
-
- {t("connection.database-name")}
-
+
{t("connection.database-name")}
{
{t("connection.username")}
+
{
/>
-
- {t("connection.password")}
-
+ {t("connection.password")}
{
onChange={(value) => setPartialConnection({ password: value })}
/>
-
-
- SSL
-
-
- {SSLTypeOptions.map((option) => (
-
- {option.label}
-
- ))}
+ {connection.engineType === Engine.TiDB ? (
+
+ {t("connection.tidb-serverless-ssl-hint")}
- {sslType !== "none" && (
- <>
-
- setSelectedSSLField("ca")}
- >
- CA Certificate
-
- {sslType === "full" && (
- <>
- setSelectedSSLField("key")}
- >
- Client Key
-
- setSelectedSSLField("cert")}
- >
- Client Certificate
-
-
- )}
-
-
-
-
-
Input or
-
- upload file
+ ) : (
+
+
SSL
+
+ {SSLTypeOptions.map((option) => (
+
+ setSSLType(value as SSLType)}
+ />
+ {option.label}
+
+ ))}
+
+ {sslType !== "preferred" && (
+ <>
+
+ setSelectedSSLField("ca")}
+ >
+ CA Certificate
+
+ {sslType === "full" && (
+ <>
+ setSelectedSSLField("key")}
+ >
+ Client Key
+
+ setSelectedSSLField("cert")}
+ >
+ Client Certificate
+
+
+ )}
+
+
+
+
+ Input or
+
+ upload file
+
+
+
+
+ )}
+ {connection.engineType === Engine.MSSQL && (
+
+
Encrypt
+
+
Encrypt connection
-
- )}
- {connection.engineType === Engine.MSSQL && (
-
-
- Encrypt
-
-
-
- Encrypt connection
-
-
-
- )}
-
+ )}
+
{isEditing && (
- setShowDeleteConnectionModal(true)}
- >
- Delete
+ setShowDeleteConnectionModal(true)}>
+ {t("common.delete")}
)}
@@ -423,14 +421,8 @@ const CreateConnectionModal = (props: Props) => {
{t("common.close")}
-
- {isRequesting && (
-
+ {isRequesting &&
diff --git a/src/components/DBHubBanner.tsx b/src/components/DBHubBanner.tsx
new file mode 100644
index 00000000..209e9816
--- /dev/null
+++ b/src/components/DBHubBanner.tsx
@@ -0,0 +1,41 @@
+import { useTranslation } from "react-i18next";
+import { useLocalStorage } from "react-use";
+import Icon from "./Icon";
+
+interface Props {
+ className?: string;
+ alwaysShow?: boolean;
+}
+
+const DBHubBanner = (props: Props) => {
+ const { className, alwaysShow } = props;
+ const { t } = useTranslation();
+ const [hideBanner, setHideBanner] = useLocalStorage("hide-github-banner", false);
+ const show = alwaysShow || !hideBanner;
+
+ return (
+
+ );
+};
+
+export default DBHubBanner;
diff --git a/src/components/DataStorageBanner.tsx b/src/components/DataStorageBanner.tsx
index af8e3775..16288310 100644
--- a/src/components/DataStorageBanner.tsx
+++ b/src/components/DataStorageBanner.tsx
@@ -10,10 +10,7 @@ interface Props {
const DataStorageBanner = (props: Props) => {
const { className, alwaysShow } = props;
const { t } = useTranslation();
- const [hideBanner, setHideBanner] = useLocalStorage(
- "hide-local-storage-banner",
- false
- );
+ const [hideBanner, setHideBanner] = useLocalStorage("hide-local-storage-banner", false);
const show = alwaysShow || !hideBanner;
return (
@@ -27,10 +24,7 @@ const DataStorageBanner = (props: Props) => {
{t("banner.data-storage")}
{!alwaysShow && (
-
setHideBanner(true)}
- >
+ setHideBanner(true)}>
)}
diff --git a/src/components/DebugView.tsx b/src/components/DebugView.tsx
new file mode 100644
index 00000000..7af93c2d
--- /dev/null
+++ b/src/components/DebugView.tsx
@@ -0,0 +1,17 @@
+import { useSession } from "next-auth/react";
+
+const PricingView = () => {
+ const { data: session } = useSession();
+
+ return (
+
+
Debug View
+
Session Info
+
+ {JSON.stringify(session, null, "\t")}
+
+
+
-
+
diff --git a/src/components/EngineIcon.tsx b/src/components/EngineIcon.tsx
index 5c031d31..e1587f42 100644
--- a/src/components/EngineIcon.tsx
+++ b/src/components/EngineIcon.tsx
@@ -10,11 +10,15 @@ const EngineIcon = (props: Props) => {
const { className, engine } = props;
if (engine === Engine.MySQL) {
- return
;
+ return
;
} else if (engine === Engine.PostgreSQL) {
- return
;
+ return
;
} else if (engine === Engine.MSSQL) {
- return
;
+ return
;
+ } else if (engine === Engine.TiDB) {
+ return
;
+ } else if (engine === Engine.OceanBase) {
+ return
;
} else {
return
;
}
diff --git a/src/components/ExecutionView/DataTableView.tsx b/src/components/ExecutionView/DataTableView.tsx
index 754cd653..4e5f6c14 100644
--- a/src/components/ExecutionView/DataTableView.tsx
+++ b/src/components/ExecutionView/DataTableView.tsx
@@ -22,9 +22,7 @@ const DataTableView = (props: Props) => {
return rawResults.length === 0 ? (
- {t("execution.message.no-data")}
-
+ {t("execution.message.no-data")}
) : (
{
const { t } = useTranslation();
return (
-
+
- {message}
-
- );
+ return {message}
;
};
export default NotificationView;
diff --git a/src/components/GitHubStarBadge.tsx b/src/components/GitHubStarBadge.tsx
index 11f57d4c..f641938b 100644
--- a/src/components/GitHubStarBadge.tsx
+++ b/src/components/GitHubStarBadge.tsx
@@ -15,15 +15,12 @@ const GitHubStarBadge = (props: Props) => {
const getRepoStarCount = async () => {
let starCount = 0;
try {
- const { data } = await axios.get(
- `https://api.github.com/repos/sqlchat/sqlchat`,
- {
- headers: {
- Accept: "application/vnd.github.v3.star+json",
- Authorization: "",
- },
- }
- );
+ const { data } = await axios.get(`https://api.github.com/repos/sqlchat/sqlchat`, {
+ headers: {
+ Accept: "application/vnd.github.v3.star+json",
+ Authorization: "",
+ },
+ });
starCount = data.stargazers_count as number;
} catch (error) {
// do nth
@@ -50,11 +47,7 @@ const GitHubStarBadge = (props: Props) => {
Star
- {isRequesting ? (
-
);
diff --git a/src/components/LocaleSelector.tsx b/src/components/LocaleSelector.tsx
index 1441ced5..05c627d5 100644
--- a/src/components/LocaleSelector.tsx
+++ b/src/components/LocaleSelector.tsx
@@ -16,26 +16,10 @@ const localeItemList: LocaleItem[] = [
value: "zh",
label: "简体中文",
},
- {
- value: "zhHant",
- label: "繁體中文",
- },
{
value: "es",
label: "Español",
},
- {
- value: "jp",
- label: "日本語",
- },
- {
- value: "de",
- label: "Deutsch",
- },
- {
- value: "th",
- label: "ยาสุฟุมิ",
- },
];
const LocaleSelector = () => {
@@ -46,14 +30,7 @@ const LocaleSelector = () => {
settingStore.setLocale(locale);
};
- return (
-
- );
+ return
;
};
export default LocaleSelector;
diff --git a/src/components/MarkdownRenderer.tsx b/src/components/MarkdownRenderer.tsx
new file mode 100644
index 00000000..7e2445d3
--- /dev/null
+++ b/src/components/MarkdownRenderer.tsx
@@ -0,0 +1,33 @@
+import React, { useState, useEffect } from "react";
+import ReactMarkdown from "react-markdown";
+
+interface MarkdownRendererProps {
+ url: string;
+}
+
+function MarkdownRenderer(props: MarkdownRendererProps) {
+ const [markdown, setMarkdown] = useState
("");
+
+ useEffect(() => {
+ const fetchData = async () => {
+ const response = await fetch(props.url);
+ const data = await response.text();
+ setMarkdown(data);
+ };
+ fetchData();
+ }, [props.url]);
+
+ return (
+ {children}
,
+ li: ({ children }) => {children} ,
+ }}
+ >
+ {markdown}
+
+ );
+}
+
+export default MarkdownRenderer;
diff --git a/src/components/OpenAIApiConfigView.tsx b/src/components/OpenAIApiConfigView.tsx
index c01c46ae..c86e56dc 100644
--- a/src/components/OpenAIApiConfigView.tsx
+++ b/src/components/OpenAIApiConfigView.tsx
@@ -3,14 +3,64 @@ import { useTranslation } from "react-i18next";
import { useDebounce } from "react-use";
import { useSettingStore } from "@/store";
import { OpenAIApiConfig } from "@/types";
+import { allowSelfOpenAIKey, hasFeature } from "@/utils";
+import Radio from "./kit/Radio";
import TextField from "./kit/TextField";
+import Tooltip from "./kit/Tooltip";
const OpenAIApiConfigView = () => {
const { t } = useTranslation();
const settingStore = useSettingStore();
- const [openAIApiConfig, setOpenAIApiConfig] = useState(
- settingStore.setting.openAIApiConfig
- );
+ const [openAIApiConfig, setOpenAIApiConfig] = useState(settingStore.setting.openAIApiConfig);
+ const [maskKey, setMaskKey] = useState(true);
+
+ const models = [
+ {
+ id: "gpt-3.5-turbo",
+ title: `GPT-3.5 Turbo`,
+ cost: 1,
+ disabled: false,
+ tooltip: "",
+ },
+ {
+ id: "gpt-4-turbo",
+ title: `GPT-4 Turbo`,
+ cost: 20,
+ disabled: false,
+ tooltip: "",
+ },
+ {
+ id: "gpt-4",
+ title: `GPT-4`,
+ cost: 60,
+ disabled: false,
+ tooltip: "",
+ },
+ {
+ id: "gpt-4o",
+ title: `GPT-4o`,
+ cost: 10,
+ disabled: false,
+ tooltip: "",
+ },
+ {
+ id: "deepseek-chat",
+ title: `Deepseek Chat`,
+ cost: 1,
+ disabled: false,
+ tooltip: "",
+ },
+ ];
+
+ const maskedKey = (str: string) => {
+ if (str.length < 7) {
+ return str;
+ }
+ const firstThree = str.slice(0, 3);
+ const lastFour = str.slice(-4);
+ const middle = ".".repeat(str.length - 7);
+ return `${firstThree}${middle}${lastFour}`;
+ };
useDebounce(
() => {
@@ -25,27 +75,78 @@ const OpenAIApiConfigView = () => {
...openAIApiConfig,
...config,
});
+ setMaskKey(false);
+ };
+
+ const modelRadio = (model: any) => {
+ return (
+
+ handleSetOpenAIApiConfig({ model: value })}
+ />
+
+ {model.title} {hasFeature("quota") ? `(${t("setting.openai-api-configuration.quota-per-ask", { count: model.cost })})` : ""}
+
+
+ );
};
return (
<>
-
- OpenAI API Key
- handleSetOpenAIApiConfig({ key: value })}
- />
-
-
-
API Endpoint
-
handleSetOpenAIApiConfig({ endpoint: value })}
- />
+
+
{t("setting.openai-api-configuration.model")}
+ {allowSelfOpenAIKey() &&
{t("setting.openai-api-configuration.model-description")}
}
+
+
+ {models.map((model) =>
+ model.disabled ? (
+
+ {modelRadio(model)}
+
+ ) : (
+ modelRadio(model)
+ )
+ )}
+
+
+
OpenAI API Key
+
{t("setting.openai-api-configuration.key-description")}
+
handleSetOpenAIApiConfig({ key: value })}
+ />
+
+
OpenAI API Endpoint
+
+
handleSetOpenAIApiConfig({ endpoint: value })}
+ />
+
);
diff --git a/src/components/PaymentHistoryTable.tsx b/src/components/PaymentHistoryTable.tsx
new file mode 100644
index 00000000..59dd1e61
--- /dev/null
+++ b/src/components/PaymentHistoryTable.tsx
@@ -0,0 +1,82 @@
+import axios from "axios";
+import { t } from "i18next";
+import { useSession } from "next-auth/react";
+import { useEffect, useState } from "react";
+import { Payment } from "@/types";
+import { getCurrencySymbol, getDateString } from "@/utils";
+
+const PaymentHistoryTable = () => {
+ const [list, setList] = useState
([]);
+ const { data: session } = useSession();
+
+ useEffect(() => {
+ const refreshPaymentList = async (userId: string) => {
+ let list: Payment[] = [];
+ try {
+ const { data } = await axios.get("/api/payment", {
+ headers: { Authorization: `Bearer ${userId}` },
+ });
+ list = data;
+ } catch (error) {
+ // do nth
+ }
+ setList(list);
+ };
+
+ if (session?.user.id) {
+ refreshPaymentList(session.user.id);
+ }
+ }, [session]);
+
+ if (list.length === 0) {
+ return <>;
+ }
+
+ return (
+
+
+
+
+
+
+
+
+ {t("common.date")}
+
+
+ {t("common.description")}
+
+
+ {t("common.amount")}
+
+
+
+
+ {list.map((payment) => (
+
+
+ {getDateString(payment.createdAt)}
+
+ {payment.description}
+
+ {getCurrencySymbol(payment.currency.toLocaleUpperCase())}
+ {payment.amount / 100}
+
+
+
+ {t("setting.subscription.view-receipt")}
+
+
+
+ ))}
+
+
+
+
+
+
-
-
- {t("setting.plan.pro-question-per-month")}
-
-
-
(session?.user?.email ? checkout() : signIn())}
- >
- {session?.user?.email
- ? t("setting.plan.early-bird-checkout")
- : t("payment.login-to-buy")}
-
+
+
+
+ {t("setting.plan.n-question-per-month", {
+ count: 1000,
+ })}
+
+
+
+
+ {tiers.map((tier) => (
+
+
+ {tier.name} - {tier.priceMonthly}
+
+ (session?.user?.email ? checkout(tier.priceId) : signIn())}
+ className="mt-6 block rounded-md bg-indigo-600 px-4 py-2.5 text-center text-xl font-semibold leading-6 text-white shadow-sm hover:bg-indigo-500 focus-visible:outline focus-visible:outline-2 focus-visible:outline-offset-2 focus-visible:outline-indigo-600"
+ >
+ {session?.user?.email ? tier.buyButton : t("payment.sign-in-to-buy")}
+
+
+ ))}
diff --git a/src/components/ProductHuntBanner.tsx b/src/components/ProductHuntBanner.tsx
index d090d4e3..2985e3e8 100644
--- a/src/components/ProductHuntBanner.tsx
+++ b/src/components/ProductHuntBanner.tsx
@@ -9,10 +9,7 @@ interface Props {
const ProductHuntBanner = (props: Props) => {
const { className } = props;
const { t } = useTranslation();
- const [hideBanner, setHideBanner] = useLocalStorage(
- "hide-product-hunt-banner",
- false
- );
+ const [hideBanner, setHideBanner] = useLocalStorage("hide-product-hunt-banner", false);
const show = !hideBanner;
return (
@@ -28,10 +25,7 @@ const ProductHuntBanner = (props: Props) => {
>
{t("banner.product-hunt")}
-
setHideBanner(true)}
- >
+ setHideBanner(true)}>
diff --git a/src/components/QueryDrawer.tsx b/src/components/QueryDrawer.tsx
index 177dc553..81c774b1 100644
--- a/src/components/QueryDrawer.tsx
+++ b/src/components/QueryDrawer.tsx
@@ -3,7 +3,7 @@ import { useEffect, useState } from "react";
import { toast } from "react-hot-toast";
import { useTranslation } from "react-i18next";
import TextareaAutosize from "react-textarea-autosize";
-import { useQueryStore } from "@/store";
+import { useQueryStore, useMessageStore } from "@/store";
import { ExecutionResult, ResponseObject } from "@/types";
import { checkStatementIsSelect, getMessageFromExecutionResult } from "@/utils";
import Tooltip from "./kit/Tooltip";
@@ -16,17 +16,14 @@ import ExecutionWarningBanner from "./ExecutionView/ExecutionWarningBanner";
const QueryDrawer = () => {
const { t } = useTranslation();
const queryStore = useQueryStore();
- const [executionResult, setExecutionResult] = useState<
- ExecutionResult | undefined
- >(undefined);
+ const messageStore = useMessageStore();
+ const [executionResult, setExecutionResult] = useState
(undefined);
+ const [originalStatement, setOriginalStatement] = useState("");
const [statement, setStatement] = useState("");
const [isLoading, setIsLoading] = useState(false);
const context = queryStore.context;
- const executionMessage = executionResult
- ? getMessageFromExecutionResult(executionResult)
- : "";
- const showExecutionWarningBanner =
- statement.trim() && !checkStatementIsSelect(statement);
+ const executionMessage = executionResult ? getMessageFromExecutionResult(executionResult) : "";
+ const showExecutionWarningBanner = statement.trim() && !checkStatementIsSelect(statement);
useEffect(() => {
if (!queryStore.showDrawer) {
@@ -35,12 +32,24 @@ const QueryDrawer = () => {
const statement = context?.statement || "";
setStatement(statement);
+ // Save original statement when open QueryDrawer.
+ if (!originalStatement) {
+ setOriginalStatement(statement);
+ }
+
if (statement !== "" && checkStatementIsSelect(statement)) {
executeStatement(statement);
}
setExecutionResult(undefined);
}, [context, queryStore.showDrawer]);
+ // Reset original statement to "" when close QueryDrawer.
+ useEffect(() => {
+ if (!queryStore.showDrawer) {
+ setOriginalStatement("");
+ }
+ }, [queryStore.showDrawer]);
+
const executeStatement = async (statement: string) => {
if (!statement) {
toast.error("Please enter a statement.");
@@ -86,82 +95,63 @@ const QueryDrawer = () => {
}
};
- const close = () => queryStore.toggleDrawer(false);
+ const close = () => {
+ if (originalStatement !== statement && context?.messageId) {
+ messageStore.updateStatement(context?.messageId, originalStatement, statement);
+ }
+ queryStore.toggleDrawer(false);
+ };
return (
-
-
-
+
+
+
{t("execution.title")}
{!context ? (
- {t("execution.message.no-connection")}
-
+ {t("execution.message.no-connection")}
) : (
<>
{t("connection.self")}:
- {context.database?.name}
- {showExecutionWarningBanner && (
-
- )}
+ {showExecutionWarningBanner &&
}
setStatement(e.target.value)}
/>
-
- executeStatement(statement)}
- >
-
-
+ executeStatement(statement)}
+ >
+ {t("common.run-sql")}
+
{isLoading ? (
- {t("execution.message.executing")}
-
+ {t("execution.message.executing")}
) : (
<>
{executionResult ? (
executionMessage ? (
-
+
) : (
-
+
)
) : (
<>
diff --git a/src/components/QuotaOverflowBanner.tsx b/src/components/QuotaOverflowBanner.tsx
index 33154de1..791b9039 100644
--- a/src/components/QuotaOverflowBanner.tsx
+++ b/src/components/QuotaOverflowBanner.tsx
@@ -1,5 +1,6 @@
import { useTranslation } from "react-i18next";
import { useLocalStorage } from "react-use";
+import { allowSelfOpenAIKey } from "@/utils";
import Link from "next/link";
import Icon from "./Icon";
@@ -10,10 +11,7 @@ interface Props {
const QuotaOverflowBanner = (props: Props) => {
const { className } = props;
const { t } = useTranslation();
- const [hideBanner, setHideBanner] = useLocalStorage(
- "hide-quota-overflow-banner",
- false
- );
+ const [hideBanner, setHideBanner] = useLocalStorage("hide-quota-overflow-banner", false);
const show = !hideBanner;
return (
@@ -25,14 +23,13 @@ const QuotaOverflowBanner = (props: Props) => {
>
{t("banner.quota-overflow")}{" "}
-
-
setHideBanner(true)}
- >
+ setHideBanner(true)}>
diff --git a/src/components/QuotaView.tsx b/src/components/QuotaView.tsx
new file mode 100644
index 00000000..e292312b
--- /dev/null
+++ b/src/components/QuotaView.tsx
@@ -0,0 +1,91 @@
+import { signIn, useSession } from "next-auth/react";
+import axios from "axios";
+import { useEffect, useState } from "react";
+import Link from "next/link";
+import { useTranslation } from "react-i18next";
+import { Quota } from "@/types";
+import getEventEmitter from "@/utils/event-emitter";
+import { allowSelfOpenAIKey } from "@/utils";
+
+interface Props {
+ className?: string;
+}
+
+const QuotaView = (props: Props) => {
+ const [quota, setQuota] = useState
({ current: 0, limit: 0 });
+ const { t } = useTranslation();
+ const { data: session } = useSession();
+
+ const showSupplyOwnKey = !session || quota.current >= quota.limit;
+ const expired = session?.user?.subscription?.expireAt && session?.user?.subscription?.expireAt < Date.now();
+ const showActionButton = !session || session.user.subscription.plan === "FREE" || expired;
+
+ const refreshQuota = async (userId: string) => {
+ let quota: Quota = { current: 0, limit: 0 };
+ try {
+ const { data } = await axios.get("/api/usage", {
+ headers: { Authorization: `Bearer ${userId}` },
+ });
+ quota = data;
+ } catch (error) {
+ // do nth
+ }
+ setQuota(quota);
+ };
+
+ getEventEmitter().on("usage.update", () => {
+ if (session?.user.id) {
+ refreshQuota(session.user.id);
+ }
+ });
+
+ useEffect(() => {
+ if (session?.user.id) {
+ refreshQuota(session.user.id);
+ }
+ }, [session]);
+
+ return (
+
+
+
+ {session ? t(`setting.plan.${session.user.subscription.plan.toLowerCase()}`) : t("setting.plan.guest")}
+
+ {!!expired && (
+
+ {t("setting.plan.expired")}
+
+ )}
+
+
+
{t("common.quota")}
+
= quota.limit ? "text-red-600" : "text-black dark:text-gray-300"}>
+ {quota.current}/{quota.limit}
+
+
+ {!!showActionButton &&
+ (session ? (
+
+ {expired ? t("setting.plan.renew") : t("setting.plan.upgrade")}
+
+ ) : (
+
signIn()}
+ >
+ {t("setting.plan.signup-for-more")}
+
+ ))}
+ {allowSelfOpenAIKey() && !!showSupplyOwnKey && (
+
+ {t("banner.use-my-key")}
+
+ )}
+
({ current: 0, limit: 0 });
- const { t } = useTranslation();
- const { data: session } = useSession();
-
- useEffect(() => {
- const refreshQuota = async () => {
- let quota: Quota = { current: 0, limit: 0 };
- try {
- const { data } = await axios.get("/api/quota", {});
- quota = data;
- } catch (error) {
- // do nth
- }
- setQuota(quota);
- };
-
- refreshQuota();
- }, []);
-
- return (
-
-
-
- {t("setting.plan.free")}
-
-
-
-
{t("common.quota")}
-
- {quota.current}/{quota.limit}
-
-
- {session ? (
-
- {t("setting.plan.upgrade")}
-
- ) : (
-
signIn()}
- >
- {t("setting.plan.signup-for-more")}
-
- )}
-
* ;
+};
+
+export default RequiredStar;
diff --git a/src/components/SchemaDrawer.tsx b/src/components/SchemaDrawer.tsx
new file mode 100644
index 00000000..95b38217
--- /dev/null
+++ b/src/components/SchemaDrawer.tsx
@@ -0,0 +1,70 @@
+import { encode } from "@nem035/gpt-3-encoder";
+import { Drawer } from "@mui/material";
+import { useEffect, useState } from "react";
+import Icon from "./Icon";
+import { getAssistantById, getPromptGeneratorOfAssistant, useConnectionStore, useConversationStore, useSettingStore } from "@/store";
+import { getModel, generateDbPromptFromContext } from "@/utils";
+import { CodeBlock } from "./CodeBlock";
+import { useTranslation } from "react-i18next";
+
+interface Props {
+ close: () => void;
+}
+
+const SchemaDrawer = (props: Props) => {
+ const { t } = useTranslation();
+ const conversationStore = useConversationStore();
+ const connectionStore = useConnectionStore();
+ const settingStore = useSettingStore();
+ const currentConversation = conversationStore.getConversationById(conversationStore.currentConversationId);
+ const maxToken = getModel(settingStore.setting.openAIApiConfig?.model || "").max_token;
+ const [prompt, setPrompt] = useState("");
+
+ const getPrompt = async () => {
+ if (!currentConversation) return;
+ const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
+ let dbPrompt = promptGenerator();
+ if (connectionStore.currentConnectionCtx?.database) {
+ const schemaList = await connectionStore.getOrFetchDatabaseSchema(connectionStore.currentConnectionCtx?.database);
+ dbPrompt = generateDbPromptFromContext(
+ promptGenerator,
+ connectionStore.currentConnectionCtx.connection.engineType,
+ schemaList,
+ currentConversation.selectedSchemaName || "",
+ currentConversation.selectedTableNameList || [],
+ maxToken
+ );
+ }
+ setPrompt(dbPrompt);
+ };
+
+ useEffect(() => {
+ // TODO: initial state with current conversation.
+ }, []);
+
+ useEffect(() => {
+ getPrompt();
+ }, []);
+
+ const close = () => props.close();
+ return (
+
+
+
+
+
+
{t("prompt.current-conversation")}
+
+ {t("connection.total-token")} {encode(prompt).length}/{maxToken}
+
+
+
+
+
+ );
+};
+
+export default SchemaDrawer;
diff --git a/src/components/SettingAvatarIcon.tsx b/src/components/SettingAvatarIcon.tsx
index 39454168..33b77d26 100644
--- a/src/components/SettingAvatarIcon.tsx
+++ b/src/components/SettingAvatarIcon.tsx
@@ -1,9 +1,9 @@
import { useSession } from "next-auth/react";
import { useTranslation } from "react-i18next";
import Link from "next/link";
+import { hasFeature } from "../utils";
import Tooltip from "./kit/Tooltip";
import Icon from "./Icon";
-import { HasFeature } from "../utils";
interface Props {}
@@ -14,7 +14,7 @@ const SettingAvatarIcon = (props: Props) => {
return (
- {(!HasFeature("account") || !session) && (
+ {(!hasFeature("account") || !session) && (
{
)}
- {HasFeature("account") && session?.user && (
+ {hasFeature("account") && session?.user && (
{session.user.image ? (
{
/>
) : (
- {session.user.name?.charAt(0)}
+ {session.user.name ? session.user.name.charAt(0) : session.user.email?.charAt(0)}
)}
diff --git a/src/components/SettingView.tsx b/src/components/SettingGeneralView.tsx
similarity index 73%
rename from src/components/SettingView.tsx
rename to src/components/SettingGeneralView.tsx
index 04e483ad..25906f5a 100644
--- a/src/components/SettingView.tsx
+++ b/src/components/SettingGeneralView.tsx
@@ -1,20 +1,21 @@
import React from "react";
import { useTranslation } from "react-i18next";
+import { hasFeature } from "../utils";
import Icon from "./Icon";
import AccountView from "./AccountView";
+import DebugView from "./DebugView";
import PricingView from "./PricingView";
import WeChatQRCodeView from "./WeChatQRCodeView";
import ClearDataButton from "./ClearDataButton";
import LocaleSelector from "./LocaleSelector";
import ThemeSelector from "./ThemeSelector";
import OpenAIApiConfigView from "./OpenAIApiConfigView";
-import { HasFeature } from "../utils";
-const SettingView = () => {
+const SettingGeneralView = () => {
const { t } = useTranslation();
return (
-
+
- {HasFeature("account") && (
+ {hasFeature("debug") && (
)}
- {HasFeature("payment") && (
+ {hasFeature("payment") && (
)}
+ {hasFeature("account") && (
+
+ )}
+
@@ -58,8 +65,18 @@ const SettingView = () => {
+
+
);
};
-export default SettingView;
+export default SettingGeneralView;
diff --git a/src/components/Sidebar/ConnectionList.tsx b/src/components/Sidebar/ConnectionList.tsx
index 9dc2e375..da0e912e 100644
--- a/src/components/Sidebar/ConnectionList.tsx
+++ b/src/components/Sidebar/ConnectionList.tsx
@@ -20,8 +20,7 @@ const ConnectionList = () => {
showCreateConnectionModal: false,
showUpdateConversationModal: false,
});
- const [editConnectionModalContext, setEditConnectionModalContext] =
- useState
();
+ const [editConnectionModalContext, setEditConnectionModalContext] = useState();
const connectionList = connectionStore.connectionList;
const currentConnectionCtx = connectionStore.currentConnectionCtx;
@@ -34,9 +33,7 @@ const ConnectionList = () => {
};
const handleConnectionSelect = async (connection: Connection) => {
- const databaseList = await connectionStore.getOrFetchDatabaseList(
- connection
- );
+ const databaseList = await connectionStore.getOrFetchDatabaseList(connection);
connectionStore.setCurrentConnectionCtx({
connection,
database: head(databaseList),
@@ -54,10 +51,7 @@ const ConnectionList = () => {
return (
<>
connectionStore.setCurrentConnectionCtx(undefined)}
>
handleConnectionSelect(connection)}
>
@@ -80,10 +73,7 @@ const ConnectionList = () => {
>
))}
@@ -97,10 +87,7 @@ const ConnectionList = () => {
{state.showCreateConnectionModal && (
- toggleCreateConnectionModal(false)}
- />
+ toggleCreateConnectionModal(false)} />
)}
);
diff --git a/src/components/Sidebar/ConversationList.tsx b/src/components/Sidebar/ConversationList.tsx
index dc536496..eb99b386 100644
--- a/src/components/Sidebar/ConversationList.tsx
+++ b/src/components/Sidebar/ConversationList.tsx
@@ -1,10 +1,6 @@
import { useState } from "react";
import { useTranslation } from "react-i18next";
-import {
- useConversationStore,
- useConnectionStore,
- useLayoutStore,
-} from "@/store";
+import { useConversationStore, useConnectionStore, useLayoutStore } from "@/store";
import { Conversation } from "@/types";
import Dropdown, { DropdownItem } from "../kit/Dropdown";
import Icon from "../Icon";
@@ -22,8 +18,7 @@ const ConversationList = () => {
const [state, setState] = useState({
showUpdateConversationModal: false,
});
- const [updateConversationModalContext, setUpdateConversationModalContext] =
- useState();
+ const [updateConversationModalContext, setUpdateConversationModalContext] = useState();
const currentConnectionCtx = connectionStore.currentConnectionCtx;
const conversationList = conversationStore.conversationList.filter(
(conversation) =>
@@ -42,10 +37,7 @@ const ConversationList = () => {
if (!currentConnectionCtx) {
conversationStore.createConversation();
} else {
- conversationStore.createConversation(
- currentConnectionCtx.connection.id,
- currentConnectionCtx.database?.name
- );
+ conversationStore.createConversation(currentConnectionCtx.connection.id, currentConnectionCtx.database?.name);
}
};
@@ -77,8 +69,7 @@ const ConversationList = () => {
handleConversationSelect(conversation)}
>
@@ -87,9 +78,7 @@ const ConversationList = () => {
) : (
)}
-
- {conversation.title || "SQL Chat"}
-
+
{conversation.title || "SQL Chat"}
@@ -125,10 +114,7 @@ const ConversationList = () => {
{updateConversationModalContext && state.showUpdateConversationModal && (
- toggleUpdateConversationModal(false)}
- conversation={updateConversationModalContext}
- />
+ toggleUpdateConversationModal(false)} conversation={updateConversationModalContext} />
)}
);
diff --git a/src/components/StripeCheckPaymentBanner.tsx b/src/components/StripeCheckPaymentBanner.tsx
index 8d68facc..b6c04ec0 100644
--- a/src/components/StripeCheckPaymentBanner.tsx
+++ b/src/components/StripeCheckPaymentBanner.tsx
@@ -1,9 +1,8 @@
+import useSWR from "swr";
import { useTranslation } from "react-i18next";
import { useRouter } from "next/router";
import { useState } from "react";
import { fetchGetJSON } from "../utils/api-helpers";
-import useSWR from "swr";
-import Link from "next/link";
import Icon from "./Icon";
interface Props {
@@ -20,10 +19,7 @@ const StripeCheckPaymentBanner = (props: Props) => {
// Fetch CheckoutSession from static page via
// https://nextjs.org/docs/basic-features/data-fetching#static-generation
- const { data, error } = useSWR(
- router.query.session_id ? `/api/checkout_sessions/${sessionId}` : null,
- fetchGetJSON
- );
+ const { data } = useSWR(router.query.session_id ? `/api/checkout_sessions/${sessionId}` : null, fetchGetJSON);
return (
<>
@@ -33,13 +29,9 @@ const StripeCheckPaymentBanner = (props: Props) => {
} relative w-full flex flex-row justify-start sm:justify-center items-center py-1 bg-gray-100 dark:bg-zinc-700`}
>
- {t("payment.self")}{" "}
- {data?.payment_intent?.status ?? t("common.loading")}
+ {t("payment.self")} {data?.payment_intent?.status ?? t("common.loading")}
- setHideBanner(true)}
- >
+ setHideBanner(true)}>
+
);
diff --git a/src/components/UpdateConversationModal.tsx b/src/components/UpdateConversationModal.tsx
index c883f39c..d6feedb7 100644
--- a/src/components/UpdateConversationModal.tsx
+++ b/src/components/UpdateConversationModal.tsx
@@ -27,9 +27,7 @@ const UpdateConversationModal = (props: Props) => {
label: assistant.name,
};
});
- const currentAssistant = assistantList.find(
- (assistant) => assistant.id === assistantId
- );
+ const currentAssistant = assistantList.find((assistant) => assistant.id === assistantId);
const handleSaveEdit = () => {
const formatedTitle = title.trim();
@@ -48,30 +46,17 @@ const UpdateConversationModal = (props: Props) => {
return (
-
- {t("conversation.title")}
-
- setTitle(value)}
- />
+ {t("conversation.title")}
+ setTitle(value)} />
diff --git a/src/components/kit/Checkbox.tsx b/src/components/kit/Checkbox.tsx
new file mode 100644
index 00000000..f64abe63
--- /dev/null
+++ b/src/components/kit/Checkbox.tsx
@@ -0,0 +1,34 @@
+import * as CheckboxUI from "@radix-ui/react-checkbox";
+import { CheckIcon, DividerHorizontalIcon } from "@radix-ui/react-icons";
+import { ReactNode } from "react";
+
+interface CheckboxProps {
+ value: boolean;
+ label: string;
+ onValueChange: (tableName: string, value: boolean) => void;
+}
+const Checkbox = (props: CheckboxProps & { children?: ReactNode }) => {
+ const { value, label, onValueChange, children } = props;
+ return (
+
+ );
+};
+
+export default Checkbox;
diff --git a/src/components/kit/Modal.tsx b/src/components/kit/Modal.tsx
index 5fccfd34..b328a54f 100644
--- a/src/components/kit/Modal.tsx
+++ b/src/components/kit/Modal.tsx
@@ -19,9 +19,7 @@ const Modal = (props: Props) => {
className || ""
} flex flex-col bg-white dark:bg-zinc-800 rounded-xl p-4 fixed top-[50%] left-[50%] h-auto max-h-[85vh] w-[90vw] max-w-[90vw] sm:max-w-lg translate-x-[-50%] translate-y-[-50%] z-100 outline-none`}
>
-
- {title}
-
+
{title}
{
>
-
- {children}
-
+
{children}
{children}
diff --git a/src/components/kit/Radio.tsx b/src/components/kit/Radio.tsx
new file mode 100644
index 00000000..694c887e
--- /dev/null
+++ b/src/components/kit/Radio.tsx
@@ -0,0 +1,39 @@
+import { HTMLInputTypeAttribute } from "react";
+
+interface Props {
+ value: string;
+ onChange?: (value: string) => void;
+ type?: HTMLInputTypeAttribute;
+ className?: string;
+ disabled?: boolean;
+ checked?: boolean;
+}
+
+const getDefaultProps = () => ({
+ value: "",
+ onChange: () => {},
+ type: "radio",
+ className: "",
+ disabled: false,
+ checked: false,
+});
+
+const Radio = (props: Props) => {
+ const { value, disabled, className, type, checked, onChange } = {
+ ...getDefaultProps(),
+ ...props,
+ };
+
+ return (
+ 7
- ? "h-80 overflow-hidden"
- : "max-h-80 overflow-auto"
+ itemList.length > 7 ? "h-80 overflow-hidden" : "max-h-80 overflow-auto"
} border dark:border-zinc-800 rounded-lg drop-shadow-lg`}
type="auto"
>
-
+
- {placeholder && (
-
- {placeholder}
-
- )}
+ {placeholder && {placeholder} }
{itemList.map((item) => (
-
- {item.label}
-
+ {item.label}
diff --git a/src/components/kit/TextField.tsx b/src/components/kit/TextField.tsx
index 604cdfe5..bcb1020d 100644
--- a/src/components/kit/TextField.tsx
+++ b/src/components/kit/TextField.tsx
@@ -26,9 +26,7 @@ const TextField = (props: Props) => {
return (
;
- execute: (
- databaseName: string,
- statement: string
- ) => Promise;
+ execute: (databaseName: string, statement: string) => Promise;
getDatabases: () => Promise;
- getTables: (databaseName: string) => Promise;
- getTableStructure: (
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
- ) => Promise;
- getTableStructureBatch: (
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
- ) => Promise;
+ getTableSchema: (databaseName: string) => Promise;
}
export const newConnector = (connection: Connection): Connector => {
@@ -31,6 +18,8 @@ export const newConnector = (connection: Connection): Connector => {
return postgres(connection);
case Engine.MSSQL:
return mssql(connection);
+ case Engine.OceanBase:
+ return mysql(connection);
default:
throw new Error("Unsupported engine type.");
}
diff --git a/src/lib/connectors/mssql/index.ts b/src/lib/connectors/mssql/index.ts
index 877014d2..929aa04e 100644
--- a/src/lib/connectors/mssql/index.ts
+++ b/src/lib/connectors/mssql/index.ts
@@ -1,12 +1,10 @@
import { ConnectionPool } from "mssql";
-import { Connection, ExecutionResult } from "@/types";
+import { Connection, ExecutionResult, Schema, Table } from "@/types";
import { Connector } from "..";
const systemDatabases = ["master", "tempdb", "model", "msdb"];
-const getMSSQLConnection = async (
- connection: Connection
-): Promise => {
+const getMSSQLConnection = async (connection: Connection): Promise => {
const connectionOptions: any = {
server: connection.host,
port: parseInt(connection.port),
@@ -34,11 +32,7 @@ const testConnection = async (connection: Connection): Promise => {
return true;
};
-const execute = async (
- connection: Connection,
- databaseName: string,
- statement: string
-): Promise => {
+const execute = async (connection: Connection, databaseName: string, statement: string): Promise => {
const pool = await getMSSQLConnection(connection);
const request = pool.request();
const result = await request.query(`USE ${databaseName}; ${statement}`);
@@ -54,11 +48,7 @@ const execute = async (
const getDatabases = async (connection: Connection): Promise => {
const pool = await getMSSQLConnection(connection);
const request = pool.request();
- const result = await request.query(
- `SELECT name FROM sys.databases WHERE name NOT IN ('${systemDatabases.join(
- "','"
- )}');`
- );
+ const result = await request.query(`SELECT name FROM sys.databases WHERE name NOT IN ('${systemDatabases.join("','")}');`);
await pool.close();
const databaseList = [];
for (const row of result.recordset) {
@@ -69,111 +59,52 @@ const getDatabases = async (connection: Connection): Promise => {
return databaseList;
};
-const getTables = async (
- connection: Connection,
- databaseName: string
-): Promise => {
+const getTableSchema = async (connection: Connection, databaseName: string): Promise => {
const pool = await getMSSQLConnection(connection);
const request = pool.request();
+ const schemaList: Schema[] = [];
const result = await request.query(
- `SELECT TABLE_NAME as table_name FROM ${databaseName}.INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE='BASE TABLE';`
+ `SELECT TABLE_NAME as table_name, TABLE_SCHEMA as table_schema FROM ${databaseName}.INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE='BASE TABLE';`
);
- await pool.close();
- const tableList = [];
for (const row of result.recordset) {
if (row["table_name"]) {
- tableList.push(row["table_name"]);
+ const schema = schemaList.find((schema) => schema.name === row["table_schema"]);
+ if (schema) {
+ schema.tables.push({ name: row["table_name"] as string, structure: "" } as Table);
+ } else {
+ schemaList.push({
+ name: row["table_schema"],
+ tables: [{ name: row["table_name"], structure: "" } as Table],
+ });
+ }
}
}
- return tableList;
-};
-const getTableStructure = async (
- connection: Connection,
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- const pool = await getMSSQLConnection(connection);
- const request = pool.request();
- const { recordset } = await request.query(
- `SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE FROM ${databaseName}.INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='dbo' AND TABLE_NAME='${tableName}';`
- );
-
- const columnList = [];
- // Transform to standard schema string.
- for (const row of recordset) {
- columnList.push(
- `${row["COLUMN_NAME"]} ${row["DATA_TYPE"].toUpperCase()} ${
- String(row["IS_NULLABLE"]).toUpperCase() === "NO" ? "NOT NULL" : ""
- }`
- );
- }
- structureFetched(
- tableName,
- `CREATE TABLE [${tableName}] (
- ${columnList.join(",\n")}
- );`
- );
-};
-
-const getTableStructureBatch = async (
- connection: Connection,
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- const pool = await getMSSQLConnection(connection);
- const request = pool.request();
-
- await Promise.all(
- tableNameList.map(async (tableName) => {
+ for (const schema of schemaList) {
+ for (const table of schema.tables) {
const { recordset } = await request.query(
- `SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE FROM ${databaseName}.INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='dbo' AND TABLE_NAME='${tableName}';`
+ `SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE FROM ${databaseName}.INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='${schema.name}' AND TABLE_NAME='${table.name}';`
);
const columnList = [];
// Transform to standard schema string.
for (const row of recordset) {
columnList.push(
- `${row["COLUMN_NAME"]} ${row["DATA_TYPE"].toUpperCase()} ${
- String(row["IS_NULLABLE"]).toUpperCase() === "NO" ? "NOT NULL" : ""
- }`
+ `${row["COLUMN_NAME"]} ${row["DATA_TYPE"].toUpperCase()} ${String(row["IS_NULLABLE"]).toUpperCase() === "NO" ? "NOT NULL" : ""}`
);
}
- structureFetched(
- tableName,
- `CREATE TABLE [${tableName}] (
- ${columnList.join(",\n")}
- );`
- );
- })
- );
+ table.structure = `CREATE TABLE [${table.name}] (\n${columnList.join(",\n")}\n);`;
+ }
+ }
+ await pool.close();
+ return schemaList;
};
const newConnector = (connection: Connection): Connector => {
return {
testConnection: () => testConnection(connection),
- execute: (databaseName: string, statement: string) =>
- execute(connection, databaseName, statement),
+ execute: (databaseName: string, statement: string) => execute(connection, databaseName, statement),
getDatabases: () => getDatabases(connection),
- getTables: (databaseName: string) => getTables(connection, databaseName),
- getTableStructure: (
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructure(connection, databaseName, tableName, structureFetched),
- getTableStructureBatch: (
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructureBatch(
- connection,
- databaseName,
- tableNameList,
- structureFetched
- ),
+ getTableSchema: (databaseName: string) => getTableSchema(connection, databaseName),
};
};
diff --git a/src/lib/connectors/mysql/index.ts b/src/lib/connectors/mysql/index.ts
index 81cf9a3b..76bde3d7 100644
--- a/src/lib/connectors/mysql/index.ts
+++ b/src/lib/connectors/mysql/index.ts
@@ -1,18 +1,11 @@
import { ConnectionOptions } from "mysql2";
import mysql, { RowDataPacket } from "mysql2/promise";
-import { Connection, ExecutionResult } from "@/types";
+import { Connection, ExecutionResult, Table, Schema } from "@/types";
import { Connector } from "..";
-const systemDatabases = [
- "information_schema",
- "mysql",
- "performance_schema",
- "sys",
-];
+const systemDatabases = ["information_schema", "mysql", "performance_schema", "sys"];
-const getMySQLConnection = async (
- connection: Connection
-): Promise => {
+const getMySQLConnection = async (connection: Connection): Promise => {
const connectionOptions: ConnectionOptions = {
host: connection.host,
port: parseInt(connection.port),
@@ -26,8 +19,28 @@ const getMySQLConnection = async (
cert: connection.ssl?.cert,
key: connection.ssl?.key,
};
+ } else {
+ // rejectUnauthorized=false to infer sslmode=prefer since hosted MySQL venders have SSL enabled.
+ connectionOptions.ssl = {
+ rejectUnauthorized: false,
+ };
+ }
+
+ let conn;
+ if (connection.ssl) {
+ conn = await mysql.createConnection(connectionOptions);
+ } else {
+ try {
+ conn = await mysql.createConnection(connectionOptions);
+ } catch (error) {
+ if (error instanceof Error && error.message.includes("Server does not support secure")) {
+ connectionOptions.ssl = undefined;
+ conn = await mysql.createConnection(connectionOptions);
+ } else {
+ throw error;
+ }
+ }
}
- const conn = await mysql.createConnection(connectionOptions);
return conn;
};
@@ -37,11 +50,7 @@ const testConnection = async (connection: Connection): Promise => {
return true;
};
-const execute = async (
- connection: Connection,
- databaseName: string,
- statement: string
-): Promise => {
+const execute = async (connection: Connection, databaseName: string, statement: string): Promise => {
connection.database = databaseName;
const conn = await getMySQLConnection(connection);
const [rows] = await conn.execute(statement);
@@ -75,89 +84,39 @@ const getDatabases = async (connection: Connection): Promise => {
return databaseList;
};
-const getTables = async (
- connection: Connection,
- databaseName: string
-): Promise => {
+const getTableSchema = async (connection: Connection, databaseName: string): Promise => {
const conn = await getMySQLConnection(connection);
+ // get All tableList from database
const [rows] = await conn.query(
- `SELECT TABLE_NAME as table_name FROM information_schema.tables WHERE TABLE_SCHEMA=? AND TABLE_TYPE='BASE TABLE';`,
+ // SYSTEM VERSIONED is a special table for MariaDB https://mariadb.com/kb/en/system-versioned-tables/
+ `SELECT TABLE_NAME as table_name FROM information_schema.tables WHERE TABLE_SCHEMA=? AND (TABLE_TYPE='BASE TABLE' || TABLE_TYPE='SYSTEM VERSIONED');`,
[databaseName]
);
- conn.destroy();
const tableList = [];
for (const row of rows) {
if (row["table_name"]) {
tableList.push(row["table_name"]);
}
}
- return tableList;
-};
-
-const getTableStructure = async (
- connection: Connection,
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- const conn = await getMySQLConnection(connection);
- const [rows] = await conn.query(
- `SHOW CREATE TABLE \`${databaseName}\`.\`${tableName}\`;`
- );
- conn.destroy();
- if (rows.length !== 1) {
- throw new Error("Unexpected number of rows.");
- }
- structureFetched(tableName, rows[0]["Create Table"] || "");
-};
+ const SchemaList: Schema[] = [{ name: "", tables: [] as Table[] }];
-const getTableStructureBatch = async (
- connection: Connection,
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- const conn = await getMySQLConnection(connection);
+ for (const tableName of tableList) {
+ const [rows] = await conn.query(`SHOW CREATE TABLE \`${databaseName}\`.\`${tableName}\`;`);
+ if (rows.length !== 1) {
+ throw new Error("Unexpected number of rows.");
+ }
- await Promise.all(
- tableNameList.map(async (tableName) => {
- const [rows] = await conn.query(
- `SHOW CREATE TABLE \`${databaseName}\`.\`${tableName}\`;`
- );
- if (rows.length !== 1) {
- throw new Error("Unexpected number of rows.");
- }
- structureFetched(tableName, rows[0]["Create Table"] || "");
- })
- ).finally(() => {
- conn.destroy();
- });
+ SchemaList[0].tables.push({ name: tableName, structure: rows[0]["Create Table"] || "" });
+ }
+ return SchemaList;
};
const newConnector = (connection: Connection): Connector => {
return {
testConnection: () => testConnection(connection),
- execute: (databaseName: string, statement: string) =>
- execute(connection, databaseName, statement),
+ execute: (databaseName: string, statement: string) => execute(connection, databaseName, statement),
getDatabases: () => getDatabases(connection),
- getTables: (databaseName: string) => getTables(connection, databaseName),
- getTableStructure: (
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructure(connection, databaseName, tableName, structureFetched),
- getTableStructureBatch: (
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructureBatch(
- connection,
- databaseName,
- tableNameList,
- structureFetched
- ),
+ getTableSchema: (databaseName: string) => getTableSchema(connection, databaseName),
};
};
diff --git a/src/lib/connectors/postgres/index.ts b/src/lib/connectors/postgres/index.ts
index 31ab8f4d..6be83bf1 100644
--- a/src/lib/connectors/postgres/index.ts
+++ b/src/lib/connectors/postgres/index.ts
@@ -1,14 +1,20 @@
import { Client, ClientConfig } from "pg";
-import { Connection, ExecutionResult } from "@/types";
+import { Connection, ExecutionResult, Table, Schema } from "@/types";
import { Connector } from "..";
-const newPostgresClient = (connection: Connection) => {
+const systemSchemas =
+ "'information_schema', 'pg_catalog', 'pg_toast', '_timescaledb_cache', '_timescaledb_catalog', '_timescaledb_internal', '_timescaledb_config', 'timescaledb_information', 'timescaledb_experimental'";
+
+const systemTables = "'_prisma_migrations'";
+
+const newPostgresClient = async (connection: Connection) => {
const clientConfig: ClientConfig = {
host: connection.host,
port: Number(connection.port),
user: connection.username,
password: connection.password,
database: connection.database,
+ application_name: "sqlchat",
};
if (connection.ssl) {
clientConfig.ssl = {
@@ -16,25 +22,43 @@ const newPostgresClient = (connection: Connection) => {
cert: connection.ssl?.cert,
key: connection.ssl?.key,
};
+ } else {
+ clientConfig.ssl = {
+ rejectUnauthorized: false,
+ };
+ }
+
+ let client = new Client(clientConfig);
+
+ if (connection.ssl) {
+ await client.connect();
+ } else {
+ try {
+ await client.connect();
+ } catch (error) {
+ // Because node-postgres didn't implement `sslmode: preferred`. So first try to connect via SSL, otherwise connect via non-SSL.
+ // Connecting postgres via non-ssl requires `clientConfig.ssl` is undefined. ref: https://github.com/sqlchat/sqlchat/issues/108
+ if (error instanceof Error && error.message.includes("The server does not support SSL connections")) {
+ clientConfig.ssl = undefined;
+ client = new Client(clientConfig);
+ await client.connect();
+ } else {
+ throw error;
+ }
+ }
}
- return new Client(clientConfig);
+ return client;
};
const testConnection = async (connection: Connection): Promise => {
- const client = newPostgresClient(connection);
- await client.connect();
+ const client = await newPostgresClient(connection);
await client.end();
return true;
};
-const execute = async (
- connection: Connection,
- databaseName: string,
- statement: string
-): Promise => {
+const execute = async (connection: Connection, databaseName: string, statement: string): Promise => {
connection.database = databaseName;
- const client = newPostgresClient(connection);
- await client.connect();
+ const client = await newPostgresClient(connection);
const { rows, rowCount } = await client.query(statement);
await client.end();
@@ -50,8 +74,7 @@ const execute = async (
};
const getDatabases = async (connection: Connection): Promise => {
- const client = newPostgresClient(connection);
- await client.connect();
+ const client = await newPostgresClient(connection);
if (connection.database) {
await client.end();
return [connection.database];
@@ -69,118 +92,58 @@ const getDatabases = async (connection: Connection): Promise => {
return databaseList;
};
-const getTables = async (
- connection: Connection,
- databaseName: string
-): Promise => {
+const getTableSchema = async (connection: Connection, databaseName: string): Promise => {
connection.database = databaseName;
- const client = newPostgresClient(connection);
- await client.connect();
+ const client = await newPostgresClient(connection);
const { rows } = await client.query(
- `SELECT table_name FROM information_schema.tables WHERE table_schema='public' AND table_type='BASE TABLE' AND table_catalog=$1;`,
+ `SELECT table_schema, table_name FROM information_schema.tables WHERE table_schema NOT IN (${systemSchemas}) AND table_name NOT IN (${systemTables}) AND (table_type='BASE TABLE' or table_type='VIEW') AND table_catalog=$1;`,
[databaseName]
);
- await client.end();
- const tableList = [];
+
+ const schemaList: Schema[] = [];
for (const row of rows) {
if (row["table_name"]) {
- tableList.push(row["table_name"]);
+ const schema = schemaList.find((schema) => schema.name === row["table_schema"]);
+ if (schema) {
+ schema.tables.push({ name: row["table_name"] as string, structure: "" } as Table);
+ } else {
+ schemaList.push({
+ name: row["table_schema"],
+ tables: [{ name: row["table_name"], structure: "" } as Table],
+ });
+ }
}
}
- return tableList;
-};
-const getTableStructure = async (
- connection: Connection,
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- connection.database = databaseName;
- const client = newPostgresClient(connection);
- await client.connect();
- const { rows } = await client.query(
- `SELECT column_name, data_type, is_nullable FROM information_schema.columns WHERE table_schema='public' AND table_name=$1;`,
- [tableName]
- );
- await client.end();
- const columnList = [];
- // TODO(steven): transform it to standard schema string.
- for (const row of rows) {
- columnList.push(
- `${row["column_name"]} ${row["data_type"].toUpperCase()} ${
- String(row["is_nullable"]).toUpperCase() === "NO" ? "NOT NULL" : ""
- }`
- );
- }
- structureFetched(
- tableName,
- `CREATE TABLE \`${tableName}\` (
- ${columnList.join(",\n")}
- );`
- );
-};
-
-const getTableStructureBatch = async (
- connection: Connection,
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
-): Promise => {
- connection.database = databaseName;
- const client = newPostgresClient(connection);
- await client.connect();
- await Promise.all(
- tableNameList.map(async (tableName) => {
- const { rows } = await client.query(
- `SELECT column_name, data_type, is_nullable FROM information_schema.columns WHERE table_schema='public' AND table_name=$1;`,
- [tableName]
+ for (const schema of schemaList) {
+ for (const table of schema.tables) {
+ const { rows: result } = await client.query(
+ `SELECT column_name, data_type, is_nullable FROM information_schema.columns WHERE table_schema NOT IN (${systemSchemas}) AND table_name=$1 AND table_schema=$2;`,
+ [table.name, schema.name]
);
const columnList = [];
// TODO(steven): transform it to standard schema string.
- for (const row of rows) {
+ for (const row of result) {
columnList.push(
- `${row["column_name"]} ${row["data_type"].toUpperCase()} ${
- String(row["is_nullable"]).toUpperCase() === "NO" ? "NOT NULL" : ""
- }`
+ `${row["column_name"]} ${row["data_type"].toUpperCase()} ${String(row["is_nullable"]).toUpperCase() === "NO" ? "NOT NULL" : ""}`
);
}
- structureFetched(
- tableName,
- `CREATE TABLE \`${tableName}\` (
- ${columnList.join(",\n")}
- );`
- );
- })
- ).finally(async () => {
- await client.end();
- });
+
+ let fullTableName = schema.name == "public" ? `"${table.name}"` : `"${schema.name}"."${table.name}"`;
+ table.structure = `CREATE TABLE ${fullTableName} (\n${columnList.join(",\n")}\n);`;
+ }
+ }
+
+ await client.end();
+ return schemaList;
};
const newConnector = (connection: Connection): Connector => {
return {
testConnection: () => testConnection(connection),
- execute: (databaseName: string, statement: string) =>
- execute(connection, databaseName, statement),
+ execute: (databaseName: string, statement: string) => execute(connection, databaseName, statement),
getDatabases: () => getDatabases(connection),
- getTables: (databaseName: string) => getTables(connection, databaseName),
- getTableStructure: (
- databaseName: string,
- tableName: string,
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructure(connection, databaseName, tableName, structureFetched),
- getTableStructureBatch: (
- databaseName: string,
- tableNameList: string[],
- structureFetched: (tableName: string, structure: string) => void
- ) =>
- getTableStructureBatch(
- connection,
- databaseName,
- tableNameList,
- structureFetched
- ),
+ getTableSchema: (databaseName: string) => getTableSchema(connection, databaseName),
};
};
diff --git a/src/locales/de.json b/src/locales/de.json
deleted file mode 100644
index 1387649e..00000000
--- a/src/locales/de.json
+++ /dev/null
@@ -1,97 +0,0 @@
-{
- "common": {
- "clear": "Löschen",
- "close": "Schließen",
- "confirm": "Bestätigen",
- "save": "Speichern",
- "edit": "Bearbeiten",
- "loading": "Laden",
- "setting": "Einstellung",
- "copy": "Kopieren",
- "delete": "Löschen",
- "execute": "Ausführen",
- "sign-in": "Anmelden",
- "sign-out": "Austragen",
- "back": "Zurückkehren",
- "quota": "Quote"
- },
- "conversation": {
- "new-chat": "Neuer Chat",
- "conversation-title": "Titel des Gesprächs",
- "update": "Gespräch aktualisieren",
- "title": "Titel"
- },
- "connection": {
- "self": "Verbindung",
- "new": "Verbindung erstellen",
- "edit": "Verbindung bearbeiten",
- "select-database": "Wählen Sie Ihre Datenbank aus",
- "database-type": "Datenbanktyp",
- "title": "Titel",
- "host": "Host",
- "port": "Port",
- "database-name": "Datenbankname",
- "username": "Benutzername",
- "password": "Passwort"
- },
- "assistant": {
- "self": "Bot",
- "create-your-bot": "Sie finden den benötigten Bot nicht? Erstellen Sie einen"
- },
- "execution": {
- "title": "SQL ausführen",
- "message": {
- "executing": "Abfrage wird ausgeführt...",
- "no-connection": "Keine Verbindung ausgewählt",
- "no-data": "Keine Daten zurückgegeben"
- }
- },
- "editor": {
- "placeholder": "Geben Sie hier Ihre Frage ein..."
- },
- "setting": {
- "self": "Einstellung",
- "general": "Allgemein",
- "plan": {
- "free": "Frei",
- "pro": "Profi",
- "signup-for-more": "Melden Sie sich für mehr an",
- "upgrade": "Aktualisierung"
- },
- "billing": {
- "self": "Abrechnung"
- },
- "account": {
- "self": "Konto"
- },
- "basic": {
- "self": "Grundlegend",
- "language": "Sprache"
- },
- "theme": {
- "self": "Thema",
- "system": "System folgen",
- "light": "Hell",
- "dark": "Dunkel"
- },
- "openai-api-configuration": {
- "self": "OpenAI API-Konfiguration"
- },
- "data": {
- "self": "Daten",
- "clear-all-data": "Alle Daten löschen"
- }
- },
- "social": {
- "join-discord-channel": "Discord-Kanal beitreten",
- "join-wechat-group": "WeChat-Gruppe beitreten"
- },
- "banner": {
- "data-storage": "Verbindungseinstellungen werden in Ihrem lokalen Browser gespeichert",
- "non-select-sql-warning": "Die aktuelle Anweisung kann eine nicht-SELECT SQL-Anweisung sein, die zu einer Änderung des Datenbankschemas oder der Daten führen wird. Stellen Sie sicher, dass Sie wissen, was Sie tun.",
- "product-hunt": "🚀🚀🚀 Wir haben gerade auf Product Hunt gestartet, bitte geben Sie uns eine Stimme! 🚀🚀🚀"
- },
- "payment": {
- "self": "Zahlung"
- }
-}
diff --git a/src/locales/en.json b/src/locales/en.json
index b51aa073..10530504 100644
--- a/src/locales/en.json
+++ b/src/locales/en.json
@@ -9,11 +9,14 @@
"setting": "Setting",
"copy": "Copy",
"delete": "Delete",
- "execute": "Execute",
+ "run-sql": "Run SQL",
"sign-in": "Sign in",
"sign-out": "Sign out",
"back": "Back",
- "quota": "Quota"
+ "quota": "Monthly Quota",
+ "date": "Date",
+ "description": "Description",
+ "amount": "Amount"
},
"conversation": {
"new-chat": "New Chat",
@@ -25,16 +28,26 @@
"self": "Connection",
"new": "Create Connection",
"edit": "Edit Connection",
+ "refresh-schema": "Refresh Schema",
"select-database": "Select your database",
"select-table": "Select your table",
- "all-tables":"All Tables",
+ "select-schema": "Select your Schema",
+ "all-tables": "All Tables",
+ "total-token": "Tokens",
"database-type": "Database type",
"title": "Title",
"host": "Host",
"port": "Port",
"database-name": "Database name",
"username": "Username",
- "password": "Password"
+ "password": "Password",
+ "empty-select": "clear selection",
+ "select-all": "select all",
+ "tidb-serverless-ssl-hint": "SSL is enabled by default"
+ },
+ "prompt": {
+ "self": "Prompt",
+ "current-conversation": "Current conversation prompt"
},
"assistant": {
"self": "Bot",
@@ -55,15 +68,23 @@
"self": "Setting",
"general": "General",
"plan": {
+ "guest": "Guest",
"free": "Free",
"pro": "Pro",
+ "pro-early-bird": "🐤 Early Bird - Pro 50% OFF",
"signup-for-more": "Sign up for more",
"upgrade": "Upgrade",
- "pro-question-per-month": "1000 questions / month",
- "early-bird-checkout": "Early bird discount, 50% off for 1 year"
+ "renew": "Renew",
+ "expired": "Expired",
+ "n-question-per-month": "{{count}} quota / month",
+ "1-month": "1 month",
+ "n-months": "{{count}} months",
+ "purhcase-1-month": "Purchase 1 month",
+ "purhcase-n-months": "Purchase {{count}} months"
},
- "billing": {
- "self": "Billing"
+ "subscription": {
+ "self": "Subscription",
+ "view-receipt": "View receipt"
},
"account": {
"self": "Account"
@@ -79,7 +100,13 @@
"dark": "Dark"
},
"openai-api-configuration": {
- "self": "OpenAI API configuration"
+ "self": "OpenAI API configuration",
+ "model": "Model",
+ "model-description": "Quota won't be consumed if you provide your own key below.",
+ "quota-per-ask": "{{count}} quota per ask",
+ "key-description": "Bring your own key to waive quota requirement.",
+ "find-my-key": "Find my key",
+ "endpoint-description": "Optional endpoint pointing to your own compatible server or gateway."
},
"data": {
"self": "Data",
@@ -99,6 +126,6 @@
},
"payment": {
"self": "Payment",
- "login-to-buy": "Login to buy"
+ "sign-in-to-buy": "Sign in to buy"
}
}
diff --git a/src/locales/es.json b/src/locales/es.json
index f59fe1b9..baead01d 100644
--- a/src/locales/es.json
+++ b/src/locales/es.json
@@ -9,15 +9,18 @@
"setting": "Configuración",
"copy": "Copiar",
"delete": "Borrar",
+ "run-sql": "Ejecutar SQL",
"sign-in": "Iniciar sesión",
"sign-out": "Desconectar",
- "back": "Devolver",
- "quota": "Cuota",
- "execute": "Ejecutar"
+ "back": "Volver",
+ "quota": "Cuota Mensual",
+ "date": "Fecha",
+ "description": "Descripción",
+ "amount": "Cantidad"
},
"conversation": {
"new-chat": "Nuevo Chat",
- "conversation-title": "Titulo de conversación",
+ "conversation-title": "Título de la conversación",
"update": "Actualizar conversación",
"title": "Titulo"
},
@@ -25,6 +28,7 @@
"self": "Conexión",
"new": "Crear Conexión",
"edit": "Editar Conexión",
+ "refresh-schema": "Actualizar Esquema",
"select-database": "Selecciona tu base de datos",
"database-type": "Tipo de Base de Datos",
"title": "Nombre",
@@ -32,7 +36,18 @@
"port": "Puerto",
"database-name": "Nombre de Base de Datos",
"username": "Usuario",
- "password": "Contraseña"
+ "password": "Contraseña",
+ "empty-select": "limpiar selección",
+ "select-all": "seleccionar todo",
+ "total-token": "Tokens",
+ "tidb-serverless-ssl-hint": "SSL está habilitado de forma predeterminada",
+ "select-table": "Selecciona tu tabla",
+ "select-schema": "Selecciona tu Esquema",
+ "all-tables": "Todas las Tablas"
+ },
+ "prompt": {
+ "view": "Prompt",
+ "current-conversation": "Mensaje de conversación actual"
},
"assistant": {
"self": "Bot",
@@ -53,13 +68,23 @@
"self": "Configuración",
"general": "General",
"plan": {
+ "guest": "Invitado",
"free": "Gratis",
"pro": "Pro",
+ "pro-early-bird": "🐤 Early Bird - Pro 50% de descuento",
"signup-for-more": "Regístrese para obtener más",
- "upgrade": "Mejora"
+ "upgrade": "Mejora",
+ "renew": "Renovar",
+ "expired": "Expirado",
+ "n-question-per-month": "{{count}} Cuota / mes",
+ "1-month": "1 mes",
+ "n-months": "{{count}} meses",
+ "purhcase-1-month": "Comprar 1 mes",
+ "purhcase-n-months": "Comprar {{count}} meses"
},
- "billing": {
- "self": "Facturación"
+ "subscription": {
+ "self": "Suscripción",
+ "view-receipt": "Ver recibo"
},
"account": {
"self": "Cuenta"
@@ -75,7 +100,12 @@
"dark": "Oscuro"
},
"openai-api-configuration": {
- "self": "Configuración del API de OpenAI"
+ "self": "Configuración del API de OpenAI",
+ "model": "Modelo",
+ "model-description": "La cuota no se consumirá si proporciona su propia clave a continuación.",
+ "key-description": "Indique su propia clave para retirarse del límite por cuota.",
+ "find-my-key": "Encuentra mi llave",
+ "endpoint-description": "Endpoint opcional que apunta a su propio servidor o gateway compatible."
},
"data": {
"self": "Datos",
@@ -87,13 +117,14 @@
"join-wechat-group": "Unirse al grupo de WeChat"
},
"banner": {
- "data-storage": "Las configuraciones de conexión y las consultas se almacenan en tu navegador.",
+ "data-storage": "Las configuraciones de conexión y las consultas se almacenan en tu navegador",
"non-select-sql-warning": "La declaración actual puede no ser SELECT SQL, lo que dará como resultado un esquema de base de datos o un cambio de datos. Asegúrate de saber lo que estás haciendo.",
- "product-hunt": "🚀🚀🚀 Acabamos de lanzar en Product Hunt, ¡por favor, vota por nosotros! 🚀🚀🚀"
+ "product-hunt": "🚀🚀🚀 Acabamos de lanzar en Product Hunt, ¡por favor, vota por nosotros! 🚀🚀🚀",
+ "quota-overflow": "Cuota excedida. Proporcione su propia clave API de OpenAI.",
+ "use-my-key": "Usar mi propia clave"
},
"payment": {
"self": "Pago",
- "quota-overflow": "Cuota excedida. Proporcione su propia clave API de OpenAI.",
- "use-my-key": "Usar mi propia clave"
+ "sign-in-to-buy": "Inicia sesión para comprar"
}
}
diff --git a/src/locales/i18n.ts b/src/locales/i18n.ts
index 5bc9febb..3c4fd4bf 100644
--- a/src/locales/i18n.ts
+++ b/src/locales/i18n.ts
@@ -2,11 +2,7 @@ import i18n from "i18next";
import { initReactI18next } from "react-i18next";
import enLocale from "./en.json";
import zhLocale from "./zh.json";
-import zhHant from "./zh-Hant.json";
import esLocale from "./es.json";
-import jpLocale from "./jp.json";
-import deLocale from "./de.json";
-import thLocale from "./th.json";
i18n.use(initReactI18next).init({
resources: {
@@ -16,21 +12,9 @@ i18n.use(initReactI18next).init({
zh: {
translation: zhLocale,
},
- zhHant: {
- translation: zhHant,
- },
es: {
translation: esLocale,
},
- jp: {
- translation: jpLocale,
- },
- de: {
- translation: deLocale,
- },
- th: {
- translation: thLocale,
- },
},
fallbackLng: "en",
});
diff --git a/src/locales/jp.json b/src/locales/jp.json
deleted file mode 100644
index 2ec08d83..00000000
--- a/src/locales/jp.json
+++ /dev/null
@@ -1,97 +0,0 @@
-{
- "common": {
- "clear": "クリア",
- "close": "閉じる",
- "confirm": "確認",
- "save": "保存",
- "edit": "編集",
- "loading": "読み込み中",
- "setting": "設定",
- "copy": "コピー",
- "delete": "削除",
- "execute": "実行",
- "sign-in": "ログイン",
- "sign-out": "サインアウト",
- "back": "戻る",
- "quota": "クォータ"
- },
- "conversation": {
- "new-chat": "新しいチャット",
- "conversation-title": "会話タイトル",
- "update": "会話を更新",
- "title": "タイトル"
- },
- "connection": {
- "self": "接続",
- "new": "接続を作成",
- "edit": "接続を編集",
- "select-database": "データベースを選択",
- "database-type": "データベースタイプ",
- "title": "タイトル",
- "host": "ホスト",
- "port": "ポート",
- "database-name": "データベース名",
- "username": "ユーザー名",
- "password": "パスワード"
- },
- "assistant": {
- "self": "ボット",
- "create-your-bot": "必要なボットが見つからない場合は、作成してください"
- },
- "execution": {
- "title": "SQLを実行",
- "message": {
- "executing": "クエリを実行中...",
- "no-connection": "接続が選択されていません",
- "no-data": "データが返されませんでした"
- }
- },
- "editor": {
- "placeholder": "ここに質問を入力してください..."
- },
- "setting": {
- "self": "設定",
- "general": "全般的",
- "plan": {
- "free": "無料版",
- "pro": "プロ",
- "signup-for-more": "サインアップしてもっと",
- "upgrade": "アップグレード"
- },
- "billing": {
- "self": "請求する"
- },
- "account": {
- "self": "アカウント"
- },
- "basic": {
- "self": "基本",
- "language": "言語"
- },
- "theme": {
- "self": "テーマ",
- "system": "システムに従う",
- "light": "ライト",
- "dark": "ダーク"
- },
- "openai-api-configuration": {
- "self": "OpenAI APIの設定"
- },
- "data": {
- "self": "データ",
- "clear-all-data": "すべてのデータをクリア"
- }
- },
- "social": {
- "join-discord-channel": "Discordチャンネルに参加",
- "join-wechat-group": "WeChatグループに参加"
- },
- "banner": {
- "data-storage": "接続設定はローカルブラウザに保存されます",
- "non-select-sql-warning": "現在のステートメントは非SELECT SQLである可能性があり、データベースのスキーマまたはデータの変更につながる可能性があります。自分が何をしているか確認してください。",
- "product-hunt": "🚀🚀🚀 Product Huntでローンチしました。投票してください！ 🚀🚀🚀"
- },
- "payment": {
- "self": "支払い"
- }
-}
diff --git a/src/locales/th.json b/src/locales/th.json
deleted file mode 100644
index 9dd9abab..00000000
--- a/src/locales/th.json
+++ /dev/null
@@ -1,97 +0,0 @@
-{
- "common": {
- "clear": "ล้าง",
- "close": "ปิด",
- "confirm": "ยืนยัน",
- "save": "บันทึก",
- "edit": "แก้ไข",
- "loading": "กำลังโหลด",
- "setting": "ตั้งค่า",
- "copy": "คัดลอก",
- "delete": "ลบ",
- "execute": "ดำเนินการ",
- "sign-in": "เข้าสู่ระบบ",
- "sign-out": "ออกจากระบบ",
- "back": "กลับ",
- "quota": "โควต้า"
- },
- "conversation": {
- "new-chat": "แชทใหม่",
- "conversation-title": "หัวข้อการสนทนา",
- "update": "อัปเดตการสนทนา",
- "title": "หัวข้อ"
- },
- "connection": {
- "self": "การเชื่อมต่อ",
- "new": "สร้างการเชื่อมต่อ",
- "edit": "แก้ไขการเชื่อมต่อ",
- "select-database": "เลือกฐานข้อมูลของคุณ",
- "database-type": "ประเภทฐานข้อมูล",
- "title": "หัวข้อ",
- "host": "โฮสต์",
- "port": "พอร์ต",
- "database-name": "ชื่อฐานข้อมูล",
- "username": "ชื่อผู้ใช้",
- "password": "รหัสผ่าน"
- },
- "assistant": {
- "self": "บอท",
- "create-your-bot": "ไม่พบบอทที่คุณต้องการ? สร้างใหม่"
- },
- "execution": {
- "title": "ดำเนินการ SQL",
- "message": {
- "executing": "กำลังดำเนินการคำสั่ง...",
- "no-connection": "ไม่ได้เลือกการเชื่อมต่อ",
- "no-data": "ไม่มีข้อมูลที่ส่งกลับ"
- }
- },
- "editor": {
- "placeholder": "กรอกคำถามของคุณที่นี่..."
- },
- "setting": {
- "self": "ตั้งค่า",
- "general": "ทั่วไป",
- "plan": {
- "free": "รุ่นฟรี",
- "pro": "รุ่นมืออาชีพ",
- "signup-for-more": "ลงทะเบียนเพื่อรับข้อมูลเพิ่มเติม",
- "upgrade": "อัปเกรด"
- },
- "billing": {
- "self": "การเรียกเก็บเงิน"
- },
- "account": {
- "self": "บัญชี"
- },
- "basic": {
- "self": "พื้นฐาน",
- "language": "ภาษา"
- },
- "theme": {
- "self": "ธีม",
- "system": "ตามระบบ",
- "light": "สว่าง",
- "dark": "เข้ม"
- },
- "openai-api-configuration": {
- "self": "การกำหนดค่า OpenAI API"
- },
- "data": {
- "self": "ข้อมูล",
- "clear-all-data": "ลบข้อมูลทั้งหมด"
- }
- },
- "social": {
- "join-discord-channel": "เข้าร่วมช่อง Discord",
- "join-wechat-group": "เข้าร่วมกลุ่ม WeChat"
- },
- "banner": {
- "data-storage": "การตั้งค่าการเชื่อมต่อจะถูกเก็บไว้ในเบราว์เซอร์ของคุณ",
- "non-select-sql-warning": "คำสั่งปัจจุบันอาจเป็นคำสั่ง SQL ที่ไม่ใช่ SELECT ซึ่งจะทำให้เกิดการเปลี่ยนแปลงโครงสร้างฐานข้อมูลหรือข้อมูล โปรดตรวจสอบว่าคุณรู้ว่าตัวเองกำลังทำอะไร",
- "product-hunt": "🚀🚀🚀 เราเพิ่งเปิดตัวบน Product Hunt โปรดโหวตเรา! 🚀🚀🚀"
- },
- "payment": {
- "self": "การชำระเงิน"
- }
-}
diff --git a/src/locales/zh-Hant.json b/src/locales/zh-Hant.json
deleted file mode 100644
index 925d35fc..00000000
--- a/src/locales/zh-Hant.json
+++ /dev/null
@@ -1,104 +0,0 @@
-{
- "common": {
- "clear": "清除",
- "close": "關閉",
- "confirm": "確認",
- "save": "儲存",
- "edit": "編輯",
- "loading": "載入中",
- "setting": "設定",
- "copy": "複製",
- "delete": "刪除",
- "execute": "執行",
- "sign-in": "登錄",
- "sign-out": "登出",
- "back": "返回",
- "quota": "額度"
- },
- "conversation": {
- "new-chat": "新聊天",
- "conversation-title": "對話標題",
- "update": "更新對話",
- "title": "標題"
- },
- "connection": {
- "self": "連線",
- "new": "建立連線",
- "edit": "編輯連線",
- "select-database": "選擇您的資料庫",
- "select-table": "選擇您的資料",
- "all-tables":"全部表",
- "database-type": "資料庫類型",
- "title": "標題",
- "host": "主機",
- "port": "埠號",
- "database-name": "資料庫名稱",
- "username": "使用者名稱",
- "password": "密碼"
- },
- "assistant": {
- "self": "機器人",
- "create-your-bot": "找不到您需要的機器人？建立一個"
- },
- "execution": {
- "title": "執行 SQL",
- "message": {
- "executing": "正在執行查詢...",
- "no-connection": "未選擇連線",
- "no-data": "未返回任何資料"
- }
- },
- "editor": {
- "placeholder": "在此輸入您的問題..."
- },
- "setting": {
- "self": "設定",
- "general": "通用",
- "plan": {
- "free": "免費版",
- "pro": "專業版",
- "signup-for-more": "注册提升额度",
- "upgrade": "升級",
- "pro-question-per-month": "1000 次提問 / 月",
- "early-bird-checkout": "早鳥優惠，5 折購買 1 年"
- },
- "billing": {
- "self": "计费"
- },
- "account": {
- "self": "賬戶"
- },
- "basic": {
- "self": "基本",
- "language": "語言"
- },
- "theme": {
- "self": "主題",
- "system": "跟隨系統",
- "light": "亮色",
- "dark": "暗色"
- },
- "openai-api-configuration": {
- "self": "OpenAI API 設定"
- },
- "data": {
- "self": "資料",
- "clear-all-data": "清除所有資料"
- }
- },
- "social": {
- "join-discord-channel": "加入 Discord 頻道",
- "join-wechat-group": "加入微信群組"
- },
- "banner": {
- "data-storage": "連線設定儲存在您的本地瀏覽器中",
- "non-select-sql-warning": "當前語句可能是非 SELECT SQL 語句，這將導致資料庫結構或資料的更改。請確保您知道自己在做什麼。",
- "product-hunt": "🚀🚀🚀 我們剛在 Product Hunt 上推出，請給我們一個投票！🚀🚀🚀",
- "quota-overflow": "我們的 OpenAI 額度已用完，請提供您自己的 OpenAI API key。",
- "use-my-key": "提供我自己的 key"
- },
- "payment": {
- "self": "支付",
- "login-to-buy": "登陸後購買"
- }
-}
diff --git a/src/locales/zh.json b/src/locales/zh.json
index 85220764..4a1c844e 100644
--- a/src/locales/zh.json
+++ b/src/locales/zh.json
@@ -9,11 +9,14 @@
"setting": "设置",
"copy": "复制",
"delete": "删除",
- "execute": "执行",
+ "run-sql": "运行 SQL",
"sign-in": "登录",
"sign-out": "登出",
"back": "返回",
- "quota": "额度"
+ "quota": "每月额度",
+ "date": "日期",
+ "description": "描述",
+ "amount": "金额"
},
"conversation": {
"new-chat": "新建会话",
@@ -25,16 +28,26 @@
"self": "连接",
"new": "创建连接",
"edit": "编辑连接",
+ "refresh-schema": "刷新 Schema",
"select-database": "选择数据库",
"select-table": "选择数据表",
- "all-tables":"全部表",
+ "select-schema": "选择 Schema",
+ "all-tables": "全部表",
+ "select-all": "选择全部",
+ "total-token": "Token",
+ "empty-select": "清空选择",
"database-type": "数据库类型",
"title": "标题",
"host": "主机",
"port": "端口",
"database-name": "数据库",
"username": "用户名",
- "password": "密码"
+ "password": "密码",
+ "tidb-serverless-ssl-hint": "SSL 默认已开启"
+ },
+ "prompt": {
+ "view": "提示词",
+ "current-conversation": "当前会话的提示词"
},
"assistant": {
"self": "机器人",
@@ -55,15 +68,23 @@
"self": "设置",
"general": "通用",
"plan": {
+ "guest": "访客",
"free": "免费版",
"pro": "专业版",
- "signup-for-more": "注册提升额度",
+ "pro-early-bird": "🐤 早鸟优惠 - 专业版 5 折",
+ "signup-for-more": "注册获得额度",
"upgrade": "升级",
- "pro-question-per-month": "1000 次提问 / 月",
- "early-bird-checkout": "早鸟优惠，5 折购买 1 年"
+ "renew": "续费",
+ "expired": "已过期",
+ "n-question-per-month": "{{count}} 点额度 / 月",
+ "1-month": "1 个月",
+ "n-months": "{{count}} 个月",
+ "purhcase-1-month": "购买 1 个月",
+ "purhcase-n-months": "购买 {{count}} 个月"
},
- "billing": {
- "self": "计费"
+ "subscription": {
+ "self": "订阅",
+ "view-receipt": "查看收据"
},
"account": {
"self": "账户"
@@ -79,7 +100,13 @@
"dark": "深色"
},
"openai-api-configuration": {
- "self": "OpenAI API 配置"
+ "self": "OpenAI API 配置",
+ "model": "模型",
+ "model-description": "如果您提供自己的 key，额度是不会消耗的。",
+ "quota-per-ask": "每一个提问消耗 {{count}} 点额度",
+ "find-my-key": "找到我的 Key",
+ "key-description": "一旦您提供了自己的 key，额度就不受限制了。",
+ "endpoint-description": "可选的 endpoint 指向接口兼容的服务器或者网关。"
},
"data": {
"self": "数据",
@@ -99,6 +126,6 @@
},
"payment": {
"self": "支付",
- "login-to-buy": "登陆后购买"
+ "sign-in-to-buy": "登录后购买"
}
}
diff --git a/src/pages/_app.tsx b/src/pages/_app.tsx
index 9f3f58b1..3dfb690f 100644
--- a/src/pages/_app.tsx
+++ b/src/pages/_app.tsx
@@ -7,6 +7,7 @@ import React, { useEffect } from "react";
import { Toaster } from "react-hot-toast";
import { useTranslation } from "react-i18next";
import { useSettingStore } from "@/store";
+import { hasDatabase } from "@/utils";
dayjs.extend(localizedFormat);
import "dayjs/locale/zh";
@@ -19,13 +20,14 @@ import "@/styles/mui.css";
import type { Session } from "next-auth";
-function MyApp({
- Component,
- pageProps: { session, ...pageProps },
-}: AppProps<{ session: Session }>) {
+function MyApp({ Component, pageProps: { session, ...pageProps } }: AppProps<{ session: Session }>) {
const { i18n } = useTranslation();
const settingStore = useSettingStore();
+ // Check whether those NEXT_PUBLIC_ envs are properly exposed in frontend.
+ // See https://github.com/vercel/next.js/discussions/17641
+ console.log("Has database:", hasDatabase());
+
useEffect(() => {
const darkMediaQuery = window.matchMedia("(prefers-color-scheme: dark)");
const handleColorSchemeChange = (e: MediaQueryListEvent) => {
@@ -58,10 +60,7 @@ function MyApp({
const theme = settingStore.setting.theme;
let currentAppearance = theme;
if (theme === "system") {
- if (
- window.matchMedia &&
- window.matchMedia("(prefers-color-scheme: dark)").matches
- ) {
+ if (window.matchMedia && window.matchMedia("(prefers-color-scheme: dark)").matches) {
currentAppearance = "dark";
} else {
currentAppearance = "light";
diff --git a/src/pages/api/auth/[...nextauth].ts b/src/pages/api/auth/[...nextauth].ts
index aecf440f..c41e4d23 100644
--- a/src/pages/api/auth/[...nextauth].ts
+++ b/src/pages/api/auth/[...nextauth].ts
@@ -4,29 +4,52 @@ import GoogleProvider from "next-auth/providers/google";
import { PrismaAdapter } from "@next-auth/prisma-adapter";
import EmailProvider from "next-auth/providers/email";
import { PrismaClient } from "@prisma/client";
+import { getSubscriptionByEmail } from "../utils/subscription";
+import { hasFeature } from "@/utils";
const prisma = new PrismaClient();
// For more information on each option (and a full list of options) go to
// https://next-auth.js.org/configuration/options
export const authOptions: NextAuthOptions = {
- adapter: PrismaAdapter(prisma),
+ adapter: hasFeature("account") ? PrismaAdapter(prisma) : undefined,
// https://next-auth.js.org/configuration/providers/oauth
- providers: [
- EmailProvider({
- server: process.env.EMAIL_SERVER,
- from: "noreply@sqlchat.ai",
- // maxAge: 24 * 60 * 60, // How long email links are valid for (default 24h)
- }),
- GoogleProvider({
- clientId: process.env.GOOGLE_CLIENT_ID,
- clientSecret: process.env.GOOGLE_CLIENT_SECRET,
- }),
- GithubProvider({
- clientId: process.env.GITHUB_ID,
- clientSecret: process.env.GITHUB_SECRET,
- }),
- ],
+ providers: hasFeature("account")
+ ? [
+ EmailProvider({
+ server: process.env.EMAIL_SERVER,
+ from: process.env.EMAIL_FROM || "noreply@sqlchat.ai",
+ // maxAge: 24 * 60 * 60, // How long email links are valid for (default 24h)
+ }),
+ GoogleProvider({
+ clientId: process.env.GOOGLE_CLIENT_ID,
+ clientSecret: process.env.GOOGLE_CLIENT_SECRET,
+ }),
+ GithubProvider({
+ clientId: process.env.GITHUB_ID,
+ clientSecret: process.env.GITHUB_SECRET,
+ }),
+ ]
+ : [],
+ cookies: {
+ sessionToken: {
+ name: `next-auth.session-token`,
+ options: {
+ httpOnly: true,
+ sameSite: "lax",
+ path: "/",
+ secure: true,
+ },
+ },
+ },
+ callbacks: {
+ async session({ session, user }) {
+ session.user.id = user.id;
+ session.user.subscription = await getSubscriptionByEmail(user.email);
+ session.user.stripeId = user.stripeId;
+ return session;
+ },
+ },
theme: {
brandColor: "#4F46E5",
logo: "/chat-logo.webp",
diff --git a/src/pages/api/auth/end-user.ts b/src/pages/api/auth/end-user.ts
index 4262b310..3b8ece94 100644
--- a/src/pages/api/auth/end-user.ts
+++ b/src/pages/api/auth/end-user.ts
@@ -1,16 +1,35 @@
+import { PrismaClient } from "@prisma/client";
import { NextApiRequest, NextApiResponse } from "next";
import { getServerSession } from "next-auth/next";
-import { authOptions } from "./[...nextauth]";
import requestIp from "request-ip";
+import { authOptions } from "./[...nextauth]";
+
+const prisma = new PrismaClient();
// Returns the login user email or the client IP address
-export const getEndUser = async (
- req: NextApiRequest,
- res: NextApiResponse
-): Promise => {
- const session = await getServerSession(req, res, authOptions);
- if (session?.user?.email) {
- return session.user.email;
+export const getEndUser = async (req: NextApiRequest, res: NextApiResponse): Promise => {
+ // Get from server session if available
+ const serverSession = await getServerSession(req, res, authOptions);
+ if (serverSession?.user?.email) {
+ return serverSession.user.email;
}
+
+ // Get from session token if available
+ const token = req.headers.authorization?.substring(7);
+ if (token) {
+ const sessionInDb = await prisma.session.findUnique({
+ where: { sessionToken: token },
+ });
+ if (sessionInDb?.userId) {
+ const user = await prisma.user.findUnique({
+ where: { id: sessionInDb.userId },
+ });
+ if (user?.email) {
+ return user.email;
+ }
+ }
+ }
+
+ // Get from client IP address
return requestIp.getClientIp(req) || "";
};
diff --git a/src/pages/api/auth/quota.ts b/src/pages/api/auth/quota.ts
deleted file mode 100644
index 2e856eb8..00000000
--- a/src/pages/api/auth/quota.ts
+++ /dev/null
@@ -1,33 +0,0 @@
-import { PrismaClient } from "@prisma/client";
-import { NextApiRequest, NextApiResponse } from "next";
-import { getServerSession } from "next-auth/next";
-import { authOptions } from "./[...nextauth]";
-import { getEndUser } from "./end-user";
-import { Quota, GUEST_QUOTA, FREE_QUOTA } from "@/types";
-
-const prisma = new PrismaClient();
-
-export const checkQuota = async (
- req: NextApiRequest,
- res: NextApiResponse
-): Promise => {
- const quota = await getQuota(req, res);
- return quota.current <= quota.limit;
-};
-
-export const getQuota = async (
- req: NextApiRequest,
- res: NextApiResponse
-): Promise => {
- const endUser = await getEndUser(req, res);
- const session = await getServerSession(req, res, authOptions);
- return {
- current: await prisma.message.count({
- where: {
- endUser: endUser,
- role: "user",
- },
- }),
- limit: session ? FREE_QUOTA : GUEST_QUOTA,
- };
-};
diff --git a/src/pages/api/chat.ts b/src/pages/api/chat.ts
index 233ca1f6..4919e952 100644
--- a/src/pages/api/chat.ts
+++ b/src/pages/api/chat.ts
@@ -1,12 +1,8 @@
-import {
- createParser,
- ParsedEvent,
- ReconnectInterval,
-} from "eventsource-parser";
+import { createParser, ParsedEvent, ReconnectInterval } from "eventsource-parser";
import { NextRequest } from "next/server";
-import { API_KEY } from "@/env";
-import { openAIApiEndpoint, openAIApiKey, gpt35 } from "@/utils";
+import { openAIApiEndpoint, openAIApiKey, openAIOrganization, hasFeature, getModel } from "@/utils";
+// Needs Edge for streaming response.
export const config = {
runtime: "edge",
};
@@ -18,51 +14,115 @@ const getApiEndpoint = (apiEndpoint: string) => {
};
const handler = async (req: NextRequest) => {
- if (API_KEY) {
- const auth = req.headers.get("Authorization");
- if (!auth || auth !== `Bearer ${API_KEY}`) {
- return new Response("Unauthorized", {
+ const reqBody = await req.json();
+ const apiKey = req.headers.get("x-openai-key") || openAIApiKey;
+
+ if (!apiKey) {
+ return new Response(
+ JSON.stringify({
+ error: {
+ message: "OpenAI API Key is missing. You can supply your own key via [Setting](/setting).",
+ },
+ }),
+ {
+ headers: {
+ "Content-Type": "application/json",
+ },
status: 401,
+ }
+ );
+ }
+
+ const useServerKey = !req.headers.get("x-openai-key");
+ const sessionToken = req.cookies.get("next-auth.session-token")?.value;
+ const currentUrl = new URL(req.url);
+ const usageUrl = new URL(currentUrl.protocol + "//" + currentUrl.host + "/api/usage");
+ const requestHeaders: any = {
+ Authorization: `Bearer ${sessionToken}`,
+ };
+ if (req.headers.get("x-openai-model")) {
+ // Pass model to usage API to calculate usage
+ requestHeaders["x-openai-model"] = req.headers.get("x-openai-model");
+ }
+
+ if (useServerKey) {
+ if (hasFeature("account") && !sessionToken) {
+ return new Response(
+ JSON.stringify({
+ error: {
+ message: "Please sign up to get free quota.",
+ },
+ }),
+ {
+ headers: {
+ "Content-Type": "application/json",
+ },
+ status: 401,
+ }
+ );
+ }
+
+ if (hasFeature("quota")) {
+ const usageRes = await fetch(usageUrl, {
+ method: "GET",
+ headers: requestHeaders,
});
+ if (!usageRes.ok) {
+ return new Response(usageRes.body, {
+ status: 500,
+ statusText: usageRes.statusText,
+ });
+ }
+
+ const usage = await usageRes.json();
+ if (usage.current >= usage.limit) {
+ return new Response(
+ JSON.stringify({
+ error: {
+ message: `You have reached your monthly quota: ${usage.current}/${usage.limit}.`,
+ },
+ }),
+ {
+ headers: {
+ "Content-Type": "application/json",
+ },
+ status: 401,
+ }
+ );
+ }
}
}
- const reqBody = await req.json();
- const openAIApiConfig = reqBody.openAIApiConfig;
- const apiKey = openAIApiConfig?.key || openAIApiKey;
+ let headers: { [key: string]: string } = {
+ "Content-Type": "application/json",
+ Authorization: `Bearer ${apiKey}`,
+ };
- if (!apiKey) {
- return new Response("Unauthorized", {
- status: 401,
- statusText:
- "OpenAI API Key is missing. You can supply your own key via Settings.",
- });
+ if (openAIOrganization) {
+ headers["OpenAI-Organization"] = openAIOrganization;
}
- const apiEndpoint = getApiEndpoint(
- openAIApiConfig?.endpoint || openAIApiEndpoint
- );
- const res = await fetch(apiEndpoint, {
- headers: {
- "Content-Type": "application/json",
- Authorization: `Bearer ${apiKey}`,
- },
+ const apiEndpoint = getApiEndpoint(req.headers.get("x-openai-endpoint") || openAIApiEndpoint);
+ const model = getModel(req.headers.get("x-openai-model") || "");
+ const remoteRes = await fetch(apiEndpoint, {
+ headers: headers,
method: "POST",
body: JSON.stringify({
- model: gpt35.name,
+ model: model.name,
messages: reqBody.messages,
- temperature: gpt35.temperature,
- frequency_penalty: gpt35.frequency_penalty,
- presence_penalty: gpt35.presence_penalty,
+ temperature: model.temperature,
+ frequency_penalty: model.frequency_penalty,
+ presence_penalty: model.presence_penalty,
stream: true,
- // Send end-user ID to help OpenAI monitor and detect abuse.
+ // Send end-user IP to help OpenAI monitor and detect abuse.
+ // It's intentionally not using email to avoid leaking user's email to OpenAI.
user: req.ip,
}),
});
- if (!res.ok) {
- return new Response(res.body, {
- status: res.status,
- statusText: res.statusText,
+ if (!remoteRes.ok) {
+ return new Response(remoteRes.body, {
+ status: remoteRes.status,
+ statusText: remoteRes.statusText,
});
}
@@ -88,11 +148,20 @@ const handler = async (req: NextRequest) => {
}
};
const parser = createParser(streamParser);
- for await (const chunk of res.body as any) {
+ for await (const chunk of remoteRes.body as any) {
parser.feed(decoder.decode(chunk));
}
},
});
+
+ if (useServerKey && hasFeature("quota")) {
+ // Increment usage count
+ await fetch(usageUrl, {
+ method: "POST",
+ headers: requestHeaders,
+ });
+ }
+
return new Response(stream);
};
diff --git a/src/pages/api/checkout_sessions/[id].ts b/src/pages/api/checkout_sessions/[id].ts
index 7dc1b320..5d2fc218 100644
--- a/src/pages/api/checkout_sessions/[id].ts
+++ b/src/pages/api/checkout_sessions/[id].ts
@@ -6,24 +6,19 @@ const stripe = new Stripe(process.env.STRIPE_API_KEY, {
apiVersion: "2022-11-15",
});
-export default async function handler(
- req: NextApiRequest,
- res: NextApiResponse
-) {
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
const id: string = req.query.id as string;
try {
if (!id.startsWith("cs_")) {
throw Error("Incorrect CheckoutSession ID.");
}
- const checkout_session: Stripe.Checkout.Session =
- await stripe.checkout.sessions.retrieve(id, {
- expand: ["payment_intent"],
- });
+ const checkout_session: Stripe.Checkout.Session = await stripe.checkout.sessions.retrieve(id, {
+ expand: ["payment_intent"],
+ });
res.status(200).json(checkout_session);
} catch (err) {
- const errorMessage =
- err instanceof Error ? err.message : "Internal server error";
+ const errorMessage = err instanceof Error ? err.message : "Internal server error";
res.status(500).json({ statusCode: 500, message: errorMessage });
}
}
diff --git a/src/pages/api/checkout_sessions/index.ts b/src/pages/api/checkout_sessions/index.ts
index 164d5c7a..6b0fa752 100644
--- a/src/pages/api/checkout_sessions/index.ts
+++ b/src/pages/api/checkout_sessions/index.ts
@@ -11,10 +11,7 @@ const stripe = new Stripe(process.env.STRIPE_API_KEY, {
const prisma = new PrismaClient();
-export default async function handler(
- req: NextApiRequest,
- res: NextApiResponse
-) {
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
const session = await getServerSession(req, res, authOptions);
if (!session?.user?.email) {
@@ -23,7 +20,6 @@ export default async function handler(
}
if (req.method === "POST") {
- const session = await getServerSession(req, res, authOptions);
const email = session?.user?.email!;
const user = await prisma.user.findUniqueOrThrow({
where: { email: email },
@@ -33,18 +29,10 @@ export default async function handler(
const params: Stripe.Checkout.SessionCreateParams = {
submit_type: "pay",
mode: "payment",
- payment_method_types: [
- "affirm",
- "alipay",
- "card",
- "cashapp",
- "klarna",
- "link",
- "wechat_pay",
- ],
+ payment_method_types: ["affirm", "alipay", "card", "cashapp", "klarna", "link", "wechat_pay"],
line_items: [
{
- price: process.env.STRIPE_ANNUAL_LICENSE_PRICE_ID,
+ price: req.body.price,
quantity: 1,
},
],
@@ -58,6 +46,8 @@ export default async function handler(
// setup_future_usage: "off_session",
metadata: {
email: session?.user?.email!,
+ plan: "PRO",
+ price: req.body.price,
},
},
// Link customer if present otherwise pass email and let Stripe create a new customer.
@@ -68,13 +58,11 @@ export default async function handler(
success_url: `${req.headers.origin}/setting?session_id={CHECKOUT_SESSION_ID}`,
cancel_url: `${req.headers.origin}/setting`,
};
- const checkoutSession: Stripe.Checkout.Session =
- await stripe.checkout.sessions.create(params);
+ const checkoutSession: Stripe.Checkout.Session = await stripe.checkout.sessions.create(params);
res.status(200).json(checkoutSession);
} catch (err) {
- const errorMessage =
- err instanceof Error ? err.message : "Internal server error";
+ const errorMessage = err instanceof Error ? err.message : "Internal server error";
res.status(500).json({ statusCode: 500, message: errorMessage });
}
} else {
diff --git a/src/pages/api/collect.ts b/src/pages/api/collect.ts
new file mode 100644
index 00000000..a1e68f5a
--- /dev/null
+++ b/src/pages/api/collect.ts
@@ -0,0 +1,59 @@
+import { PrismaClient } from "@prisma/client";
+import { NextApiRequest, NextApiResponse } from "next";
+import { Conversation, Message } from "@/types";
+import { getModel } from "@/utils";
+import { getEndUser } from "./auth/end-user";
+
+const prisma = new PrismaClient();
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+ if (req.method !== "POST") {
+ return res.status(405).json([]);
+ }
+ const conversation = req.body.conversation as Conversation;
+ const messages = req.body.messages as Message[];
+ try {
+ const chat = await prisma.chat.findUnique({
+ where: {
+ id: conversation.id,
+ },
+ });
+ const endUser = await getEndUser(req, res);
+ if (chat) {
+ await prisma.message.createMany({
+ data: messages.map((message) => ({
+ chatId: chat.id,
+ createdAt: new Date(message.createdAt),
+ endUser: endUser,
+ role: message.creatorRole,
+ content: message.content,
+ upvote: false,
+ downvote: false,
+ })),
+ });
+ } else {
+ await prisma.chat.create({
+ data: {
+ id: conversation.id,
+ createdAt: new Date(conversation.createdAt),
+ model: getModel((req.headers["x-openai-model"] as string) || ""),
+ ctx: {},
+ messages: {
+ create: messages.map((message) => ({
+ createdAt: new Date(message.createdAt),
+ endUser: endUser,
+ role: message.creatorRole,
+ content: message.content,
+ upvote: false,
+ downvote: false,
+ })),
+ },
+ },
+ });
+ }
+ } catch (err) {
+ console.error(err);
+ }
+
+ res.status(200).json(true);
+}
diff --git a/src/pages/api/connection/db.ts b/src/pages/api/connection/db.ts
index 84c5ee44..b4f57064 100644
--- a/src/pages/api/connection/db.ts
+++ b/src/pages/api/connection/db.ts
@@ -1,6 +1,8 @@
import { NextApiRequest, NextApiResponse } from "next";
import { newConnector } from "@/lib/connectors";
import { Connection } from "@/types";
+import { changeTiDBConnectionToMySQL } from "@/utils";
+import { Engine } from "@/types/connection";
// POST /api/connection/db
// req body: { connection: Connection }
@@ -10,7 +12,11 @@ const handler = async (req: NextApiRequest, res: NextApiResponse) => {
return res.status(405).json([]);
}
- const connection = req.body.connection as Connection;
+ let connection = req.body.connection as Connection;
+ if (connection.engineType === Engine.TiDB) {
+ connection = changeTiDBConnectionToMySQL(connection);
+ }
+
try {
const connector = newConnector(connection);
const databaseNameList = await connector.getDatabases();
diff --git a/src/pages/api/connection/db_schema.ts b/src/pages/api/connection/db_schema.ts
index 1c2388df..d5ff9863 100644
--- a/src/pages/api/connection/db_schema.ts
+++ b/src/pages/api/connection/db_schema.ts
@@ -1,6 +1,8 @@
import { NextApiRequest, NextApiResponse } from "next";
import { newConnector } from "@/lib/connectors";
-import { Connection, Table } from "@/types";
+import { Connection, Schema } from "@/types";
+import { changeTiDBConnectionToMySQL } from "@/utils";
+import { Engine } from "@/types/connection";
// POST /api/connection/db_schema
// req body: { connection: Connection, db: string }
@@ -10,26 +12,19 @@ const handler = async (req: NextApiRequest, res: NextApiResponse) => {
return res.status(405).json([]);
}
- const connection = req.body.connection as Connection;
+ let connection = req.body.connection as Connection;
+ if (connection.engineType === Engine.TiDB) {
+ connection = changeTiDBConnectionToMySQL(connection);
+ }
+
const db = req.body.db as string;
+
try {
const connector = newConnector(connection);
- const tableStructures: Table[] = [];
- const rawTableNameList = await connector.getTables(db);
- const structureFetched = (tableName: string, structure: string) => {
- tableStructures.push({
- name: tableName,
- structure,
- });
- };
- await connector.getTableStructureBatch(
- db,
- rawTableNameList,
- structureFetched
- );
+ const schemaList: Schema[] = await connector.getTableSchema(db);
res.status(200).json({
- data: tableStructures,
+ data: schemaList,
});
} catch (error: any) {
res.status(400).json({
diff --git a/src/pages/api/connection/execute.ts b/src/pages/api/connection/execute.ts
index 6ff613fe..900f1837 100644
--- a/src/pages/api/connection/execute.ts
+++ b/src/pages/api/connection/execute.ts
@@ -1,6 +1,8 @@
import { NextApiRequest, NextApiResponse } from "next";
import { newConnector } from "@/lib/connectors";
import { Connection } from "@/types";
+import { changeTiDBConnectionToMySQL } from "@/utils";
+import { Engine } from "@/types/connection";
// POST /api/connection/execute
// req body: { connection: Connection, db: string, statement: string }
@@ -9,7 +11,10 @@ const handler = async (req: NextApiRequest, res: NextApiResponse) => {
return res.status(405).json(false);
}
- const connection = req.body.connection as Connection;
+ let connection = req.body.connection as Connection;
+ if (connection.engineType === Engine.TiDB) {
+ connection = changeTiDBConnectionToMySQL(connection);
+ }
const db = req.body.db as string;
const statement = req.body.statement as string;
diff --git a/src/pages/api/connection/test.ts b/src/pages/api/connection/test.ts
index d9ec669c..5a54803f 100644
--- a/src/pages/api/connection/test.ts
+++ b/src/pages/api/connection/test.ts
@@ -1,6 +1,8 @@
import { NextApiRequest, NextApiResponse } from "next";
import { newConnector } from "@/lib/connectors";
import { Connection } from "@/types";
+import { changeTiDBConnectionToMySQL } from "@/utils";
+import { Engine } from "@/types/connection";
// POST /api/connection/test
// req body: { connection: Connection }
@@ -10,7 +12,10 @@ const handler = async (req: NextApiRequest, res: NextApiResponse) => {
return res.status(405).json(false);
}
- const connection = req.body.connection as Connection;
+ let connection = req.body.connection as Connection;
+ if (connection.engineType === Engine.TiDB) {
+ connection = changeTiDBConnectionToMySQL(connection);
+ }
try {
const connector = newConnector(connection);
await connector.testConnection();
diff --git a/src/pages/api/payment.ts b/src/pages/api/payment.ts
new file mode 100644
index 00000000..fcfe19af
--- /dev/null
+++ b/src/pages/api/payment.ts
@@ -0,0 +1,16 @@
+import { NextApiRequest, NextApiResponse } from "next";
+import { getEndUser } from "./auth/end-user";
+import { getPaymentListByEmail } from "./utils/payment";
+
+const handler = async (req: NextApiRequest, res: NextApiResponse) => {
+ if (req.method !== "GET") {
+ return res.status(405).json([]);
+ }
+
+ const endUser = await getEndUser(req, res);
+ const paymentList = await getPaymentListByEmail(endUser);
+
+ res.status(200).json(paymentList);
+};
+
+export default handler;
diff --git a/src/pages/api/quota.ts b/src/pages/api/quota.ts
deleted file mode 100644
index d26d5b81..00000000
--- a/src/pages/api/quota.ts
+++ /dev/null
@@ -1,11 +0,0 @@
-import { NextApiRequest, NextApiResponse } from "next";
-import { getQuota } from "./auth/quota";
-
-const handler = async (req: NextApiRequest, res: NextApiResponse) => {
- if (req.method !== "GET") {
- return res.status(405).json([]);
- }
- res.status(200).json(await getQuota(req, res));
-};
-
-export default handler;
diff --git a/src/pages/api/usage.ts b/src/pages/api/usage.ts
index c40bcbe5..87c6690f 100644
--- a/src/pages/api/usage.ts
+++ b/src/pages/api/usage.ts
@@ -1,62 +1,40 @@
-import { PrismaClient } from "@prisma/client";
import { NextApiRequest, NextApiResponse } from "next";
-import { Conversation, Message } from "@/types";
-import { gpt35 } from "@/utils";
+import { getServerSession } from "next-auth/next";
+import { authOptions } from "./auth/[...nextauth]";
+import { getSubscriptionByEmail } from "./utils/subscription";
+import { getModel } from "@/utils";
+import { addUsage, getCurrentMonthUsage } from "./utils/usage";
import { getEndUser } from "./auth/end-user";
+import { Quota } from "@/types";
-const prisma = new PrismaClient();
-
-export default async function handler(
- req: NextApiRequest,
- res: NextApiResponse
-) {
- if (req.method !== "POST") {
+const handler = async (req: NextApiRequest, res: NextApiResponse) => {
+ if (req.method !== "GET" && req.method !== "POST") {
return res.status(405).json([]);
}
- const conversation = req.body.conversation as Conversation;
- const messages = req.body.messages as Message[];
- try {
- const chat = await prisma.chat.findUnique({
- where: {
- id: conversation.id,
- },
- });
- const endUser = await getEndUser(req, res);
- if (chat) {
- await prisma.message.createMany({
- data: messages.map((message) => ({
- chatId: chat.id,
- createdAt: new Date(message.createdAt),
- endUser: endUser,
- role: message.creatorRole,
- content: message.content,
- upvote: false,
- downvote: false,
- })),
- });
- } else {
- await prisma.chat.create({
- data: {
- id: conversation.id,
- createdAt: new Date(conversation.createdAt),
- model: gpt35,
- ctx: {},
- messages: {
- create: messages.map((message) => ({
- createdAt: new Date(message.createdAt),
- endUser: endUser,
- role: message.creatorRole,
- content: message.content,
- upvote: false,
- downvote: false,
- })),
- },
- },
- });
- }
- } catch (err) {
- console.error(err);
+
+ const endUser = await getEndUser(req, res);
+
+ // Get from server session if available
+ const serverSession = await getServerSession(req, res, authOptions);
+ let subscripion = serverSession?.user?.subscription;
+ if (!subscripion) {
+ subscripion = await getSubscriptionByEmail(endUser);
}
- res.status(200).json(true);
-}
+ let usage = 0;
+ if (req.method === "GET") {
+ usage = await getCurrentMonthUsage(endUser);
+ } else if (req.method === "POST") {
+ const model = getModel((req.headers["x-openai-model"] as string) || "");
+ usage = await addUsage(endUser, model.cost_per_call);
+ }
+
+ const quota: Quota = {
+ current: usage,
+ limit: subscripion.quota,
+ };
+
+ res.status(200).json(quota);
+};
+
+export default handler;
diff --git a/src/pages/api/utils/payment.ts b/src/pages/api/utils/payment.ts
new file mode 100644
index 00000000..179fe009
--- /dev/null
+++ b/src/pages/api/utils/payment.ts
@@ -0,0 +1,25 @@
+import { Payment } from "@/types";
+import { PrismaClient } from "@prisma/client";
+
+const prisma = new PrismaClient();
+
+export const getPaymentListByEmail = async (email: string): Promise => {
+ const payments = await prisma.payment.findMany({
+ where: { email: email },
+ orderBy: { createdAt: "desc" },
+ });
+
+ const result: Payment[] = [];
+ for (const payment of payments) {
+ result.push({
+ id: payment.id,
+ email: payment.email,
+ createdAt: payment.createdAt.getTime(),
+ amount: payment.amount,
+ currency: payment.currency,
+ receipt: payment.receipt,
+ description: payment.description,
+ });
+ }
+ return result;
+};
diff --git a/src/pages/api/utils/subscription.ts b/src/pages/api/utils/subscription.ts
new file mode 100644
index 00000000..54282fd2
--- /dev/null
+++ b/src/pages/api/utils/subscription.ts
@@ -0,0 +1,47 @@
+import { PlanConfig, PlanType, Subscription } from "@/types";
+import { PrismaClient } from "@prisma/client";
+
+const prisma = new PrismaClient();
+
+export const getSubscriptionByEmail = async (email: string): Promise => {
+ const subscriptions = await prisma.subscription.findMany({
+ where: { email: email },
+ orderBy: { expireAt: "desc" },
+ });
+
+ // Return the latest active subscription if exists.
+ for (const subscription of subscriptions) {
+ const result: Subscription = {
+ id: subscription.id,
+ plan: subscription.plan as PlanType,
+ quota: PlanConfig[subscription.plan].quota,
+ startAt: subscription.startAt.getTime(),
+ expireAt: subscription.expireAt.getTime(),
+ canceledAt: subscription.canceledAt?.getTime(),
+ };
+ if (!result.canceledAt && result.expireAt > Date.now()) {
+ return result;
+ }
+ }
+
+ // Return the latest subscripion if exists.
+ for (const subscription of subscriptions) {
+ return {
+ id: subscription.id,
+ plan: subscription.plan as PlanType,
+ quota: PlanConfig["FREE"].quota,
+ startAt: subscription.startAt.getTime(),
+ expireAt: subscription.expireAt.getTime(),
+ canceledAt: subscription.canceledAt?.getTime(),
+ };
+ }
+
+ // Return a FREE subscription.
+ return {
+ id: "",
+ plan: "FREE",
+ quota: PlanConfig["FREE"].quota,
+ startAt: 0,
+ expireAt: 0,
+ };
+};
diff --git a/src/pages/api/utils/usage.ts b/src/pages/api/utils/usage.ts
new file mode 100644
index 00000000..c8fa0e1d
--- /dev/null
+++ b/src/pages/api/utils/usage.ts
@@ -0,0 +1,59 @@
+import { PrismaClient } from "@prisma/client";
+
+const prisma = new PrismaClient();
+
+export const getCurrentMonthUsage = async (endUser: string): Promise => {
+ const now = new Date();
+ const start = new Date(now.getFullYear(), now.getMonth(), 1);
+ const end = new Date(now.getFullYear(), now.getMonth() + 1, 1);
+ const aggregations = await prisma.usage.aggregate({
+ _sum: {
+ count: true,
+ },
+ where: {
+ endUser: endUser,
+ createdAt: {
+ gte: start,
+ lt: end,
+ },
+ },
+ });
+
+ return aggregations._sum.count || 0;
+};
+
+// We coerce individual usage to the begining of the day to reduce the usage records.
+export const addUsage = async (endUser: string, addition: number): Promise => {
+ const now = new Date();
+ const today = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate()));
+ const usage = await prisma.usage.findFirst({
+ where: {
+ endUser: endUser,
+ createdAt: today,
+ },
+ });
+
+ let newUsage = 0;
+ if (usage) {
+ newUsage = usage.count + addition;
+ await prisma.usage.update({
+ where: {
+ id: usage.id,
+ },
+ data: {
+ count: newUsage,
+ },
+ });
+ } else {
+ newUsage = addition;
+ await prisma.usage.create({
+ data: {
+ endUser: endUser,
+ createdAt: today,
+ count: newUsage,
+ },
+ });
+ }
+
+ return newUsage;
+};
diff --git a/src/pages/api/webhook/index.ts b/src/pages/api/webhook/index.ts
index 27e88500..dbaf7aa2 100644
--- a/src/pages/api/webhook/index.ts
+++ b/src/pages/api/webhook/index.ts
@@ -1,9 +1,11 @@
+import { PrismaClient, Prisma, SubscriptionPlan } from "@prisma/client";
import { buffer } from "micro";
import Cors from "micro-cors";
import { NextApiRequest, NextApiResponse } from "next";
-import { PrismaClient, Prisma } from "@prisma/client";
-
import Stripe from "stripe";
+import { getSubscriptionByEmail } from "../utils/subscription";
+import { getPlanFromPriceId } from "@/utils";
+
const stripe = new Stripe(process.env.STRIPE_API_KEY, {
// https://github.com/stripe/stripe-node#configuration
apiVersion: "2022-11-15",
@@ -31,11 +33,7 @@ const webhookHandler = async (req: NextApiRequest, res: NextApiResponse) => {
let event: Stripe.Event;
try {
- event = stripe.webhooks.constructEvent(
- buf.toString(),
- sig,
- webhookSecret
- );
+ event = stripe.webhooks.constructEvent(buf.toString(), sig, webhookSecret);
} catch (err) {
const errorMessage = err instanceof Error ? err.message : "Unknown error";
// On error, log and return the error message.
@@ -48,6 +46,17 @@ const webhookHandler = async (req: NextApiRequest, res: NextApiResponse) => {
// Cast event data to Stripe object.
if (event.type === "payment_intent.succeeded") {
const paymentIntent = event.data.object as Stripe.PaymentIntent;
+ const charge = await stripe.charges.retrieve(paymentIntent.latest_charge as string);
+
+ let plan;
+ try {
+ plan = getPlanFromPriceId(paymentIntent.metadata.price);
+ } catch (err) {
+ console.log(err);
+ res.status(400).send(`Invalid price id: ${paymentIntent.metadata.price}`);
+ return;
+ }
+
const customerId = paymentIntent.customer as string;
if (customerId) {
// Save the stripe customer id so that we can relate this customer to future payments.
@@ -65,22 +74,48 @@ const webhookHandler = async (req: NextApiRequest, res: NextApiResponse) => {
where: { email: paymentIntent.metadata.email },
});
- const today = new Date(new Date().setHours(0, 0, 0, 0));
- const subscription: Prisma.SubscriptionUncheckedCreateInput = {
+ const payment: Prisma.PaymentUncheckedCreateInput = {
userId: user.id,
+ email: paymentIntent.metadata.email,
createdAt: new Date(paymentIntent.created * 1000),
- status: "ACTIVE",
- startAt: today,
- expireAt: new Date(today.setFullYear(today.getFullYear() + 1)),
paymentId: paymentIntent.id,
customerId: customerId || "",
+ description: plan.description,
+ amount: paymentIntent.amount,
+ currency: paymentIntent.currency,
+ receipt: charge.receipt_url as string,
};
- await prisma.subscription.create({ data: subscription });
+ await prisma.payment.create({ data: payment });
+
+ const currentSubscription = await getSubscriptionByEmail(paymentIntent.metadata.email);
+ // Create a new subscription if there is no active paid subscription.
+ if (currentSubscription.plan === "FREE" || currentSubscription.canceledAt || currentSubscription.expireAt < new Date().getTime()) {
+ const today = new Date(new Date().setHours(0, 0, 0, 0));
+ // Subtract 1 second from the year from now to make it 23:59:59
+ const yearFromNow = new Date(new Date(new Date().setHours(0, 0, 0, 0)).setMonth(today.getMonth() + plan.month) - 1000);
+ const subscription: Prisma.SubscriptionUncheckedCreateInput = {
+ userId: user.id,
+ email: paymentIntent.metadata.email,
+ createdAt: new Date(paymentIntent.created * 1000),
+ startAt: today,
+ expireAt: yearFromNow,
+ plan: paymentIntent.metadata.plan as SubscriptionPlan,
+ };
+ await prisma.subscription.create({ data: subscription });
+ } else {
+ // Extend the current subscription if there is an active paid subscription.
+ const expireAt = new Date(Math.max(currentSubscription.expireAt, new Date().getTime()));
+ expireAt.setMonth(expireAt.getMonth() + plan.month);
+ await prisma.subscription.update({
+ where: { id: currentSubscription.id },
+ data: {
+ expireAt,
+ },
+ });
+ }
} else if (event.type === "payment_intent.payment_failed") {
const paymentIntent = event.data.object as Stripe.PaymentIntent;
- console.log(
- `❌ Payment failed: ${paymentIntent.last_payment_error?.message}`
- );
+ console.log(`❌ Payment failed: ${paymentIntent.last_payment_error?.message}`);
}
// Return a response to acknowledge receipt of the event.
diff --git a/src/pages/index.tsx b/src/pages/index.tsx
index 4cd999cd..9b5f777d 100644
--- a/src/pages/index.tsx
+++ b/src/pages/index.tsx
@@ -6,18 +6,12 @@ import React from "react";
// Use dynamic import to avoid page hydrated.
// reference: https://github.com/pmndrs/zustand/issues/1145#issuecomment-1316431268
-const ConnectionSidebar = dynamic(
- () => import("@/components/ConnectionSidebar"),
- {
- ssr: false,
- }
-);
-const ConversationView = dynamic(
- () => import("@/components/ConversationView"),
- {
- ssr: false,
- }
-);
+const ConnectionSidebar = dynamic(() => import("@/components/ConnectionSidebar"), {
+ ssr: false,
+});
+const ConversationView = dynamic(() => import("@/components/ConversationView"), {
+ ssr: false,
+});
const QueryDrawer = dynamic(() => import("@/components/QueryDrawer"), {
ssr: false,
});
@@ -26,31 +20,14 @@ const IndexPage: NextPage = () => {
return (
-
- SQL Chat - Chat-based SQL Client and Editor for the next decade
-
- SQL Chat - Chat-based SQL Client and Editor for the next decade
+ SQL Chat
@@ -61,11 +38,7 @@ const IndexPage: NextPage = () => {
Privacy
+
+
+ {t("setting.self")}
+
-
+
-
+
{
leaveTo="opacity-0"
>
- setSidebarOpen(false)}
- >
+ setSidebarOpen(false)}>
Close sidebar
-
{/* Sidebar component, swap this element with another sidebar if you like */}
-
+
@@ -116,17 +95,13 @@ const SettingPage: NextPage = () => {
{
{/* Static sidebar for desktop */}
{/* Sidebar component, swap this element with another sidebar if you like */}
-
+
@@ -163,17 +138,15 @@ const SettingPage: NextPage = () => {
href={item.href}
className={classNames(
item.current
- ? "bg-gray-50 text-indigo-600"
- : "text-gray-700 hover:text-indigo-600 hover:bg-gray-50",
- "group flex gap-x-3 rounded-md p-2 text-sm leading-6 font-semibold"
+ ? "bg-gray-50 !text-indigo-600 dark:bg-zinc-800"
+ : "text-gray-700 hover:text-indigo-600 hover:bg-gray-50 dark:hover:bg-zinc-800",
+ "group flex gap-x-3 rounded-md p-2 text-sm leading-6 font-semibold dark:text-gray-400"
)}
>
@@ -188,25 +161,17 @@ const SettingPage: NextPage = () => {
-
-
setSidebarOpen(true)}
- >
+
+ setSidebarOpen(true)}>
Open sidebar
- {router.query.session_id && (
-
-
+ {router.query.session_id &&
}
+
+
Terms
+
+
+ ;
- getOrFetchDatabaseSchema: (
- database: Database,
- skipCache?: boolean
- ) => Promise;
+ setCurrentConnectionCtx: (connectionCtx: ConnectionContext | undefined) => void;
+ getOrFetchDatabaseList: (connection: Connection, skipCache?: boolean) => Promise;
+ getOrFetchDatabaseSchema: (database: Database, skipCache?: boolean) => Promise;
getConnectionById: (connectionId: string) => Connection | undefined;
- updateConnection: (
- connectionId: string,
- connection: Partial
- ) => void;
+ updateConnection: (connectionId: string, connection: Partial) => void;
clearConnection: (filter: (connection: Connection) => boolean) => void;
}
@@ -66,33 +55,25 @@ export const useConnectionStore = create()(
...state,
currentConnectionCtx: connectionCtx,
})),
- getOrFetchDatabaseList: async (
- connection: Connection,
- skipCache = false
- ) => {
+ getOrFetchDatabaseList: async (connection: Connection, skipCache = false) => {
const state = get();
if (!skipCache) {
- if (
- state.databaseList.some(
- (database) => database.connectionId === connection.id
- )
- ) {
- return state.databaseList.filter(
- (database) => database.connectionId === connection.id
- );
+ if (state.databaseList.some((database) => database.connectionId === connection.id)) {
+ return state.databaseList.filter((database) => database.connectionId === connection.id);
}
}
const { data } = await axios.post("/api/connection/db", {
connection,
});
+
const fetchedDatabaseList = data.map(
(dbName) =>
({
connectionId: connection.id,
name: dbName,
- tableList: [],
+ schemaList: [],
} as Database)
);
const databaseList = uniqBy(
@@ -103,76 +84,54 @@ export const useConnectionStore = create()(
...state,
databaseList,
}));
- return databaseList.filter(
- (database) => database.connectionId === connection.id
- );
+ return databaseList.filter((database) => database.connectionId === connection.id);
},
- getOrFetchDatabaseSchema: async (
- database: Database,
- skipCache = false
- ) => {
+ getOrFetchDatabaseSchema: async (database: Database, skipCache = false) => {
const state = get();
if (!skipCache) {
- const db = state.databaseList.find(
- (db) =>
- db.connectionId === database.connectionId &&
- db.name === database.name
- );
- if (
- db !== undefined &&
- Array.isArray(db.tableList) &&
- db.tableList.length !== 0
- ) {
- return db.tableList;
+ const db = state.databaseList.find((db) => db.connectionId === database.connectionId && db.name === database.name);
+ if (db !== undefined && Array.isArray(db.schemaList) && db.schemaList.length !== 0) {
+ return db.schemaList;
}
}
- const connection = state.connectionList.find(
- (connection) => connection.id === database.connectionId
- );
+ const connection = state.connectionList.find((connection) => connection.id === database.connectionId);
if (!connection) {
return [];
}
- const { data: result } = await axios.post>(
- "/api/connection/db_schema",
- {
- connection,
- db: database.name,
- }
- );
+ const { data: result } = await axios.post>("/api/connection/db_schema", {
+ connection,
+ db: database.name,
+ });
+
if (result.message) {
throw result.message;
}
- const fetchedTableList = result.data;
+ const fetchedTableList: Schema[] = result.data;
+ fetchedTableList.forEach((schema) => {
+ schema.tables.forEach((table) => {
+ table.token = countTextTokens(table.structure);
+ });
+ });
set((state) => ({
...state,
databaseList: state.databaseList.map((item) =>
- item.connectionId === database.connectionId &&
- item.name === database.name
- ? { ...item, tableList: fetchedTableList }
- : item
+ item.connectionId === database.connectionId && item.name === database.name ? { ...item, schemaList: fetchedTableList } : item
),
}));
return fetchedTableList;
},
getConnectionById: (connectionId: string) => {
- return get().connectionList.find(
- (connection) => connection.id === connectionId
- );
+ return get().connectionList.find((connection) => connection.id === connectionId);
},
- updateConnection: (
- connectionId: string,
- connection: Partial
- ) => {
+ updateConnection: (connectionId: string, connection: Partial) => {
set((state) => ({
...state,
- connectionList: state.connectionList.map((item) =>
- item.id === connectionId ? { ...item, ...connection } : item
- ),
+ connectionList: state.connectionList.map((item) => (item.id === connectionId ? { ...item, ...connection } : item)),
}));
},
clearConnection: (filter: (connection: Connection) => boolean) => {
@@ -184,6 +143,16 @@ export const useConnectionStore = create()(
}),
{
name: "connection-storage",
+ version: 1,
+ migrate: (persistedState: any, version: number) => {
+ let state = persistedState as ConnectionState;
+ if (version === 0) {
+ console.info(`migrate from ${version} to 1`);
+ // to clear old data. it will make refetch new schema List
+ state.databaseList = [];
+ }
+ return state;
+ },
}
)
);
diff --git a/src/store/conversation.ts b/src/store/conversation.ts
index 44a19988..59f782f2 100644
--- a/src/store/conversation.ts
+++ b/src/store/conversation.ts
@@ -3,12 +3,12 @@ import { create } from "zustand";
import { persist } from "zustand/middleware";
import { Conversation, Id } from "@/types";
import { generateUUID } from "@/utils";
-import { GeneralBotId, SQLChatBotId } from ".";
+import { SQLChatBotId } from ".";
const getDefaultConversation = (): Conversation => {
return {
id: generateUUID(),
- assistantId: GeneralBotId,
+ assistantId: SQLChatBotId,
title: dayjs().format("LTS"),
createdAt: Date.now(),
};
@@ -18,25 +18,18 @@ interface ConversationState {
getState: () => ConversationState;
conversationList: Conversation[];
currentConversationId?: Id;
- createConversation: (
- connectionId?: Id,
- databaseName?: string
- ) => Conversation;
+ createConversation: (connectionId?: Id, databaseName?: string) => Conversation;
setCurrentConversationId: (conversationId: Id | undefined) => void;
- getConversationById: (
- conversationId: Id | undefined
- ) => Conversation | undefined;
- updateConversation: (
- conversationId: Id,
- conversation: Partial
- ) => void;
+ getConversationById: (conversationId: Id | undefined) => Conversation | undefined;
+ updateConversation: (conversationId: Id, conversation: Partial) => void;
clearConversation: (filter: (conversation: Conversation) => boolean) => void;
- updateTableName: (tableName: string) => void;
+ updateSelectedTablesNameList: (selectedTableNameList: string[]) => void;
+ updateSelectedSchemaName: (selectedSchemaName: string) => void;
}
export const useConversationStore = create()(
persist(
- (set: any, get: any) => ({
+ (set, get) => ({
getState: () => get(),
conversationList: [],
createConversation: (connectionId?: Id, databaseName?: string) => {
@@ -54,17 +47,11 @@ export const useConversationStore = create()(
}));
return conversation;
},
- setCurrentConversationId: (conversation: Id | undefined) =>
- set(() => ({ currentConversationId: conversation })),
+ setCurrentConversationId: (conversation: Id | undefined) => set(() => ({ currentConversationId: conversation })),
getConversationById: (conversationId: Id | undefined) => {
- return get().conversationList.find(
- (item: Conversation) => item.id === conversationId
- );
+ return get().conversationList.find((item: Conversation) => item.id === conversationId);
},
- updateConversation: (
- conversationId: Id,
- conversation: Partial
- ) => {
+ updateConversation: (conversationId: Id, conversation: Partial) => {
set((state: ConversationState) => ({
...state,
conversationList: state.conversationList.map((item: Conversation) =>
@@ -78,13 +65,19 @@ export const useConversationStore = create()(
conversationList: state.conversationList.filter(filter),
}));
},
- updateTableName: (tableName: string) => {
- const currentConversation = get().getConversationById(
- get().currentConversationId
- );
+ updateSelectedTablesNameList: (selectedTableNameList: string[]) => {
+ const currentConversation = get().getConversationById(get().currentConversationId);
if (currentConversation) {
get().updateConversation(currentConversation.id, {
- tableName,
+ selectedTableNameList,
+ });
+ }
+ },
+ updateSelectedSchemaName: (selectedSchemaName: string) => {
+ const currentConversation = get().getConversationById(get().currentConversationId);
+ if (currentConversation) {
+ get().updateConversation(currentConversation.id, {
+ selectedSchemaName,
});
}
},
@@ -96,11 +89,7 @@ export const useConversationStore = create()(
let state = persistedState as ConversationState;
if (version === 0) {
for (const conversation of state.conversationList) {
- if (!conversation.connectionId) {
- conversation.assistantId = "general-bot";
- } else {
- conversation.assistantId = "sql-chat-bot";
- }
+ conversation.assistantId = SQLChatBotId;
}
state.currentConversationId = undefined;
}
diff --git a/src/store/message.ts b/src/store/message.ts
index f820879f..666b18cc 100644
--- a/src/store/message.ts
+++ b/src/store/message.ts
@@ -7,6 +7,7 @@ interface MessageState {
getState: () => MessageState;
addMessage: (message: Message) => void;
updateMessage: (messageId: Id, message: Partial) => void;
+ updateStatement: (messageId: Id, originalStatement: string, replacementStatement: string) => void;
clearMessage: (filter: (message: Message) => boolean) => void;
}
@@ -15,18 +16,24 @@ export const useMessageStore = create()(
(set, get) => ({
messageList: [],
getState: () => get(),
- addMessage: (message: Message) =>
- set((state) => ({ messageList: [...state.messageList, message] })),
+ addMessage: (message: Message) => set((state) => ({ messageList: [...state.messageList, message] })),
updateMessage: (messageId: Id, message: Partial) => {
set((state) => ({
...state,
- messageList: state.messageList.map((item) =>
- item.id === messageId ? { ...item, ...message } : item
- ),
+ messageList: state.messageList.map((item) => (item.id === messageId ? { ...item, ...message } : item)),
}));
},
- clearMessage: (filter: (message: Message) => boolean) =>
- set((state) => ({ messageList: state.messageList.filter(filter) })),
+ updateStatement: (messageId: Id, originalStatement: string, replacementStatement: string) => {
+ if (!originalStatement) return;
+ const newMessage = get().messageList.find((message) => message.id == messageId);
+ if (!newMessage) return;
+ newMessage.content = newMessage.content.replace(originalStatement, replacementStatement);
+ set((state) => ({
+ ...state,
+ messageList: state.messageList.map((item) => (item.id === messageId ? newMessage : item)),
+ }));
+ },
+ clearMessage: (filter: (message: Message) => boolean) => set((state) => ({ messageList: state.messageList.filter(filter) })),
}),
{
name: "message-storage",
diff --git a/src/store/query.ts b/src/store/query.ts
index 166b0383..81cb7a99 100644
--- a/src/store/query.ts
+++ b/src/store/query.ts
@@ -1,11 +1,12 @@
import { merge } from "lodash-es";
import { create } from "zustand";
import { persist } from "zustand/middleware";
-import { Connection, Database, Timestamp } from "@/types";
+import { Connection, Database, Id, Timestamp } from "@/types";
interface ExecuteQueryContext {
connection: Connection;
database?: Database;
+ messageId?: Id;
statement: string;
}
diff --git a/src/store/setting.ts b/src/store/setting.ts
index 9524faf4..b51abaac 100644
--- a/src/store/setting.ts
+++ b/src/store/setting.ts
@@ -10,6 +10,7 @@ const getDefaultSetting = (): Setting => {
openAIApiConfig: {
key: "",
endpoint: "",
+ model: "gpt-3.5-turbo",
},
};
};
@@ -54,8 +55,7 @@ export const useSettingStore = create()(
}),
{
name: "setting-storage",
- merge: (persistedState, currentState) =>
- merge(currentState, persistedState),
+ merge: (persistedState, currentState) => merge(currentState, persistedState),
}
)
);
diff --git a/src/types/connection.ts b/src/types/connection.ts
index 522eba5f..b7bb64c1 100644
--- a/src/types/connection.ts
+++ b/src/types/connection.ts
@@ -4,12 +4,18 @@ export enum Engine {
MySQL = "MYSQL",
PostgreSQL = "POSTGRESQL",
MSSQL = "MSSQL",
+ TiDB = "TiDB",
+ Snowflake = "SNOWFLAKE",
+ Hive = "HIVE",
+ OceanBase = "OCEANBASE",
}
export interface SSLOptions {
ca?: string;
cert?: string;
key?: string;
+ minVersion?: string;
+ rejectUnauthorized?: boolean;
}
export interface Connection {
diff --git a/src/types/conversation.ts b/src/types/conversation.ts
index d2e9cc0a..b629c922 100644
--- a/src/types/conversation.ts
+++ b/src/types/conversation.ts
@@ -4,7 +4,8 @@ export interface Conversation {
id: string;
connectionId?: Id;
databaseName?: string;
- tableName?: string;
+ selectedTableNameList?: string[];
+ selectedSchemaName?: string;
assistantId: Id;
title: string;
createdAt: Timestamp;
diff --git a/src/types/database.ts b/src/types/database.ts
index 2d86fdb9..30b12612 100644
--- a/src/types/database.ts
+++ b/src/types/database.ts
@@ -1,9 +1,8 @@
import { Id } from ".";
-
export interface Database {
connectionId: Id;
name: string;
- tableList: Table[];
+ schemaList: Schema[];
}
export interface Table {
@@ -11,4 +10,9 @@ export interface Table {
// structure is a string of the table structure.
// It's mainly used for providing a chat context for the assistant.
structure: string;
+ token?: number;
+}
+export interface Schema {
+ name: string;
+ tables: Table[];
}
diff --git a/src/types/index.ts b/src/types/index.ts
index dd3fd4cc..bdeb2706 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -9,3 +9,4 @@ export * from "./api";
export * from "./connector";
export * from "./assistant";
export * from "./quota";
+export * from "./subscription";
diff --git a/src/types/quota.ts b/src/types/quota.ts
index 6054f511..c742d0bb 100644
--- a/src/types/quota.ts
+++ b/src/types/quota.ts
@@ -2,8 +2,3 @@ export interface Quota {
current: number;
limit: number;
}
-
-// By month
-export const GUEST_QUOTA = 10;
-export const FREE_QUOTA = 25;
-export const PRO_QUOTA = 1000;
diff --git a/src/types/setting.ts b/src/types/setting.ts
index 636a85f2..886dbf26 100644
--- a/src/types/setting.ts
+++ b/src/types/setting.ts
@@ -5,6 +5,7 @@ export type Theme = "light" | "dark" | "system";
export interface OpenAIApiConfig {
key: string;
endpoint: string;
+ model: string;
}
export interface Setting {
diff --git a/src/types/subscription.ts b/src/types/subscription.ts
new file mode 100644
index 00000000..f3bbdc10
--- /dev/null
+++ b/src/types/subscription.ts
@@ -0,0 +1,39 @@
+import { SubscriptionPlan } from "@prisma/client";
+
+export type PlanType = SubscriptionPlan | "GUEST" | "FREE";
+
+// Quota is by month
+export const PlanConfig: {
+ [key: string]: {
+ quota: number;
+ };
+} = {
+ GUEST: {
+ quota: 0,
+ },
+ FREE: {
+ quota: 20,
+ },
+ PRO: {
+ quota: 1000,
+ },
+};
+
+export interface Subscription {
+ id: string;
+ plan: PlanType;
+ quota: number;
+ startAt: number;
+ expireAt: number;
+ canceledAt?: number;
+}
+
+export interface Payment {
+ id: string;
+ email: string;
+ createdAt: number;
+ amount: number;
+ currency: string;
+ receipt: string;
+ description: string;
+}
diff --git a/src/utils/event-emitter.ts b/src/utils/event-emitter.ts
new file mode 100644
index 00000000..268fe3a3
--- /dev/null
+++ b/src/utils/event-emitter.ts
@@ -0,0 +1,37 @@
+export type EventType = "usage.update";
+
+type Callback = () => void;
+
+class EventEmitter {
+ private static instance: EventEmitter;
+ private events: { [eventName: string]: Callback[] } = {};
+
+ private constructor() {}
+
+ public static getInstance(): EventEmitter {
+ if (!EventEmitter.instance) {
+ EventEmitter.instance = new EventEmitter();
+ }
+ return EventEmitter.instance;
+ }
+
+ on(eventName: EventType, callback: Callback) {
+ if (!this.events[eventName]) {
+ this.events[eventName] = [];
+ }
+ this.events[eventName].push(callback);
+ }
+
+ emit(eventName: EventType) {
+ const callbacks = this.events[eventName];
+ if (callbacks) {
+ callbacks.forEach((callback) => callback());
+ }
+ }
+}
+
+const getEventEmitter = () => {
+ return EventEmitter.getInstance();
+};
+
+export default getEventEmitter;
diff --git a/src/utils/execution.ts b/src/utils/execution.ts
index 905e9188..da5ab287 100644
--- a/src/utils/execution.ts
+++ b/src/utils/execution.ts
@@ -1,12 +1,11 @@
import { ExecutionResult } from "@/types";
-export const getMessageFromExecutionResult = (
- result: ExecutionResult
-): string => {
+export const getMessageFromExecutionResult = (result: ExecutionResult): string => {
if (result.error) {
return result.error;
}
- if (result.affectedRows) {
+ // Only return the "rows affected" message if there are no raw results to display
+ if (result.affectedRows && (!result.rawResult || result.rawResult.length === 0)) {
return `${result.affectedRows} rows affected.`;
}
return "";
diff --git a/src/utils/feature.ts b/src/utils/feature.ts
index 087b6dac..0ac417f7 100644
--- a/src/utils/feature.ts
+++ b/src/utils/feature.ts
@@ -1,8 +1,26 @@
-type FeatureType = "account" | "payment" | "quota";
+type FeatureType = "debug" | "account" | "payment" | "quota" | "collect";
-export const HasFeature = (feature: FeatureType) => {
- if (process.env.NODE_ENV === "development") {
- return true;
- }
- return false;
+export const hasDatabase = () => {
+ return process.env.NEXT_PUBLIC_USE_DATABASE == "true";
+};
+
+const matrix: { [key: string]: { [feature: string]: boolean } } = {
+ development: {
+ debug: true,
+ account: hasDatabase(),
+ payment: hasDatabase(),
+ quota: hasDatabase(),
+ collect: hasDatabase(),
+ },
+ production: {
+ debug: false,
+ account: hasDatabase(),
+ payment: hasDatabase(),
+ quota: hasDatabase(),
+ collect: hasDatabase(),
+ },
+};
+
+export const hasFeature = (feature: FeatureType) => {
+ return matrix[process.env.NODE_ENV][feature];
};
diff --git a/src/utils/index.ts b/src/utils/index.ts
index 9d730bfb..5886fcbf 100644
--- a/src/utils/index.ts
+++ b/src/utils/index.ts
@@ -4,3 +4,6 @@ export * from "./sql";
export * from "./execution";
export * from "./model";
export * from "./feature";
+export * from "./tidb";
+export * from "./misc";
+export * from "./plan";
diff --git a/src/utils/misc.ts b/src/utils/misc.ts
new file mode 100644
index 00000000..a753dfda
--- /dev/null
+++ b/src/utils/misc.ts
@@ -0,0 +1,22 @@
+export const getCurrencySymbol = (currencyCode: string): string => {
+ try {
+ return new Intl.NumberFormat("en-US", {
+ style: "currency",
+ currency: currencyCode,
+ })
+ .formatToParts(1)
+ .find((part) => part.type === "currency")!.value;
+ } catch (error) {
+ console.error(`Invalid currency code: ${currencyCode}`);
+ return "";
+ }
+};
+
+export const getDateString = (timeStamp = Date.now()): string => {
+ const dateOptions: Intl.DateTimeFormatOptions = {
+ year: "numeric",
+ month: "long",
+ day: "numeric",
+ };
+ return new Date(timeStamp).toLocaleDateString(undefined, dateOptions);
+};
diff --git a/src/utils/model.ts b/src/utils/model.ts
index af555a4b..2541de72 100644
--- a/src/utils/model.ts
+++ b/src/utils/model.ts
@@ -1,6 +1,55 @@
-export const gpt35 = {
+const gpt35turbo = {
name: "gpt-3.5-turbo",
temperature: 0,
frequency_penalty: 0.0,
presence_penalty: 0.0,
+ max_token: 4000,
+ cost_per_call: 1,
+};
+
+const gpt4 = {
+ name: "gpt-4",
+ temperature: 0,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ max_token: 8000,
+ cost_per_call: 60,
+};
+
+const gpt4turbo = {
+ name: "gpt-4-turbo",
+ temperature: 0,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ max_token: 4000,
+ cost_per_call: 20,
+};
+
+const gpt4ho = {
+ name: "gpt-4o",
+ temperature: 0,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ max_token: 4000,
+ cost_per_call: 10,
+};
+
+const deepseekChat = {
+ name: "deepseek-chat",
+ temperature: 0,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ max_token: 12000,
+ cost_per_call: 1,
+};
+
+export const models = [gpt35turbo, gpt4, gpt4turbo, gpt4ho, deepseekChat];
+
+export const getModel = (name: string) => {
+ for (const model of models) {
+ if (model.name === name) {
+ return model;
+ }
+ }
+ return gpt35turbo;
};
diff --git a/src/utils/openai.ts b/src/utils/openai.ts
index cc74b0e3..a152406d 100644
--- a/src/utils/openai.ts
+++ b/src/utils/openai.ts
@@ -1,12 +1,59 @@
import { encode } from "@nem035/gpt-3-encoder";
+import { Engine, Schema, Table } from "@/types";
// openAIApiKey is the API key for OpenAI API.
export const openAIApiKey = process.env.OPENAI_API_KEY;
// openAIApiEndpoint is the API endpoint for OpenAI API. Defaults to https://api.openai.com.
-export const openAIApiEndpoint =
- process.env.OPENAI_API_ENDPOINT || "https://api.openai.com";
+export const openAIApiEndpoint = process.env.OPENAI_API_ENDPOINT || "https://api.openai.com";
+
+// openAIOrganization a header to specify which organization is used for an API request.
+export const openAIOrganization = process.env.OPENAI_ORGANIZATION;
export const countTextTokens = (text: string) => {
return encode(text).length;
};
+
+export function generateDbPromptFromContext(
+ promptGenerator: (engine: Engine | undefined, schema: string | undefined) => string,
+ engine: Engine,
+ schemaList: Schema[],
+ selectedSchemaName: string,
+ selectedTableNameList: string[],
+ maxToken: number,
+ userPrompt?: string
+): string {
+ // userPrompt is the message that user want to send to bot. When to look prompt in drawer, userPrompt is undefined.
+ let tokens = countTextTokens(userPrompt || "");
+
+ // Empty table name(such as []) denote all table. [] and `undefined` both are false in `if`
+ // The above comment is out of date. [] is true in `if` now. And no selected table should not denote all table now.
+ // Because in have Token custom number in connectionSidebar. If [] denote all table. the Token will be inconsistent.
+ const tableList: string[] = [];
+ const selectedSchema = schemaList.find((schema: Schema) => schema.name == (selectedSchemaName || ""));
+ if (selectedTableNameList) {
+ selectedTableNameList.forEach((tableName: string) => {
+ const table = selectedSchema?.tables.find((table: Table) => table.name == tableName);
+ tableList.push(table!.structure);
+ });
+ } else {
+ for (const table of selectedSchema?.tables || []) {
+ tableList.push(table!.structure);
+ }
+ }
+
+ let finalTableList = [];
+ if (tableList) {
+ for (const table of tableList) {
+ if (tokens < maxToken / 2) {
+ tokens += countTextTokens(table + "\n\n");
+ finalTableList.push(table);
+ }
+ }
+ }
+ return promptGenerator(engine, finalTableList.join("\n\n"));
+}
+
+export function allowSelfOpenAIKey() {
+ return process.env.NEXT_PUBLIC_ALLOW_SELF_OPENAI_KEY == "true";
+}
diff --git a/src/utils/plan.ts b/src/utils/plan.ts
new file mode 100644
index 00000000..83536774
--- /dev/null
+++ b/src/utils/plan.ts
@@ -0,0 +1,22 @@
+export const getPlanFromPriceId = (priceId: string) => {
+ switch (priceId) {
+ case process.env.NEXT_PUBLIC_STRIPE_PRICE_ID_PRO_1_MONTH_SUBSCRIPTION:
+ return {
+ month: 1,
+ description: "Pro 1 Month (Early Bird)",
+ };
+ case process.env.NEXT_PUBLIC_STRIPE_PRICE_ID_PRO_3_MONTH_SUBSCRIPTION:
+ return {
+ month: 1,
+ description: "Pro 3 Months (Early Bird)",
+ };
+ case process.env.NEXT_PUBLIC_STRIPE_PRICE_ID_PRO_1_YEAR_SUBSCRIPTION:
+ return {
+ month: 12,
+ description: "Pro 1 Year (Early Bird)",
+ };
+ default: {
+ throw Error(`Invalid price ID ${priceId}`);
+ }
+ }
+};
diff --git a/src/utils/tidb.ts b/src/utils/tidb.ts
new file mode 100644
index 00000000..2d4d18ec
--- /dev/null
+++ b/src/utils/tidb.ts
@@ -0,0 +1,16 @@
+import { Connection } from "@/types";
+import { Engine } from "@/types/connection";
+
+export const changeTiDBConnectionToMySQL = (connection: Connection) => {
+ if (connection.engineType === Engine.TiDB) {
+ return {
+ ...connection,
+ engineType: Engine.MySQL,
+ ssl: {
+ minVersion: "TLSv1.2",
+ rejectUnauthorized: true,
+ },
+ };
+ }
+ return connection;
+};
diff --git a/tailwind.config.js b/tailwind.config.js
index c68b0430..fbb7a447 100644
--- a/tailwind.config.js
+++ b/tailwind.config.js
@@ -1,9 +1,6 @@
/** @type {import('tailwindcss').Config} */
module.exports = {
- content: [
- "./src/pages/**/*.{js,ts,jsx,tsx}",
- "./src/components/**/*.{js,ts,jsx,tsx}",
- ],
+ content: ["./src/pages/**/*.{js,ts,jsx,tsx}", "./src/components/**/*.{js,ts,jsx,tsx}"],
darkMode: "class",
theme: {
extend: {
@@ -13,5 +10,5 @@ module.exports = {
},
},
},
- plugins: [require("@tailwindcss/typography")],
+ plugins: [require("@tailwindcss/typography"), require("tailwind-scrollbar-hide")],
};
diff --git a/tsconfig.json b/tsconfig.json
index 6344ce50..a62f1358 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -18,6 +18,6 @@
"@/*": ["./src/*"]
}
},
- "include": ["next-env.d.ts", "process.d.ts", "**/*.ts", "**/*.tsx"],
+ "include": ["next-env.d.ts", "process.d.ts", "auth.d.ts", "**/*.ts", "**/*.tsx"],
"exclude": ["node_modules"]
}
+
+
+
+
+## Patrocinadores
+
+
+
+
-```bash
-docker run --name sqlchat --platform linux/amd64 --env OPENAI_API_KEY=xxx --env OPENAI_API_ENDPOINT=yyy -p 3000:3000 sqlchat/sqlchat
-```
+## Startup options
-## Local Development
+## TL;DR
-1. Make a copy of the example environment variables file;
+- If you just want to use for yourself, then run without database. Check [.env.nodb](https://github.com/sqlchat/sqlchat/blob/main/.env.nodb).
+- If you want to offer a similar service as [sqlchat.ai](https://sqlchat.ai), then run with database, check [.env.usedb](https://github.com/sqlchat/sqlchat/blob/main/.env.usedb). The database is used to store account, usage info.
- ```bash
- cp .env.example .env
- ```
+### OpenAI related
-2. Add your [API key](https://platform.openai.com/account/api-keys) and OpenAI API Endpoint(optional) to the newly created `.env` file;
+- `OPENAI_API_KEY`: OpenAI API key. You can get one from [here](https://platform.openai.com/api-keys).
-3. Install dependencies and start the dev server;
+- `OPENAI_API_ENDPOINT`: OpenAI API endpoint. Defaults to `https://api.openai.com`. Use [Ollama](https://github.com/ollama/ollama) to set up self-host AI model and set the endpoint to it.
- ```bash
- pnpm i && pnpm dev
- ```
+- `NEXT_PUBLIC_ALLOW_SELF_OPENAI_KEY`: Set to `true` to allow users to bring their own OpenAI API key.
+
+
+### Database related
-### Database Setup
+- `NEXT_PUBLIC_USE_DATABASE`: Set to `true` to start SQL Chat with database. This will
+ enable following features:
+ 1. Account system.
+ 1. Per-user quota enforcement.
+ 1. Payment.
+ 1. Usage data collection.
+- `DATABASE_URL`: Applicable if `NEXT_PUBLIC_USE_DATABASE` is `true`. Postgres connection string to store data. e.g. `postgresql://postgres:YOUR_PASSWORD@localhost:5432/sqlchat?schema=sqlchat`.
+
+## Local Development
1. Install dependencies
@@ -83,19 +99,41 @@ docker run --name sqlchat --platform linux/amd64 --env OPENAI_API_KEY=xxx --env
pnpm i
```
-1. Generate prisma client from the model
+1. Generate prisma client
```bash
pnpm prisma generate
```
-1. Migrate schema
+1. Make a copy of the example environment variables file:
+
+ ```bash
+ cp .env.usedb .env
+ ```
+
+1. Add your [API key](https://platform.openai.com/account/api-keys) and OpenAI API Endpoint(optional) to the newly created `.env` file.
+
+### Setup database
+
+**You can skip this section with `NEXT_PUBLIC_USE_DATABASE=false` if you don't build features requiring database**
+
+1. Start a Postgres instance. For mac, you can use [StackbBricks](https://stackbricks.app/), [DBngin](https://dbngin.com/) or [Postgres.app](https://postgresapp.com/).
+
+1. Create a database:
+
+ ```sql
+ CREATE DATABASE sqlchat;
+ ```
+
+ In `.env` file, assign the connection string to environment variable `DATABASE_URL` and `DATABASE_DIRECT_URL`. [This article](https://www.prisma.io/docs/data-platform/data-proxy/prisma-cli-with-data-proxy#set-a-direct-database-connection-url-in-your-prisma-schema) explains why we need two URLs.
+
+1. Set up database schema
```bash
pnpm prisma migrate dev
```
-1. Seed data
+1. (Optional) Seed data
```bash
pnpm prisma db seed
@@ -107,8 +145,6 @@ docker run --name sqlchat --platform linux/amd64 --env OPENAI_API_KEY=xxx --env
## Community
-[](https://discord.gg/z6kakemDjm)
-
[](https://twitter.com/Bytebase)