39 lines
897 B
TypeScript
39 lines
897 B
TypeScript
|
import type { ChatRequest, ChatReponse } from "./api/chat/typing";
|
||
|
|
import { Message } from "./store";
|
||
|
|
|
||
|
|
export async function requestChat(messages: Message[]) {
|
||
|
|
const req: ChatRequest = {
|
||
|
|
model: "gpt-3.5-turbo",
|
||
|
|
messages: messages
|
||
|
|
.map((v) => ({
|
||
|
|
role: v.role,
|
||
|
|
content: v.content,
|
||
|
|
}))
|
||
|
|
.filter((m) => m.role !== "assistant"),
|
||
|
|
};
|
||
|
|
|
||
|
|
const res = await fetch("/api/chat", {
|
||
|
|
method: "POST",
|
||
|
|
headers: {
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
},
|
||
|
|
body: JSON.stringify(req),
|
||
|
|
});
|
||
|
|
|
||
|
|
return (await res.json()) as ChatReponse;
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function requestWithPrompt(messages: Message[], prompt: string) {
|
||
|
|
messages = messages.concat([
|
||
|
|
{
|
||
|
|
role: "system",
|
||
|
|
content: prompt,
|
||
|
|
date: new Date().toLocaleString(),
|
||
|
|
},
|
||
|
|
]);
|
||
|
|
|
||
|
|
const res = await requestChat(messages);
|
||
|
|
|
||
|
|
return res.choices.at(0)?.message?.content ?? "";
|
||
|
|
}
|