diff --git a/src/api/ai/writer/index.ts b/src/api/ai/write/index.ts similarity index 36% rename from src/api/ai/writer/index.ts rename to src/api/ai/write/index.ts index 7494b2595135238cfa4a7c095db1980653235999..0b4b18a440b578a6c875f087c5f8cbfd55c7a361 100644 --- a/src/api/ai/writer/index.ts +++ b/src/api/ai/write/index.ts @@ -1,9 +1,9 @@ import { fetchEventSource } from '@microsoft/fetch-event-source' -import request from '@/config/axios' import { getAccessToken } from '@/utils/auth' import { config } from '@/config/axios/config' import { AiWriteTypeEnum } from '@/views/ai/utils/constants' +import request from '@/config/axios' export interface WriteVO { type: AiWriteTypeEnum.WRITING | AiWriteTypeEnum.REPLY // 1:撰写 2:回复 @@ -21,45 +21,65 @@ export interface WriteVO { createTime?: Date // 创建时间 } -// TODO @hhero:搞成 WriteApi,类似 ConversationApi 一样。这样更有类的概念,后续引入某个 Api,然后调用它的方法就可以了。 -export const writeStream = ({ - data, - onClose, - onMessage, - onError, - ctrl -}: { - data: WriteVO - onMessage?: (res: any) => void - onError?: (...args: any[]) => void - onClose?: (...args: any[]) => void - ctrl: AbortController -}) => { - const token = getAccessToken() - return fetchEventSource(`${config.base_url}/ai/write/generate-stream`, { - method: 'post', - headers: { - 'Content-Type': 'application/json', - Authorization: `Bearer ${token}` - }, - openWhenHidden: true, - body: JSON.stringify(data), - onmessage: onMessage, - onerror: onError, - onclose: onClose, - signal: ctrl.signal - }) +export interface AiWritePageReqVO extends PageParam { + userId?: number // 用户编号 + type?: AiWriteTypeEnum // 写作类型 + platform?: string // 平台 + createTime?: [string, string] // 创建时间 +} + +export interface AiWriteRespVo { + id: number + userId: number + type: number + platform: string + model: string + prompt: string + generatedContent: string + originalContent: string + length: number + format: number + tone: number + language: number + errorMessage: string + createTime: string } -// AI 写作 API export const WriteApi = { - // 查询AI 写作分页 - getWritePage: async (params: any) => { - return await request.get({ url: `/ai/write/page`, params }) + writeStream: ({ + data, + onClose, + onMessage, + onError, + ctrl + }: { + data: WriteVO + onMessage?: (res: any) => void + onError?: (...args: any[]) => void + onClose?: (...args: any[]) => void + ctrl: AbortController + }) => { + const token = getAccessToken() + return fetchEventSource(`${config.base_url}/ai/write/generate-stream`, { + method: 'post', + headers: { + 'Content-Type': 'application/json', + Authorization: `Bearer ${token}` + }, + openWhenHidden: true, + body: JSON.stringify(data), + onmessage: onMessage, + onerror: onError, + onclose: onClose, + signal: ctrl.signal + }) }, - - // 删除AI 写作 - deleteWrite: async (id: number) => { - return await request.delete({ url: `/ai/write/delete?id=` + id }) + // 获取写作列表 + getWritePage: (params: AiWritePageReqVO) => { + return request.get>({ url: `/ai/write/page`, params }) + }, + // 删除写作 + deleteWrite(id: number) { + return request.delete({ url: `/ai/write/delete`, params: { id } }) } } diff --git a/src/views/ai/utils/constants.ts b/src/views/ai/utils/constants.ts index 483f782ef5cf9137138e993fd9ccb5f26c53efd3..4fee2dc60c24752d616a9de0a616166aaf1587be 100644 --- a/src/views/ai/utils/constants.ts +++ b/src/views/ai/utils/constants.ts @@ -64,6 +64,12 @@ export enum AiWriteTypeEnum { REPLY // 回复 } +// 表格展示对照map +export const AiWriteTypeTableRender = { + [AiWriteTypeEnum.WRITING]: '撰写', + [AiWriteTypeEnum.REPLY]: '回复', +} + // ========== 【图片 UI】相关的枚举 ========== export const ImageHotWords = [ '中国旗袍', diff --git a/src/views/ai/writer/index/components/Left.vue b/src/views/ai/write/index/components/Left.vue similarity index 81% rename from src/views/ai/writer/index/components/Left.vue rename to src/views/ai/write/index/components/Left.vue index 7e7744c7e1163dab58675b23f82f11f448f0c054..05cc04a5de12f9e8d41c73051ecf812625ae218f 100644 --- a/src/views/ai/writer/index/components/Left.vue +++ b/src/views/ai/write/index/components/Left.vue @@ -24,26 +24,28 @@ -
+
-
-
- +
+
+
+ +