Browse Source

change db

master
godo 6 months ago
parent
commit
e3b2b67812
  1. 4
      frontend/src/components/window/IframeFile.vue
  2. 5
      frontend/src/hook/useAi.ts
  3. 7
      frontend/src/stores/db.ts

4
frontend/src/components/window/IframeFile.vue

@ -199,9 +199,9 @@ const eventHandler = async (e: MessageEvent) => {
if(eventData.category){
postData.category = eventData.category
}
//console.log(postData,eventData.action)
// AI
const res = await askAi(postData, eventData.action);
const res:any = await askAi(postData, eventData.action);
storeRef.value?.contentWindow?.postMessage(
{
type: 'aiReciver',

5
frontend/src/hook/useAi.ts

@ -2,6 +2,7 @@ import { getSystemConfig, fetchPost } from "@/system/config";
import { useAssistantStore } from '@/stores/assistant';
import { useModelStore } from "@/stores/model";
export async function askAi(question: any, action: string) {
try {
const assistantStore = useAssistantStore();
const modelStore = useModelStore();
const config = getSystemConfig();
@ -41,5 +42,9 @@ export async function askAi(question: any, action: string) {
}
const data = await complain.json()
return data.choices[0].message.content
} catch (error) {
return '请求失败'+error
}
}

7
frontend/src/stores/db.ts

@ -1,6 +1,6 @@
import Dexie from 'dexie';
export type ChatTable = 'prompts' | 'modelslabel' | 'modelslist' | 'chatuser' | 'chatmsg' |'systemChatRecord'| 'workbenchChatRecord' | 'workbenchChatUser' | 'workbenchSessionList' | 'groupSessionList' | 'workbenchGroupChatRecord' | 'workbenchGroupUserList' | 'workbenchGroupInviteMessage';
export type ChatTable = 'prompts' | 'modelslabel' | 'modelslist' | 'chatuser' | 'chatmsg' | 'systemChatRecord' | 'workbenchChatRecord' | 'workbenchChatUser' | 'workbenchSessionList' | 'groupSessionList' | 'workbenchGroupChatRecord' | 'workbenchGroupUserList' | 'workbenchGroupInviteMessage';
export const dbInit: any = new Dexie('GodoOSDatabase');
dbInit.version(1).stores({
@ -110,9 +110,14 @@ export const db = {
},
async get(tableName: ChatTable, whereObj: any) {
//console.log(whereObj)
try {
const data = await dbInit[tableName].where(whereObj).first()
//console.log(data)
return data ? data : false
} catch (error) {
return false
}
},
async rows(tableName: ChatTable, whereObj: any) {
return dbInit[tableName].where(whereObj).toArray()

Loading…
Cancel
Save