feat: 暂存添加ai chat and prompt generate

This commit is contained in:
2024-09-26 01:18:04 +08:00
parent 3da62fd254
commit 553e4d62f0
16 changed files with 1265 additions and 60 deletions

View File

@@ -0,0 +1,58 @@
import { query } from '@/modules';
import { message } from 'antd';
import { create } from 'zustand';
export type AiStore = {
open: boolean;
setOpen: (open: boolean) => void;
type?: string;
key: string;
setKey: (key: string) => void;
setType?: (type: string) => void;
sendMsg: (msg: string) => void;
formData: any;
setFormData: (data: any) => void;
runAi: () => any;
title: string;
setTitle: (title: string) => void;
};
export const useAiStore = create<AiStore>((set, get) => {
return {
open: false,
setOpen: (open) => set({ open }),
key: '',
setKey: (key) => set({ key }),
sendMsg: (msg) => {
console.log(msg);
},
formData: {},
setFormData: (data) => set({ formData: data }),
runAi: async () => {
const { formData } = get();
const res = await query.post({
path: 'ai',
key: 'run',
data: {
key: formData.key,
inputs: [
{
key: 'title',
value: '根据描述生成代码',
},
{
key: 'description',
value: '我想获取一个card, 包含标题和内容标题是evision内容是这是一个测试',
},
],
},
});
if (res.code === 200) {
console.log(res.data);
message.success('Success');
}
},
title: '',
setTitle: (title) => set({ title }),
};
});