update pay function
This commit is contained in:
93
boilerplate-chakra-pro-main/utils/streams/essayStream.ts
Normal file
93
boilerplate-chakra-pro-main/utils/streams/essayStream.ts
Normal file
@@ -0,0 +1,93 @@
|
||||
import endent from 'endent';
|
||||
import {
|
||||
createParser,
|
||||
ParsedEvent,
|
||||
ReconnectInterval,
|
||||
} from 'eventsource-parser';
|
||||
|
||||
const createPrompt = (topic: string, words: string, essayType: string) => {
|
||||
const data = (topic: any, words: string, essayType: string) => {
|
||||
return endent`
|
||||
You are an expert formal essay writer and generator.
|
||||
You know very well all types of essays. Generate an formal ${essayType} essay about ${topic}, which has a number of maximum ${words} words.
|
||||
The generated content should NOT be longer than ${words} words.
|
||||
The essay must be in markdown format but not rendered, it must include all markdown characteristics. The title must be bold, and there should be a between every paragraph.
|
||||
Do not include informations about console logs or print messages.
|
||||
`;
|
||||
};
|
||||
|
||||
if (essayType) {
|
||||
return data(topic, words, essayType);
|
||||
}
|
||||
};
|
||||
|
||||
export const OpenAIStream = async (
|
||||
topic: string,
|
||||
essayType: string,
|
||||
words: string,
|
||||
model: string,
|
||||
key: string | undefined,
|
||||
) => {
|
||||
const prompt = createPrompt(topic, words, essayType);
|
||||
|
||||
const system = { role: 'system', content: prompt };
|
||||
|
||||
const res = await fetch(`https://api.openai.com/v1/chat/completions`, {
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
Authorization: `Bearer ${key || process.env.NEXT_PUBLIC_OPENAI_API_KEY}`,
|
||||
},
|
||||
method: 'POST',
|
||||
body: JSON.stringify({
|
||||
model,
|
||||
messages: [system],
|
||||
temperature: 0,
|
||||
stream: true,
|
||||
}),
|
||||
});
|
||||
|
||||
const encoder = new TextEncoder();
|
||||
const decoder = new TextDecoder();
|
||||
|
||||
if (res.status !== 200) {
|
||||
const statusText = res.statusText;
|
||||
const result = await res.body?.getReader().read();
|
||||
throw new Error(
|
||||
`OpenAI API returned an error: ${
|
||||
decoder.decode(result?.value) || statusText
|
||||
}`,
|
||||
);
|
||||
}
|
||||
|
||||
const stream = new ReadableStream({
|
||||
async start(controller) {
|
||||
const onParse = (event: ParsedEvent | ReconnectInterval) => {
|
||||
if (event.type === 'event') {
|
||||
const data = event.data;
|
||||
|
||||
if (data === '[DONE]') {
|
||||
controller.close();
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
const json = JSON.parse(data);
|
||||
const text = json.choices[0].delta.content;
|
||||
const queue = encoder.encode(text);
|
||||
controller.enqueue(queue);
|
||||
} catch (e) {
|
||||
controller.error(e);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const parser = createParser(onParse);
|
||||
|
||||
for await (const chunk of res.body as any) {
|
||||
parser.feed(decoder.decode(chunk));
|
||||
}
|
||||
},
|
||||
});
|
||||
|
||||
return stream;
|
||||
};
|
||||
Reference in New Issue
Block a user