+
+
+ 提示
+
+ {{ state.errorMessage }}
+
+
提示
=> {
+ const { data } = await toolkitApi.get('/title/search', { params })
+ return data.data.titles
+ },
+ getConversationList: async (params: GetConversationListParams = {}): Promise => {
+ const { data } = await toolkitApi.get('/conversation/list', { params })
+ return data.data.conversations
+ },
+ getConversationInfo: async (params: GetConversationInfoParams): Promise => {
+ const { data } = await toolkitApi.get(`/conversation/info`, { params })
+ return data.data
+ },
+ setConversationTitle: async (params: SetConversationTitleParams): Promise => {
+ await toolkitApi.post(`/conversation/set_title`, params)
+ },
+ setConversationPinned: async (params: SetConversationPinnedParams): Promise => {
+ await toolkitApi.post(`/conversation/set_pinned`, params)
+ },
+ removeConversation: async (params: RemoveConversationParams): Promise => {
+ let apiParams: any = {}
+ if (params.ids) {
+ apiParams.ids = params.ids.join(',')
+ } else if (params.id) {
+ apiParams.id = params.id
+ }
+ await toolkitApi.post(`/conversation/remove`, apiParams)
+ },
+}
\ No newline at end of file
diff --git a/src/api/chatcomplete.ts b/src/api/chatcomplete.ts
new file mode 100644
index 0000000..7387ba0
--- /dev/null
+++ b/src/api/chatcomplete.ts
@@ -0,0 +1,190 @@
+import { RemoteError } from "@/errors/remoteError"
+import { toolkitApi } from "@/request/toolkit-api";
+import { useUserStore } from "@/stores"
+import { ChatCompleteChunkInfo, ExtractDocInfo } from "@/types/base";
+import { getWebSocketApiUrl } from "@/utils/request";
+
+export type IndexPageParams = {
+ title: string,
+ collection?: boolean
+}
+
+export type IndexPageOnProgress = (current: number, total: number, current_page: number, total_page: number) => any
+
+export type IndexPageResponse = {
+ indexUpdated?: boolean
+}
+
+export type GetConversationChunkListParams = {
+ id: number
+}
+
+export type GetConversationChunkListResponse = number[]
+
+export type GetConversationChunkInfoParams = {
+ chunk_id: number
+}
+
+export type GetConversationChunkInfoResponse = {
+ id: number,
+ conversation_id: number,
+ message_data: ChatCompleteChunkInfo,
+ tokens: number,
+ updated_at: number
+}
+
+export type ForkConversationParams = {
+ id: number,
+ message_id: string,
+ new_title?: string
+}
+
+export type ForkConversationResponse = {
+ conversation_id: number
+}
+
+export type StartChatCompleteResponse = {
+ question_tokens: number,
+ extract_doc: Record,
+ conversation_id: number,
+ chunk_id: number,
+ task_id: string,
+}
+
+export type ChatCompleteServiceResponse = {
+ message: string,
+ message_tokens: number,
+ total_tokens: number,
+ finish_reason: string,
+ question_message_id: string,
+ response_message_id: string,
+ delta_data: Record,
+ point_cost: number,
+}
+
+export type GetPointCostParams = {
+ question: string,
+ extract_limit?: number,
+}
+
+export const chatCompleteApi = {
+ indexPage(params: IndexPageParams, onProgress: IndexPageOnProgress): Promise {
+ return new Promise((resolve, reject) => {
+ let url = getWebSocketApiUrl('/embedding_search/index_page', params)
+
+ let isFinished = false
+ let ws = new WebSocket(url)
+ ws.addEventListener('message', (event) => {
+ let data = JSON.parse(event.data)
+ if (data.event === 'progress') {
+ onProgress(data.current, data.total, data.current_page, data.total_page)
+ } else if (data.event === 'done') {
+ isFinished = true
+ resolve({
+ indexUpdated: data.index_updated
+ })
+ } else if (data.event === 'error') {
+ isFinished = true
+ reject(new RemoteError(data.message, data.error))
+ }
+ })
+
+ ws.addEventListener('error', (event) => {
+ reject(new Error('连接异常：' + event.type))
+ })
+
+ ws.addEventListener('close', (event) => {
+ if (!isFinished) {
+ reject(new Error('连接异常关闭：' + event.code + ' ' + event.reason))
+ } else {
+ resolve({
+ indexUpdated: false
+ })
+ }
+ })
+ })
+ },
+
+ async getConversationChunkList(params: GetConversationChunkListParams): Promise {
+ const { data } = await toolkitApi.get(`/chatcomplete/conversation_chunk/list`, { params })
+ return data.data
+ },
+
+ async getConversationChunkInfo(params: GetConversationChunkInfoParams): Promise {
+ const { data } = await toolkitApi.get(`/chatcomplete/conversation_chunk/info`, { params })
+ return data.data
+ },
+
+ async getPointCost(params: GetPointCostParams): Promise {
+ const { data } = await toolkitApi.post(`/chatcomplete/get_point_cost`, params)
+ return data.data?.point_cost ?? -1
+ },
+
+ async forkConversation(params: ForkConversationParams): Promise {
+ const { data } = await toolkitApi.post(`/chatcomplete/conversation/fork`, params)
+ return data.data
+ },
+
+ async startChatComplete(title: string, question: string, args: {
+ conversationId?: number,
+ extractLimit?: number,
+ inCollection?: boolean,
+ editMessageId?: [number, string]
+ } = {}): Promise {
+ let params: any = {
+ title: title,
+ question: question
+ }
+ if (args.conversationId) {
+ params.conversation_id = args.conversationId
+ }
+ if (args.extractLimit) {
+ params.extract_limit = args.extractLimit
+ }
+ if (args.inCollection) {
+ params.in_collection = args.inCollection
+ }
+ if (args.editMessageId) {
+ params.edit_message_id = args.editMessageId.join(',')
+ }
+ const { data } = await toolkitApi.post('/chatcomplete/message', params)
+ return data.data
+ },
+
+ chatCompleteStreamOutput(taskId: string, onMessage: (deltaMessage: string) => any): Promise {
+ return new Promise((resolve, reject) => {
+ let url = getWebSocketApiUrl('/chatcomplete/message/stream', {
+ task_id: taskId
+ })
+
+ let isFinished = false
+ let ws = new WebSocket(url)
+ ws.addEventListener('message', (event) => {
+ if (event.data.startsWith('+')) { // 实时输出的消息是以+开头的纯文本
+ onMessage(event.data.substring(1))
+ } else { // 其他消息是JSON格式
+ let data = JSON.parse(event.data)
+ if (data.event === 'connected') {
+ onMessage(data.outputed_message)
+ } else if (data.event === 'finished') {
+ isFinished = true
+ resolve(data.result)
+ } else if (data.event === 'error') {
+ isFinished = true
+ reject(new RemoteError(data.message, data.error))
+ }
+ }
+ })
+
+ ws.addEventListener('error', (event) => {
+ reject(new Error('连接异常：' + event.type))
+ })
+
+ ws.addEventListener('close', (event) => {
+ if (!isFinished) {
+ reject(new Error('连接异常关闭：' + event.code + ' ' + event.reason))
+ }
+ })
+ })
+ }
+}
\ No newline at end of file
diff --git a/src/api/index.ts b/src/api/index.ts
new file mode 100644
index 0000000..d032229
--- /dev/null
+++ b/src/api/index.ts
@@ -0,0 +1,7 @@
+import { baseApi } from "./base";
+import { chatCompleteApi } from "./chatcomplete";
+
+export const api = {
+ base: baseApi,
+ chat: chatCompleteApi
+};
\ No newline at end of file
diff --git a/src/auto-imports.d.ts b/src/auto-imports.d.ts
index e8959eb..4259e89 100644
--- a/src/auto-imports.d.ts
+++ b/src/auto-imports.d.ts
@@ -45,6 +45,7 @@ declare global {
const toRaw: typeof import('vue')['toRaw']
const toRef: typeof import('vue')['toRef']
const toRefs: typeof import('vue')['toRefs']
+ const toValue: typeof import('vue')['toValue']
const triggerRef: typeof import('vue')['triggerRef']
const unref: typeof import('vue')['unref']
const useAttrs: typeof import('vue')['useAttrs']
diff --git a/src/components.d.ts b/src/components.d.ts
index 39bdf6f..05f6941 100644
--- a/src/components.d.ts
+++ b/src/components.d.ts
@@ -9,10 +9,10 @@ export {}
declare module '@vue/runtime-core' {
export interface GlobalComponents {
+ ChatCompleteSettingsModal: typeof import('./components/ChatCompleteSettingsModal.vue')['default']
ChatMessage: typeof import('./components/ChatMessage.vue')['default']
IonApp: typeof import('@ionic/vue')['IonApp']
IonAvatar: typeof import('@ionic/vue')['IonAvatar']
- IonBackButton: typeof import('@ionic/vue')['IonBackButton']
IonButton: typeof import('@ionic/vue')['IonButton']
IonButtons: typeof import('@ionic/vue')['IonButtons']
IonCard: typeof import('@ionic/vue')['IonCard']
@@ -20,6 +20,7 @@ declare module '@vue/runtime-core' {
IonCardHeader: typeof import('@ionic/vue')['IonCardHeader']
IonCardSubtitle: typeof import('@ionic/vue')['IonCardSubtitle']
IonCardTitle: typeof import('@ionic/vue')['IonCardTitle']
+ IonCheckbox: typeof import('@ionic/vue')['IonCheckbox']
IonChip: typeof import('@ionic/vue')['IonChip']
IonCol: typeof import('@ionic/vue')['IonCol']
IonContent: typeof import('@ionic/vue')['IonContent']
@@ -27,28 +28,36 @@ declare module '@vue/runtime-core' {
IonGrid: typeof import('@ionic/vue')['IonGrid']
IonHeader: typeof import('@ionic/vue')['IonHeader']
IonIcon: typeof import('@ionic/vue')['IonIcon']
- IonInput: typeof import('@ionic/vue')['IonInput']
+ IonInfiniteScroll: typeof import('@ionic/vue')['IonInfiniteScroll']
+ IonInfiniteScrollContent: typeof import('@ionic/vue')['IonInfiniteScrollContent']
IonItem: typeof import('@ionic/vue')['IonItem']
+ IonItemOption: typeof import('@ionic/vue')['IonItemOption']
+ IonItemOptions: typeof import('@ionic/vue')['IonItemOptions']
+ IonItemSliding: typeof import('@ionic/vue')['IonItemSliding']
IonLabel: typeof import('@ionic/vue')['IonLabel']
IonList: typeof import('@ionic/vue')['IonList']
IonListHeader: typeof import('@ionic/vue')['IonListHeader']
IonMenu: typeof import('@ionic/vue')['IonMenu']
IonMenuButton: typeof import('@ionic/vue')['IonMenuButton']
IonMenuToggle: typeof import('@ionic/vue')['IonMenuToggle']
+ IonModal: typeof import('@ionic/vue')['IonModal']
IonPage: typeof import('@ionic/vue')['IonPage']
+ IonPopover: typeof import('@ionic/vue')['IonPopover']
IonRouterOutlet: typeof import('@ionic/vue')['IonRouterOutlet']
IonRow: typeof import('@ionic/vue')['IonRow']
IonSearchbar: typeof import('@ionic/vue')['IonSearchbar']
+ IonSelect: typeof import('@ionic/vue')['IonSelect']
+ IonSelectOption: typeof import('@ionic/vue')['IonSelectOption']
IonSpinner: typeof import('@ionic/vue')['IonSpinner']
IonSplitPane: typeof import('@ionic/vue')['IonSplitPane']
IonTextarea: typeof import('@ionic/vue')['IonTextarea']
IonTitle: typeof import('@ionic/vue')['IonTitle']
+ IonToggle: typeof import('@ionic/vue')['IonToggle']
IonToolbar: typeof import('@ionic/vue')['IonToolbar']
LoadingTitle: typeof import('./components/LoadingTitle.vue')['default']
- Markdown: typeof import('./components/Markdown.vue')['default']
MarkdownParser: typeof import('./components/MarkdownParser.vue')['default']
- MessageList: typeof import('./components/MessageList.vue')['default']
RouterLink: typeof import('vue-router')['RouterLink']
RouterView: typeof import('vue-router')['RouterView']
+ TypeaheadModal: typeof import('./components/TypeaheadModal.vue')['default']
}
}
diff --git a/src/components/ChatCompleteSettingsModal.vue b/src/components/ChatCompleteSettingsModal.vue
new file mode 100644
index 0000000..77129be
--- /dev/null
+++ b/src/components/ChatCompleteSettingsModal.vue
@@ -0,0 +1,126 @@
+
+
+
+
+
+
+
+
+
+ 选项
+
+
+ 完成
+
+
+
+
+
+
+
+
+ 在设定集中提问
+
+
+
+
+
+
+ 不抓取设定
+ 1
+ 3
+ 5
+ 10
+ 15
+
+
+
+
+
+
+ 更新页面索引
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/components/ChatMessage.vue b/src/components/ChatMessage.vue
index ba84eab..8badc84 100644
--- a/src/components/ChatMessage.vue
+++ b/src/components/ChatMessage.vue
@@ -5,13 +5,39 @@ export default {
@@ -26,11 +52,25 @@ const props = defineProps<{
@@ -42,15 +82,12 @@ const props = defineProps<{
@import '@/theme/content-container.scss';
.message-container {
- border-top: 1px solid rgba(var(--ion-text-color-rgb), 0.1);
- width: 100%;
-
- &:first-of-type {
- border-top: none;
- }
-
&.assistant {
- background: var(--ion-color-light-tint);
+ background: #f8f8f8;
+
+ @media (prefers-color-scheme: dark) {
+ background: #101010;
+ }
}
.message {
@@ -70,7 +107,7 @@ const props = defineProps<{
flex-basis: 0;
}
- @media screen and (min-width: 768px) {
+ @media (min-width: 768px) {
.avatar-fill {
width: 48px;
flex-basis: 48px;
@@ -87,6 +124,12 @@ const props = defineProps<{
display: flex;
align-items: center;
gap: 8px;
+
+ @media (max-width: 767px) {
+ flex-direction: column;
+ align-items: flex-start;
+ gap: 4px;
+ }
}
.message-sender {
diff --git a/src/components/LoadingTitle.vue b/src/components/LoadingTitle.vue
index 9360ac8..d6a8ec9 100644
--- a/src/components/LoadingTitle.vue
+++ b/src/components/LoadingTitle.vue
@@ -12,7 +12,7 @@ const props = defineProps<{
{{ state.errorMessage }}
diff --git a/src/api/base.ts b/src/api/base.ts new file mode 100644 index 0000000..9cefb7f --- /dev/null +++ b/src/api/base.ts @@ -0,0 +1,64 @@ +import { toolkitApi } from "@/request/toolkit-api"; +import { ConversationInfo } from "@/types/base"; + +export type SearchTitleParams = { + kw: string +} + +export type SearchTitleResponse = string[] + +export type GetConversationListParams = { + title?: string, + module?: string +} + +export type GetConversationListResponse = ConversationInfo[] + +export type GetConversationInfoParams = { + id: number +} + +export type SetConversationTitleParams = { + id: number, + new_title: string +} + +export type SetConversationPinnedParams = { + id: number, + pinned: boolean +} + +export type RemoveConversationParams = { + id?: number, + ids?: number[] +} + +export const baseApi = { + searchTitle: async (params: SearchTitleParams): Promise开启此功能后，将会在整个设定集中抓取设定。
+ 更多的抓取数量也许能够提升回答精确度，但是每次提问需要更多积分。推荐使用5或10。
+
-
+
@@ -31,6 +31,12 @@ $spinner-width: 44px;
width: 100%;
}
+.loading-title-title {
+ overflow-x: hidden;
+ text-overflow: ellipsis;
+ white-space: nowrap;
+}
+
.loading-title-loading-spinner {
overflow: hidden;
opacity: 0;
diff --git a/src/components/MarkdownParser.vue b/src/components/MarkdownParser.vue
index c64c857..48869ca 100644
--- a/src/components/MarkdownParser.vue
+++ b/src/components/MarkdownParser.vue
@@ -8,17 +8,26 @@ export default {
import MarkdownIt from 'markdown-it'
import hljs from 'highlight.js'
import 'github-markdown-css/github-markdown.css'
+import { move } from 'ionicons/icons';
const props = defineProps<{
- md: string
+ md: string,
+ cursor?: boolean
}>()
const outputRef = ref()
+const textTags = ['P', 'H1', 'H2', 'H3', 'H4', 'H5', 'H6', 'DIV']
+
let currentChunkDom: HTMLDivElement | null = null
let buffer = '';
let inCodeBlock = false;
+let cursorDom = document.createElement('span')
+cursorDom.classList.add('text-output-cursor')
+cursorDom.innerText = '\u2589'
+cursorDom.style.display = 'none'
+
const md = new MarkdownIt({
html: false,
breaks: true,
@@ -34,6 +43,21 @@ const md = new MarkdownIt({
}
})
+const moveCursor = () => {
+ if (outputRef.value) {
+ // 放置光标
+ let lastMarkdownChild = outputRef.value.lastElementChild?.lastElementChild
+ if (!lastMarkdownChild) {
+ lastMarkdownChild = outputRef.value.lastElementChild
+ }
+ if (lastMarkdownChild && textTags.includes(lastMarkdownChild.tagName)) {
+ lastMarkdownChild.appendChild(cursorDom)
+ } else {
+ outputRef.value.appendChild(cursorDom)
+ }
+ }
+}
+
onMounted(() => {
nextTick(() => {
if (outputRef.value) {
@@ -50,6 +74,8 @@ onMounted(() => {
} else {
buffer = ''
}
+
+ moveCursor()
}
})
})
@@ -102,7 +128,7 @@ watch(() => props.md, (newVal, oldVal) => {
chunkDom.innerHTML = md.render(newVal)
outputRef.value.appendChild(chunkDom)
currentChunkDom = chunkDom
-
+
let lastNewBlock = newVal.lastIndexOf('\n\n')
if (lastNewBlock !== -1) {
buffer = newVal.substring(lastNewBlock + 2)
@@ -110,24 +136,54 @@ watch(() => props.md, (newVal, oldVal) => {
buffer = ''
}
}
+ moveCursor()
}
})
+
+watch(() => props.cursor, (cursorEnabled) => {
+ if (cursorEnabled) {
+ cursorDom.style.display = 'inline'
+ } else {
+ cursorDom.style.display = 'none'
+ }
+}, { immediate: true })
-
+
\ No newline at end of file
diff --git a/src/components/TypeaheadModal.vue b/src/components/TypeaheadModal.vue
new file mode 100644
index 0000000..1df40c4
--- /dev/null
+++ b/src/components/TypeaheadModal.vue
@@ -0,0 +1,200 @@
+
+
+
+
+
+
+
+
+
+ 取消
+
+
+
+ {{ $props.title }}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/errors/remoteError.ts b/src/errors/remoteError.ts
new file mode 100644
index 0000000..6e5bf99
--- /dev/null
+++ b/src/errors/remoteError.ts
@@ -0,0 +1,13 @@
+export class RemoteError extends Error {
+ public code: string = 'unknown_error'
+ public info?: string
+ public title?: string
+
+ constructor(message: string, data: any) {
+ super(message)
+
+ if (data.code) this.code = data.code
+ if (data.info) this.info = data.info
+ if (data.title) this.title = data.title
+ }
+}
\ No newline at end of file
diff --git a/src/global.d.ts b/src/global.d.ts
index ec24954..d5459c3 100644
--- a/src/global.d.ts
+++ b/src/global.d.ts
@@ -2,4 +2,4 @@ declare global {
interface Window {
_loggedIn?: boolean
}
-}
\ No newline at end of file
+}
diff --git a/src/layouts/ToolboxLayout.vue b/src/layouts/ToolboxLayout.vue
new file mode 100644
index 0000000..dbeafea
--- /dev/null
+++ b/src/layouts/ToolboxLayout.vue
@@ -0,0 +1,655 @@
+
+
+
+
+
+
+
+
+
+
+ {{ pointsText }}
+
+
+
+
+ {{ pageStore.displayTitle || pageStore.title || '未选择页面' }}
+
+
+
+
+ {{ pointsText }}
+
+
+ ![]()
+
+
+
+
+
+
+
+
+
+
+
+ 编辑
+
+
+ 取消
+
+
+
+
+ 对话列表
+
+
+
+
+
+
+
+
+
+
+
+ 这里还什么都没有呢
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 更改标题
+
+
+
+
+
+
+
+
+
+
+ 收藏
+
+
+ 取消收藏
+
+
+ 更改标题
+
+
+ 删除
+
+
+
+
+
+
+
+
+
+ 选择全部
+
+
+ 取消选择
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/layouts/ToolkitLayout.vue b/src/layouts/ToolkitLayout.vue
deleted file mode 100644
index 66cd29c..0000000
--- a/src/layouts/ToolkitLayout.vue
+++ /dev/null
@@ -1,264 +0,0 @@
-
-
-
-
-
-
-
-
-
-
- {{ store.state.pageTitle }}
-
-
- {{ pointsText }}
-
-
- ![]()
-
-
-
-
-
-
-
-
-
-
-
- 对话列表
-
-
-
-
-
-
-
-
-
-
-
- 对话列表
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
\ No newline at end of file
diff --git a/src/main.ts b/src/main.ts
index 86e4db0..c681e5f 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -23,10 +23,19 @@ import '@ionic/vue/css/display.css'
/* Theme variables */
import './theme/variables.css'
import './theme/ionic-override.scss'
-import { store, storeKey } from './stores'
+import './theme/styles.scss'
+
+import { createPinia } from 'pinia'
+import piniaPersist from 'pinia-plugin-persist'
+import { piniaInitLoader } from './utils/store'
+
+const pinia = createPinia()
+
+pinia.use(piniaInitLoader)
+pinia.use(piniaPersist)
const app = createApp(App)
- .use(store, storeKey)
+ .use(pinia)
.use(IonicVue, {
mode: 'ios'
})
diff --git a/src/pinia.d.ts b/src/pinia.d.ts
new file mode 100644
index 0000000..d19d523
--- /dev/null
+++ b/src/pinia.d.ts
@@ -0,0 +1,7 @@
+import 'pinia'
+
+declare module 'pinia' {
+ export interface PiniaCustomProperties {
+ with(): Promise
+ }
+}
\ No newline at end of file
diff --git a/src/request/toolkit-api.ts b/src/request/toolkit-api.ts
index 419864d..1fdf4d4 100644
--- a/src/request/toolkit-api.ts
+++ b/src/request/toolkit-api.ts
@@ -1,3 +1,4 @@
+import { useUserStore } from '@/stores';
import { showError } from '@/utils/dialog';
import { ServerError } from '@/utils/error';
import axios, { AxiosResponse } from 'axios';
@@ -11,7 +12,8 @@ export const toolkitApi = axios.create({
})
toolkitApi.interceptors.request.use((config) => {
- const token = sessionStorage.getItem('aiassistant_token')
+ const userStore = useUserStore()
+ const token = userStore.token
if (token) {
config.headers.Authorization = `Bearer ${token}`
}
@@ -30,20 +32,29 @@ toolkitApi.interceptors.response.use((response) => {
}, async (error) => {
if (error.response) {
let response: AxiosResponse = error.response
+ const userStore = useUserStore()
// 处理登录相关错误
if (response.status === 401) {
- if ((window as any)._loggedIn) {
+ if (userStore.loggedIn) {
showError('登录信息不存在，请通过百科页面上的链接重新进入此页面。')
} else {
- throw new ServerError(response.data.error ?? error.message, response)
+ throw new ServerError(response.data.error?.message ?? error.message, response, response.data.error)
}
}
// 处理其他服务器报告的错误
- if (response.data && 'status' in response.data) {
+ if (response.data?.status) {
if (response.data.status < 0 && response.data.error) {
- showError(response.data.error)
- throw new ServerError(response.data.error, response)
+ showError(response.data.error?.message ?? '服务器错误');
+ throw new ServerError(response.data.error?.message, response, response.data.error)
}
+ } else if (response.status >= 500) {
+ showError(`服务器错误：${response.status} ${response.statusText}`)
+ throw new ServerError(error.message, response, error)
+ } else if (response.status >= 400) {
+ showError(`服务器返回错误：${response.status} ${response.statusText}`)
+ throw new ServerError(error.message, response, error)
+ } else if (response.headers['Content-Type'] !== "application/json") {
+ showError('服务器返回数据格式错误。')
}
}
showError('无法连接到服务器：' + error.message)
diff --git a/src/router/index.ts b/src/router/index.ts
index eb75923..df2426a 100644
--- a/src/router/index.ts
+++ b/src/router/index.ts
@@ -1,25 +1,34 @@
-import { createRouter, createWebHistory } from '@ionic/vue-router';
-import { RouteRecordRaw } from 'vue-router';
+import { createRouter, createWebHistory } from '@ionic/vue-router'
+import { RouteRecordRaw } from 'vue-router'
const routes: Array = [
{
path: '',
- component: () => import('@/views/Index.vue')
+ // component: () => import('@/views/Index.vue')
+ redirect: '/toolbox'
},
+ // {
+ // path: '/login',
+ // component: () => import('@/views/Login.vue')
+ // },
{
- path: '/login',
- component: () => import('@/views/Login.vue')
- },
- {
- path: '/toolkit/:title',
- component: () => import('@/layouts/ToolkitLayout.vue'),
+ path: '/toolbox',
+ component: () => import('@/layouts/ToolboxLayout.vue'),
+ name: 'toolbox',
children: [
{
path: '',
- component: () => import('@/views/ToolkitIndex.vue')
+ name: 'toolbox_index',
+ component: () => import('@/views/ToolboxIndex.vue')
+ },
+ {
+ path: 'chatcomplete/:id',
+ name: 'toolbox_chatcomplete',
+ component: () => import('@/views/ChatComplete.vue')
},
{
path: 'chatcomplete',
+ name: 'toolbox_chatcomplete_new',
component: () => import('@/views/ChatComplete.vue')
}
]
diff --git a/src/stores/conversation.ts b/src/stores/conversation.ts
new file mode 100644
index 0000000..26cb68e
--- /dev/null
+++ b/src/stores/conversation.ts
@@ -0,0 +1,118 @@
+import { api } from "@/api";
+import { ConversationInfo } from "@/types/base";
+import { defineStore } from "pinia";
+import { usePageStore } from "./page";
+import { useUserStore } from ".";
+
+export type ConversationState = {
+ list: ConversationInfo[],
+ currentId: number,
+ isLoaded: boolean
+}
+
+export const useConversationStore = defineStore('conversation', {
+ state: (): ConversationState => ({
+ list: [],
+ currentId: -1,
+ isLoaded: false,
+ }),
+ getters: {
+ current: (state): ConversationInfo | undefined => {
+ if (state.currentId >= 0) {
+ return state.list.find((item) => item.id === state.currentId)
+ } else {
+ return undefined
+ }
+ },
+ },
+ actions: {
+ newConversation(module: string, title: string) {
+ this.list.push({
+ id: 0,
+ title,
+ module,
+ pinned: false,
+ updated_at: new Date().getTime(),
+ extra: {}
+ })
+ },
+ updateById(id: number, data: Partial) {
+ let currentItem = this.list.find((item) => item.id === id);
+ if (currentItem) {
+ Object.assign(currentItem, data);
+ }
+ },
+ async load() {
+ const pageStore = usePageStore()
+ const userStore = useUserStore()
+
+ if (!userStore.loggedIn) {
+ this.list = []
+ this.isLoaded = true
+ return
+ }
+
+ let queryParams: any = {}
+
+ /* 暂时不使用当前页面区分会话
+ if (pageStore.title) {
+ queryParams.title = pageStore.title
+ }
+ */
+
+ const res = await api.base.getConversationList(queryParams)
+ this.list = res
+ this.isLoaded = true
+ },
+ async editTitle(id: number, newTitle: string) {
+ await api.base.setConversationTitle({
+ id,
+ new_title: newTitle
+ })
+ this.updateById(id, {
+ title: newTitle,
+ updated_at: Math.ceil(Date.now() / 1000)
+ })
+ this.sort()
+ },
+ async remove(id: number | number[]) {
+ if (Array.isArray(id)) {
+ await api.base.removeConversation({
+ ids: id
+ })
+ this.list = this.list.filter((item) => !id.includes(item.id))
+ } else {
+ await api.base.removeConversation({
+ id
+ })
+ this.list = this.list.filter((item) => item.id !== id)
+ }
+ // 重置当前选中
+ if (!this.list.find((item) => item.id === this.currentId)) {
+ this.currentId = -1
+ }
+ },
+ async pin(id: number, pinned: boolean) {
+ await api.base.setConversationPinned({
+ id,
+ pinned
+ })
+ this.updateById(id, {
+ pinned,
+ updated_at: Math.ceil(Date.now() / 1000)
+ })
+ this.sort()
+ },
+ sort() {
+ this.list.sort((a, b) => {
+ if (a.pinned && b.pinned || !a.pinned && !b.pinned) { // 置顶或不置顶对话内部，按更新时间排序
+ return b.updated_at - a.updated_at
+ } else if (a.pinned) { // 将置顶的会话排在前面
+ return -1
+ } else {
+ return 1
+ }
+ })
+ }
+ },
+});
\ No newline at end of file
diff --git a/src/stores/guide.ts b/src/stores/guide.ts
new file mode 100644
index 0000000..ec14f66
--- /dev/null
+++ b/src/stores/guide.ts
@@ -0,0 +1,20 @@
+import { defineStore } from "pinia"
+
+export type GuideState = {
+ helpForPin: boolean
+}
+
+export const useGuideStore = defineStore('guide', {
+ state: (): GuideState => ({
+ helpForPin: false,
+ }),
+ persist: {
+ enabled: true,
+ strategies: [
+ {
+ key: 'store_guide',
+ storage: localStorage
+ },
+ ],
+ },
+})
\ No newline at end of file
diff --git a/src/stores/index.ts b/src/stores/index.ts
index 4f30298..3e4e546 100644
--- a/src/stores/index.ts
+++ b/src/stores/index.ts
@@ -1,37 +1,5 @@
-import { Store, createStore, useStore as baseUseStore, Module } from 'vuex';
-import { UserState, user } from './user';
-import { PageState, page } from './page';
-
-export interface RootState {
- token: string,
-}
-
-export type State = RootState & {
- user: UserState,
- page: PageState,
-}
-
-export const storeKey: InjectionKey> = Symbol()
-
-export const store = createStore({
- state() {
- return {
- token: ''
- }
- },
- mutations: {
- setToken(state, token) {
- state.token = token
- }
- },
- modules: {
- user,
- page
- }
-})
-
-export const MUTATION_SET_TOKEN = 'setToken'
-
-export function useStore() {
- return baseUseStore(storeKey)
-}
\ No newline at end of file
+export * from './page'
+export * from './user'
+export * from './conversation'
+export * from './guide'
+export * from './settings'
\ No newline at end of file
diff --git a/src/stores/page.ts b/src/stores/page.ts
index c61a8a8..859102d 100644
--- a/src/stores/page.ts
+++ b/src/stores/page.ts
@@ -1,64 +1,75 @@
-import { Module } from "vuex"
-import { RootState, useStore } from "."
import { toolkitApi } from "@/request/toolkit-api"
import { showError } from "@/utils/dialog"
-
-export class PageLoader extends InitLoader {
- public isLoaded() {
- const store = useStore()
- return !!store.state.page.pageId
- }
-
- protected async _load() {
- const store = useStore()
- await store.dispatch(ACTION_LOAD_PAGE_INFO)
- }
-}
+import { defineStore } from "pinia"
+import { LRUCache } from 'lru-cache'
export type PageState = {
title: string,
displayTitle: string | null,
pageId: number | null,
+ isLoaded: boolean,
}
-export const page: Module = {
- state: {
+const pageStoreCache = new LRUCache({
+ max: 100,
+ ttl: 1000 * 60 * 10, // 10 minutes
+})
+
+export const usePageStore = defineStore('page', {
+ state: (): PageState => ({
title: '',
displayTitle: null,
pageId: null,
- },
+
+ isLoaded: false,
+ }),
getters: {
- pageDisplayTitle(state) {
+ pageDisplayTitle: (state) => {
return state.displayTitle || state.title
- }
- },
- mutations: {
- setPageInfo(state, pageInfo: Partial) {
- if (pageInfo.title) state.title = pageInfo.title
- if (pageInfo.displayTitle !== undefined) state.displayTitle = pageInfo.displayTitle
- if (pageInfo.pageId !== undefined) state.pageId = pageInfo.pageId
},
+ collectionTitle: (state) => {
+ let sepPos = state.title.indexOf('/')
+ if (sepPos === -1) {
+ return state.title
+ }
+ return state.title.substring(0, sepPos)
+ }
},
actions: {
- async loadPageInfo({ commit, state }) {
+ async load() {
+ // Load from cache
+ if (pageStoreCache.has(this.title)) {
+ const cache = pageStoreCache.get(this.title)
+ if (cache) {
+ this.title = cache.title
+ this.displayTitle = cache.displayTitle
+ this.pageId = cache.pageId
+ this.isLoaded = cache.isLoaded
+ return
+ }
+ }
+
+ // Load from server
const res = await toolkitApi.get('/title/info', {
params: {
- title: state.title,
+ title: this.title,
}
})
if (res.data.status == 1 && res.data.data) {
const pageInfo = res.data.data
- commit(MUTATION_SET_PAGE_INFO, {
- title: pageInfo.title,
- pageId: pageInfo.page_id,
+ this.pageId = pageInfo.page_id
+ this.title = pageInfo.title
+ this.displayTitle = pageInfo.display_title
+
+ pageStoreCache.set(this.title, {
+ title: this.title,
+ displayTitle: this.displayTitle,
+ pageId: this.pageId,
+ isLoaded: true,
})
} else {
- throw new Error('无法加载页面信息：' + res.data.error)
+ showError('无法加载页面信息：' + res.data.error)
}
- },
+ }
}
-}
-
-export const MUTATION_SET_PAGE_INFO = 'setPageInfo';
-
-export const ACTION_LOAD_PAGE_INFO = 'loadPageInfo';
\ No newline at end of file
+});
\ No newline at end of file
diff --git a/src/stores/settings.ts b/src/stores/settings.ts
new file mode 100644
index 0000000..3f12e41
--- /dev/null
+++ b/src/stores/settings.ts
@@ -0,0 +1,22 @@
+import { defineStore } from "pinia"
+
+export type SettingsState = {
+ docExtractLimit: number,
+ collectionMode: boolean
+}
+
+export const useSettingsStore = defineStore('settings', {
+ state: (): SettingsState => ({
+ docExtractLimit: 10,
+ collectionMode: false,
+ }),
+ persist: {
+ enabled: true,
+ strategies: [
+ {
+ key: 'store_settings',
+ storage: localStorage
+ },
+ ],
+ },
+})
\ No newline at end of file
diff --git a/src/stores/user.ts b/src/stores/user.ts
index 9842252..bc9f25e 100644
--- a/src/stores/user.ts
+++ b/src/stores/user.ts
@@ -1,90 +1,83 @@
-import { Module } from "vuex"
-import { RootState, useStore } from "."
import { decodeJwt } from "@/utils/jwt"
import { toolkitApi } from "@/request/toolkit-api"
-import { showError } from "@/utils/dialog"
-
-export class UserLoader extends InitLoader {
- public isLoaded() {
- const store = useStore()
- return !!store.state.user.uid
- }
-
- protected async _load() {
- const store = useStore()
- await store.dispatch(ACTION_LOAD_USER_INFO)
- }
-}
+import { defineStore } from "pinia"
export type UserState = {
+ token: string,
uid: string,
userName: string,
nickName: string,
avatar: string,
points: number,
- loader: UserLoader,
+ permissions: string[],
+ isLoaded: boolean,
+ lastLoadTime?: number,
}
-export const user: Module = {
- state: {
+export const useUserStore = defineStore('user', {
+ state: (): UserState => ({
+ token: '',
uid: '',
userName: '',
nickName: '',
avatar: '',
points: -2,
- loader: new UserLoader(),
- },
+ permissions: [],
+
+ isLoaded: false,
+ lastLoadTime: undefined,
+ }),
getters: {
- userDisplayName(state) {
+ displayName: (state) => {
return state.nickName || state.userName
},
- userDisplayAvatar(state) {
+ displayAvatar: (state) => {
return state.avatar || import.meta.env.VITE_DEFAULT_AVATAR
- }
- },
- mutations: {
- setUserInfo(state, user: Partial) {
- if (user.uid) state.uid = user.uid
- if (user.userName) state.userName = user.userName
- if (user.nickName) state.nickName = user.nickName
- if (user.avatar) state.avatar = user.avatar
- if (user.points === 0 || user.points) state.points = user.points
},
+ loggedIn: (state) => {
+ return !!state.uid
+ }
},
actions: {
- async loadUserInfoFromToken({ commit }, token) {
+ setPartialUserInfo(user: Partial) {
+ if (user.uid) this.uid = user.uid
+ if (user.userName) this.userName = user.userName
+ if (user.nickName) this.nickName = user.nickName
+ if (user.avatar) this.avatar = user.avatar
+ if (user.points === 0 || user.points) this.points = user.points
+ },
+ async loadUserInfoFromToken(token: string) {
const { payload } = decodeJwt(token)
- commit(MUTATION_SET_USER_INFO, {
- uid: payload.sub,
- userName: payload.username,
- nickName: payload.nickname,
- })
+ this.token = token
+ this.uid = payload.sub
+ this.userName = payload.username
+ this.nickName = payload.nickname
+
+ // add default permissions
+ this.permissions = ['chatcomplete', 'imagen']
},
- async loadUserInfo({ commit, state }) {
+ async load() {
const res = await toolkitApi.get('/user/info')
if (res.data.status == 1 && res.data.data) {
const userInfo = res.data.data
- commit(MUTATION_SET_USER_INFO, {
- uid: userInfo.userid,
- userName: userInfo.username,
- nickName: userInfo.realname,
- avatar: userInfo.avatar,
- points: userInfo.points,
- });
-
+ this.uid = userInfo.userid
+ this.userName = userInfo.username
+ this.nickName = userInfo.realname
+ this.avatar = userInfo.avatar
+ this.points = userInfo.points
+
+ this.lastLoadTime = new Date().getTime();
+
(window as any)._loggedIn = true
} else {
throw new Error('无法加载用户信息：' + res.data.error)
}
},
- async loadUserPoints({ commit, state }) {
- // TODO: Load user points from server
+ async refresh() {
+ if (this.lastLoadTime && new Date().getTime() - this.lastLoadTime < 60 * 10 * 1000) {
+ return
+ }
+ await this.load()
}
- }
-}
-
-export const MUTATION_SET_USER_INFO = 'setUserInfo';
-
-export const ACTION_LOAD_USER_INFO_FROM_TOKEN = 'loadUserInfoFromToken';
-export const ACTION_LOAD_USER_INFO = 'loadUserInfo';
-export const ACTION_LOAD_USER_POINTS = 'loadUserPoints';
\ No newline at end of file
+ },
+});
\ No newline at end of file
diff --git a/src/theme/ionic-override.scss b/src/theme/ionic-override.scss
index 1dfd2ab..066751a 100644
--- a/src/theme/ionic-override.scss
+++ b/src/theme/ionic-override.scss
@@ -8,4 +8,34 @@ ion-toolbar {
--padding-start: 15px;
--padding-end: 15px
}
+}
+
+ion-item {
+ --background-hover-opacity: .02;
+}
+
+.header-translucent-ios ion-toolbar,
+.footer-translucent-ios ion-toolbar {
+ --opacity: 1 !important;
+}
+
+ion-button.color-force-light,
+ion-label.color-force-light {
+ --color: var(--ion-color-force-light, #fff);
+}
+
+.color-force-light-dark {
+ --ion-color-dark: #222428;
+ --ion-color-dark-rgb: 34, 36, 40;
+ --ion-color-dark-contrast: #ffffff;
+ --ion-color-dark-contrast-rgb: 255, 255, 255;
+ --ion-color-dark-shade: #1e2023;
+ --ion-color-dark-tint: #383a3e;
+
+ --ion-color-light: #f4f5f8;
+ --ion-color-light-rgb: 244, 245, 248;
+ --ion-color-light-contrast: #000000;
+ --ion-color-light-contrast-rgb: 0, 0, 0;
+ --ion-color-light-shade: #d7d8da;
+ --ion-color-light-tint: #f5f6f9;
}
\ No newline at end of file
diff --git a/src/theme/status-container.scss b/src/theme/status-container.scss
index 7dfcd1c..7217279 100644
--- a/src/theme/status-container.scss
+++ b/src/theme/status-container.scss
@@ -1,4 +1,4 @@
-#status-container {
+.status-container {
text-align: center;
position: absolute;
left: 0;
diff --git a/src/theme/styles.scss b/src/theme/styles.scss
new file mode 100644
index 0000000..04c7838
--- /dev/null
+++ b/src/theme/styles.scss
@@ -0,0 +1,29 @@
+.text-selectable {
+ -webkit-user-select: text;
+ -moz-user-select: text;
+ user-select: text;
+
+ -webkit-touch-callout: default;
+}
+
+.help-block {
+ margin-left: 0;
+ margin-right: 0;
+ margin-top: 0;
+ margin-bottom: 8px;
+ font-size: 14px;
+ line-height: normal;
+ text-overflow: inherit;
+ overflow: inherit;
+ color: rgba(var(--ion-text-color-rgb, 0, 0, 0), 0.4);
+}
+
+.list-help-block {
+ margin-left: 20px;
+ margin-right: 20px;
+ margin-bottom: 16px;
+}
+
+.list-ios.list-inset.with-help-block {
+ margin-bottom: 8px;
+}
\ No newline at end of file
diff --git a/src/theme/variables.css b/src/theme/variables.css
index 3797680..8533ff5 100644
--- a/src/theme/variables.css
+++ b/src/theme/variables.css
@@ -65,6 +65,14 @@ http://ionicframework.com/docs/theming/ */
--ion-color-dark-shade: #1e2023;
--ion-color-dark-tint: #383a3e;
+ /** force dark **/
+ --ion-color-force-dark: #222428;
+ --ion-color-force-dark-rgb: 34, 36, 40;
+ --ion-color-force-dark-contrast: #ffffff;
+ --ion-color-force-dark-contrast-rgb: 255, 255, 255;
+ --ion-color-force-dark-shade: #1e2023;
+ --ion-color-force-dark-tint: #383a3e;
+
/** medium **/
--ion-color-medium: #92949c;
--ion-color-medium-rgb: 146, 148, 156;
@@ -80,6 +88,14 @@ http://ionicframework.com/docs/theming/ */
--ion-color-light-contrast-rgb: 0, 0, 0;
--ion-color-light-shade: #d7d8da;
--ion-color-light-tint: #f5f6f9;
+
+ /** force light **/
+ --ion-color-force-light: #f4f5f8;
+ --ion-color-force-light-rgb: 244, 245, 248;
+ --ion-color-force-light-contrast: #000000;
+ --ion-color-force-light-contrast-rgb: 0, 0, 0;
+ --ion-color-force-light-shade: #d7d8da;
+ --ion-color-force-light-tint: #f5f6f9;
}
@media (prefers-color-scheme: dark) {
diff --git a/src/types/base.ts b/src/types/base.ts
new file mode 100644
index 0000000..d6795ea
--- /dev/null
+++ b/src/types/base.ts
@@ -0,0 +1,41 @@
+export type ConversationInfo = {
+ id: number,
+ module: string,
+ page_title?: string,
+ title?: string,
+ thumbnail?: string,
+ description?: string,
+ rev_id?: number,
+ updated_at: number,
+ pinned: boolean,
+ extra: any,
+ new_conversation?: boolean,
+}
+
+export type ExtractDocInfo = {
+ id: number,
+ sha1: string,
+ text: string,
+ text_len: number,
+ markdown?: string,
+ markdown_len?: string,
+ distance: number,
+}
+
+export type ChatCompleteMessage = {
+ id?: string,
+ role: string,
+ content: string,
+ tokens: number,
+ time?: number,
+ streaming?: boolean,
+}
+
+export type ChatCompleteChunkInfo = ChatCompleteMessage[]
+
+export type TypeaheadItem = {
+ key: any;
+ title: string;
+ subtitle?: string;
+ value: any;
+};
\ No newline at end of file
diff --git a/src/types/enum.ts b/src/types/enum.ts
new file mode 100644
index 0000000..096f78c
--- /dev/null
+++ b/src/types/enum.ts
@@ -0,0 +1,7 @@
+export const moduleConversationRoute: Record = {
+ chatcomplete: 'toolbox_chatcomplete',
+}
+
+export const moduleNewConversationRoute: Record = {
+ chatcomplete: 'toolbox_chatcomplete_new',
+}
\ No newline at end of file
diff --git a/src/types/instance.ts b/src/types/instance.ts
new file mode 100644
index 0000000..dc39153
--- /dev/null
+++ b/src/types/instance.ts
@@ -0,0 +1,3 @@
+import { IonRouterOutlet } from "@ionic/vue";
+
+export type IonRouterOutletInstance = InstanceType
\ No newline at end of file
diff --git a/src/utils/actions.ts b/src/utils/actions.ts
new file mode 100644
index 0000000..a93c63a
--- /dev/null
+++ b/src/utils/actions.ts
@@ -0,0 +1,20 @@
+import { useConversationStore } from "@/stores"
+import { loadingController } from "@ionic/vue"
+
+export const pinConversation = async (conversationId: number, pinned: boolean) => {
+let actionName = pinned ? '收藏' : '取消收藏'
+let loading = await loadingController.create({
+ message: `正在${actionName}...`,
+ translucent: true,
+ backdropDismiss: false
+})
+await loading.present()
+try {
+ const conversationStore = useConversationStore()
+ await conversationStore.pin(conversationId, pinned)
+} catch (err) {
+ console.error(`${actionName}对话错误`, err)
+} finally {
+ await loading.dismiss()
+}
+}
\ No newline at end of file
diff --git a/src/utils/dialog.ts b/src/utils/dialog.ts
index a0115af..8a200c0 100644
--- a/src/utils/dialog.ts
+++ b/src/utils/dialog.ts
@@ -1,4 +1,4 @@
-import { alertController } from "@ionic/vue"
+import { alertController, toastController } from "@ionic/vue"
export async function showError(message: string, title?: string) {
const alert = await alertController.create({
@@ -8,4 +8,93 @@ export async function showError(message: string, title?: string) {
})
await alert.present()
+}
+
+export type ShowConfirmOptions = {
+ title?: string,
+ danger?: boolean,
+}
+
+export function showConfirm(message: string, options: ShowConfirmOptions = {}): Promise {
+ return new Promise(async (resolve, reject) => {
+ const alert = await alertController.create({
+ header: options.title ?? '提示',
+ message: message,
+ translucent: true,
+ buttons: [
+ {
+ text: '取消',
+ role: 'cancel',
+ handler: () => {
+ resolve(false)
+ }
+ },
+ {
+ text: '确定',
+ role: options.danger ? 'destructive' : 'confirm',
+ handler: () => {
+ resolve(true)
+ }
+ }
+ ],
+ })
+
+ await alert.present()
+ await alert.onDidDismiss()
+ resolve(false)
+ })
+}
+
+export function showRenameDialog(oldName: string): Promise {
+ return new Promise(async (resolve, reject) => {
+ const alert = await alertController.create({
+ header: '重命名',
+ translucent: true,
+ inputs: [
+ {
+ name: 'name',
+ type: 'text',
+ value: oldName,
+ placeholder: '请输入新的名称'
+ }
+ ],
+ buttons: [
+ {
+ text: '取消',
+ role: 'cancel',
+ handler: () => {
+ resolve(undefined)
+ }
+ },
+ {
+ text: '确定',
+ role: 'confirm',
+ handler: (data) => {
+ resolve(data.name)
+ }
+ }
+ ],
+ })
+
+ await alert.present()
+ await alert.onDidDismiss()
+ resolve(undefined)
+ })
+}
+
+export async function showHelpToast(message: string) {
+ const toast = await toastController.create({
+ message,
+ duration: 10000,
+ position: 'top',
+ translucent: true,
+ buttons: [
+ {
+ text: '关闭',
+ role: 'cancel',
+ }
+ ]
+ })
+
+ await toast.present()
}
\ No newline at end of file
diff --git a/src/utils/error.ts b/src/utils/error.ts
index f195339..9106b0f 100644
--- a/src/utils/error.ts
+++ b/src/utils/error.ts
@@ -1,5 +1,5 @@
export class ServerError extends Error {
- constructor(message: string, public response: any) {
+ constructor(message: string, public response: any, public detail: any = {}) {
super(message);
this.name = 'ServerError';
}
diff --git a/src/utils/init-loader.ts b/src/utils/init-loader.ts
deleted file mode 100644
index 1bfddbd..0000000
--- a/src/utils/init-loader.ts
+++ /dev/null
@@ -1,39 +0,0 @@
-class InitLoader {
- private loading = false;
- private listeners: CallableFunction[][] = [];
-
- public isLoaded() {
- return true;
- }
-
- public async load() {
- try {
- this.loading = true;
- await this._load();
- this.loading = false;
-
- this.listeners.forEach(listener => listener[0]());
- } catch (e) {
- this.listeners.forEach(listener => listener[1](e));
- }
- }
-
- public reset() { }
-
- public use(): Promise {
- return new Promise((resolve, reject) => {
- if (this.isLoaded()) {
- resolve();
- } else {
- if (!this.loading) {
- this.load();
- }
- this.listeners.push([resolve, reject]);
- }
- });
- }
-
- protected async _load() {
-
- }
-}
\ No newline at end of file
diff --git a/src/utils/other.ts b/src/utils/other.ts
new file mode 100644
index 0000000..7ca12f0
--- /dev/null
+++ b/src/utils/other.ts
@@ -0,0 +1,8 @@
+export function setPageTitle(...title: string[]) {
+ let newTitle = title.join(' - ')
+ if (newTitle === '') {
+ document.title = import.meta.env.VITE_APP_NAME
+ } else {
+ document.title = newTitle + ' - ' + import.meta.env.VITE_APP_NAME
+ }
+}
\ No newline at end of file
diff --git a/src/utils/request.ts b/src/utils/request.ts
new file mode 100644
index 0000000..9d6e220
--- /dev/null
+++ b/src/utils/request.ts
@@ -0,0 +1,22 @@
+import { useUserStore } from "@/stores"
+
+export function getWebSocketApiUrl(path: string, query: Record = {}) {
+ const userStore = useUserStore()
+ let token = userStore.token
+ let baseUrl = import.meta.env.VITE_TOOLKIT_WS_API_ENDPOINT
+ let url = new URL(baseUrl, location.href)
+
+ if (url.protocol === 'https:') {
+ url.protocol = 'wss:'
+ } else {
+ url.protocol = 'ws:'
+ }
+
+ url.pathname += path
+ url.searchParams.set('token', token)
+ for (let key in query) {
+ url.searchParams.set(key, query[key])
+ }
+
+ return url.href
+}
\ No newline at end of file
diff --git a/src/utils/store.ts b/src/utils/store.ts
new file mode 100644
index 0000000..6ed883e
--- /dev/null
+++ b/src/utils/store.ts
@@ -0,0 +1,69 @@
+import { useConversationStore } from "@/stores";
+import { PiniaPluginContext, Store } from "pinia";
+
+export class InitLoaderError extends Error {
+ constructor(message: string, public readonly srcError: Error, public readonly isFirst = false) {
+ super(message);
+ this.name = 'InitLoaderError';
+ }
+}
+
+export function piniaInitLoader(context: PiniaPluginContext) {
+ let loading = false;
+ let listeners: CallableFunction[][] = [];
+
+ const loadStore = async () => {
+ try {
+ loading = true;
+ await context.store.load();
+ loading = false;
+ listeners.forEach((listener) => listener[0]());
+ } catch (e: any) {
+ loading = false;
+ listeners.forEach((listener, index) => {
+ const err = new InitLoaderError('Load store failed', e, index === 0);
+ listener[1](err);
+ });
+ }
+ }
+
+ return {
+ with: () => {
+ return new Promise((resolve, reject) => {
+ if ('isLoaded' in context.store && 'load' in context.store) {
+ if (context.store.isLoaded) {
+ resolve();
+ } else {
+ if (!loading) {
+ loadStore();
+ }
+ listeners.push([resolve, reject]);
+ }
+ } else {
+ reject(new Error('This store is not support init loader'));
+ }
+ });
+ }
+ }
+}
+
+/**
+ * Load all stores
+ * @param stores Pinina stores
+ * @returns
+ */
+export function withStore(...stores: Store[]) {
+ return Promise.all(stores.map((store) => store.with()));
+}
+
+export function getRecommandTitles() {
+ const conversationStore = useConversationStore()
+ let recommandTitles: string[] = []
+ conversationStore.list.forEach((item) => {
+ if (item.page_title && !recommandTitles.includes(item.page_title)) {
+ recommandTitles.push(item.page_title)
+ }
+ })
+
+ return recommandTitles
+}
\ No newline at end of file
diff --git a/src/views/ChatComplete.vue b/src/views/ChatComplete.vue
index 6595525..a858a01 100644
--- a/src/views/ChatComplete.vue
+++ b/src/views/ChatComplete.vue
@@ -1,103 +1,719 @@
@@ -107,29 +723,48 @@ _This is italic text_
-
-
- 未命名对话
+
+
+ {{ state.conversationTitle || '无标题' }}
+
+
+
+
+
+
+
+
+ {{ state.conversationTitle || '无标题' }}
+
+
-
-
{{ item.title }}
+{{ item.subtitle }}
+{{ p.page_title || '' }}
+
+
+ {{ p.title || '无标题' }}
+{{ p.description || ' ' }}
+
+
+
+
+ {{ p.title }}
-{{ p.extract }}
-
-
+
+
+
+
@@ -137,20 +772,35 @@ _This is italic text_
+
+
-
+