@@ -98,7 +103,7 @@ export const SettingOther = () => {
-
+
{t("generalSettings.system.deleteChatHistory.label")}
@@ -122,6 +127,37 @@ export const SettingOther = () => {
{t("generalSettings.system.deleteChatHistory.button")}
+
+
+ {t("generalSettings.system.export.label")}
+
+
+
+
+
+ {t("generalSettings.system.import.label")}
+
+
+ {
+ if (e.target.files) {
+ importPageAssistData(e.target.files[0])
+ }
+ }}
+ />
+
)
diff --git a/src/db/index.ts b/src/db/index.ts
index bc8b2f8..6ca341f 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -444,3 +444,15 @@ export const importChatHistory = async (
}
}
}
+
+export const exportPrompts = async () => {
+ const db = new PageAssitDatabase()
+ return await db.getAllPrompts()
+}
+
+export const importPrompts = async (prompts: Prompts) => {
+ const db = new PageAssitDatabase()
+ for (const prompt of prompts) {
+ await db.addPrompt(prompt)
+ }
+}
diff --git a/src/db/vector.ts b/src/db/vector.ts
index 2c2e281..09624fc 100644
--- a/src/db/vector.ts
+++ b/src/db/vector.ts
@@ -112,6 +112,22 @@ export class PageAssistVectorDb {
})
})
}
+
+ saveImportedData = async (data: VectorData[]): Promise
=> {
+ return new Promise((resolve, reject) => {
+ const obj: Record = {}
+ data.forEach((d) => {
+ obj[d.id] = d
+ })
+ this.db.set(obj, () => {
+ if (chrome.runtime.lastError) {
+ reject(chrome.runtime.lastError)
+ } else {
+ resolve()
+ }
+ })
+ })
+ }
}
export const insertVector = async (
@@ -148,7 +164,5 @@ export const exportVectors = async () => {
export const importVectors = async (data: VectorData[]) => {
const db = new PageAssistVectorDb()
- for (const d of data) {
- await db.insertVector(d.id, d.vectors)
- }
+ return db.saveImportedData(data)
}
diff --git a/src/libs/export-import.ts b/src/libs/export-import.ts
new file mode 100644
index 0000000..50bda8b
--- /dev/null
+++ b/src/libs/export-import.ts
@@ -0,0 +1,66 @@
+import {
+ exportChatHistory,
+ exportPrompts,
+ importChatHistory,
+ importPrompts
+} from "@/db"
+import { exportKnowledge, importKnowledge } from "@/db/knowledge"
+import { exportVectors, importVectors } from "@/db/vector"
+import { message } from "antd"
+
+export const exportPageAssistData = async () => {
+ const knowledge = await exportKnowledge()
+ const chat = await exportChatHistory()
+ const vector = await exportVectors()
+ const prompts = await exportPrompts()
+
+ const data = {
+ knowledge,
+ chat,
+ vector,
+ prompts
+ }
+
+ const dataStr = JSON.stringify(data)
+
+ const blob = new Blob([dataStr], { type: "application/json" })
+ const url = URL.createObjectURL(blob)
+
+ const a = document.createElement("a")
+ a.href = url
+ a.download = `page-assist-${new Date().toISOString()}.json`
+ a.click()
+ URL.revokeObjectURL(url)
+}
+
+export const importPageAssistData = async (file: File) => {
+ const reader = new FileReader()
+ reader.onload = async () => {
+ try {
+ const data = JSON.parse(reader.result as string)
+
+ if (data?.knowledge) {
+ await importKnowledge(data.knowledge)
+ }
+
+ if (data?.chat) {
+ await importChatHistory(data.chat)
+ }
+
+ if (data?.vector) {
+ await importVectors(data.vector)
+ }
+
+ if (data?.prompts) {
+ await importPrompts(data.prompts)
+ }
+
+ message.success("Data imported successfully")
+ } catch (e) {
+ console.error(e)
+ message.error("Failed to import data")
+ }
+ }
+
+ reader.readAsText(file)
+}