feat: add in prettier and format code for consistency (#168)

This commit is contained in:
Simon Holmes
2023-03-26 05:13:18 +00:00
committed by GitHub
parent b843f6e0e0
commit d6973b9ccc
72 changed files with 1140 additions and 4573 deletions
+24 -15
View File
@@ -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;