feat: add in prettier and format code for consistency (#168)
This commit is contained in:
+24
-15
@@ -1,27 +1,36 @@
|
||||
import { Message, OpenAIModel } from "@/types";
|
||||
import { createParser, ParsedEvent, ReconnectInterval } from "eventsource-parser";
|
||||
import { OPENAI_API_HOST } from "../app/const";
|
||||
import { Message, OpenAIModel } from '@/types';
|
||||
import {
|
||||
createParser,
|
||||
ParsedEvent,
|
||||
ReconnectInterval,
|
||||
} from 'eventsource-parser';
|
||||
import { OPENAI_API_HOST } from '../app/const';
|
||||
|
||||
export const OpenAIStream = async (model: OpenAIModel, systemPrompt: string, key: string, messages: Message[]) => {
|
||||
export const OpenAIStream = async (
|
||||
model: OpenAIModel,
|
||||
systemPrompt: string,
|
||||
key: string,
|
||||
messages: Message[],
|
||||
) => {
|
||||
const res = await fetch(`${OPENAI_API_HOST}/v1/chat/completions`, {
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
Authorization: `Bearer ${key ? key : process.env.OPENAI_API_KEY}`
|
||||
'Content-Type': 'application/json',
|
||||
Authorization: `Bearer ${key ? key : process.env.OPENAI_API_KEY}`,
|
||||
},
|
||||
method: "POST",
|
||||
method: 'POST',
|
||||
body: JSON.stringify({
|
||||
model: model.id,
|
||||
messages: [
|
||||
{
|
||||
role: "system",
|
||||
content: systemPrompt
|
||||
role: 'system',
|
||||
content: systemPrompt,
|
||||
},
|
||||
...messages
|
||||
...messages,
|
||||
],
|
||||
max_tokens: 1000,
|
||||
temperature: 1,
|
||||
stream: true
|
||||
})
|
||||
stream: true,
|
||||
}),
|
||||
});
|
||||
|
||||
if (res.status !== 200) {
|
||||
@@ -35,10 +44,10 @@ export const OpenAIStream = async (model: OpenAIModel, systemPrompt: string, key
|
||||
const stream = new ReadableStream({
|
||||
async start(controller) {
|
||||
const onParse = (event: ParsedEvent | ReconnectInterval) => {
|
||||
if (event.type === "event") {
|
||||
if (event.type === 'event') {
|
||||
const data = event.data;
|
||||
|
||||
if (data === "[DONE]") {
|
||||
if (data === '[DONE]') {
|
||||
controller.close();
|
||||
return;
|
||||
}
|
||||
@@ -59,7 +68,7 @@ export const OpenAIStream = async (model: OpenAIModel, systemPrompt: string, key
|
||||
for await (const chunk of res.body as any) {
|
||||
parser.feed(decoder.decode(chunk));
|
||||
}
|
||||
}
|
||||
},
|
||||
});
|
||||
|
||||
return stream;
|
||||
|
||||
Reference in New Issue
Block a user