import { module } from '/@/cool';
|
import { useBase } from '/$/base';
|
|
export function useAi() {
|
const { api } = module.config('helper');
|
const { user } = useBase();
|
|
// 调用流程
|
async function invokeFlow(
|
label: string,
|
params: any,
|
streamCb?: ({ isEnd, content }: { isEnd: boolean; content: string }) => void
|
): Promise<any> {
|
const stream = !!streamCb;
|
|
let cacheText = '';
|
|
return new Promise((resolve, reject) => {
|
fetch(api + '/open/code/gen/data', {
|
method: 'POST',
|
headers: {
|
Authorization: user.token,
|
'Content-Type': 'application/json'
|
},
|
body: JSON.stringify({
|
params,
|
label,
|
stream
|
})
|
})
|
.then(res => {
|
if (res.body) {
|
if (stream) {
|
const reader = res.body.getReader();
|
const decoder = new TextDecoder('utf-8');
|
const stream = new ReadableStream({
|
start(controller) {
|
function push() {
|
reader.read().then(({ done, value }) => {
|
if (done) {
|
controller.close();
|
return;
|
}
|
|
let text = decoder.decode(value, { stream: true });
|
|
if (streamCb) {
|
if (cacheText) {
|
text = cacheText + text;
|
}
|
|
if (text.indexOf('data:') == 0) {
|
text = '\n\n' + text;
|
}
|
|
try {
|
const arr = text
|
.split(/\n\ndata:/g)
|
.filter(Boolean)
|
.map(e => JSON.parse(e));
|
|
arr.forEach(streamCb);
|
|
cacheText = '';
|
} catch (err) {
|
cacheText = text;
|
}
|
}
|
|
controller.enqueue(text);
|
push();
|
});
|
}
|
push();
|
}
|
});
|
|
return new Response(stream);
|
} else {
|
return res.json();
|
}
|
}
|
})
|
.then(res => {
|
if (stream) {
|
return res;
|
}
|
|
if (res.code == 1000) {
|
resolve(res.data.result);
|
} else {
|
reject(res);
|
}
|
})
|
.catch(reject);
|
});
|
}
|
|
return {
|
invokeFlow
|
};
|
}
|
