schihei/chatbot-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
pages/api/chat.ts
+9 -9
View File
@@ -1,13 +1,13 @@
import { ChatBody, Message, OpenAIModelID } from "@/types";
import { DEFAULT_SYSTEM_PROMPT } from "@/utils/app/const";
import { OpenAIStream } from "@/utils/server";
import tiktokenModel from "@dqbd/tiktoken/encoders/cl100k_base.json";
import { init, Tiktoken } from "@dqbd/tiktoken/lite/init";
import { ChatBody, Message, OpenAIModelID } from '@/types';
import { DEFAULT_SYSTEM_PROMPT } from '@/utils/app/const';
import { OpenAIStream } from '@/utils/server';
import tiktokenModel from '@dqbd/tiktoken/encoders/cl100k_base.json';
import { init, Tiktoken } from '@dqbd/tiktoken/lite/init';
// @ts-expect-error
import wasm from "../../node_modules/@dqbd/tiktoken/lite/tiktoken_bg.wasm?module";
import wasm from '../../node_modules/@dqbd/tiktoken/lite/tiktoken_bg.wasm?module';
export const config = {
runtime: "edge",
runtime: 'edge',
};
const handler = async (req: Request): Promise<Response> => {
@@ -18,7 +18,7 @@ const handler = async (req: Request): Promise<Response> => {
const encoding = new Tiktoken(
tiktokenModel.bpe_ranks,
tiktokenModel.special_tokens,
tiktokenModel.pat_str
tiktokenModel.pat_str,
);
const tokenLimit = model.id === OpenAIModelID.GPT_4 ? 6000 : 3000;
@@ -51,7 +51,7 @@ const handler = async (req: Request): Promise<Response> => {
return new Response(stream);
} catch (error) {
console.error(error);
return new Response("Error", { status: 500 });
return new Response('Error', { status: 500 });
}
};
pages/api/models.ts
+18 -17
View File
@@ -1,8 +1,8 @@
import { OpenAIModel, OpenAIModelID, OpenAIModels } from "@/types";
import { OPENAI_API_HOST } from "@/utils/app/const";
import { OpenAIModel, OpenAIModelID, OpenAIModels } from '@/types';
import { OPENAI_API_HOST } from '@/utils/app/const';
export const config = {
runtime: "edge"
runtime: 'edge',
};
const handler = async (req: Request): Promise<Response> => {
@@ -13,22 +13,23 @@ const handler = async (req: Request): Promise<Response> => {
const response = await fetch(`${OPENAI_API_HOST}/v1/models`, {
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}`,
},
});
if (response.status === 401) {
return new Response(
response.body,
{
status: 500,
headers: response.headers
}
);
return new Response(response.body, {
status: 500,
headers: response.headers,
});
} else if (response.status !== 200) {
console.error(`OpenAI API returned an error ${response.status}: ${await response.text()}`)
throw new Error("OpenAI API returned an error");
console.error(
`OpenAI API returned an error ${
response.status
}: ${await response.text()}`,
);
throw new Error('OpenAI API returned an error');
}
const json = await response.json();
@@ -39,7 +40,7 @@ const handler = async (req: Request): Promise<Response> => {
if (value === model.id) {
return {
id: model.id,
name: OpenAIModels[value].name
name: OpenAIModels[value].name,
};
}
}
@@ -49,7 +50,7 @@ const handler = async (req: Request): Promise<Response> => {
return new Response(JSON.stringify(models), { status: 200 });
} catch (error) {
console.error(error);
return new Response("Error", { status: 500 });
return new Response('Error', { status: 500 });
}
};