xiao
初始化
6a37520
raw
history blame
1.97 kB
import { createParser } from "eventsource-parser";
import { NextRequest } from "next/server";
async function createStream(req: NextRequest) {
const encoder = new TextEncoder();
const decoder = new TextDecoder();
const res = await fetch(
"http://lemurchat.anfans.cn/api/chat/conversation-trial",
{
headers: {
"Content-Type": "application/json",
},
method: "POST",
body: req.body,
},
);
const stream = new ReadableStream({
async start(controller) {
function onParse(event: any) {
if (event.type === "event") {
const data = event.data;
if (event.id == "1") {
let text1 = data.slice(data.indexOf("content"));
const text = text1.slice(12, text1.indexOf("index") - 6);
const queue = encoder.encode(text);
controller.enqueue(queue);
return;
}
// https://beta.openai.com/docs/api-reference/completions/create#completions/create-stream
try {
const json = JSON.parse(data);
// console.log(data.indexOf("content"))
if (data.indexOf("content") == -1) {
controller.close();
return;
}
// console.log(event.data)
const text = JSON.parse(json.data.slice(5)).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;
}
export async function POST(req: NextRequest) {
try {
const stream = await createStream(req);
return new Response(stream);
} catch (error) {
console.error("[Chat Stream]", error);
}
}
export const config = {
runtime: "edge",
};