forked from i-am-alice/3rd-devs
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathOpenAIService.ts
33 lines (29 loc) · 926 Bytes
/
OpenAIService.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import OpenAI from "openai";
import type { ChatCompletionMessageParam } from "openai/resources/chat/completions";
export class OpenAIService {
private openai: OpenAI;
constructor() {
this.openai = new OpenAI();
}
async completion(
messages: ChatCompletionMessageParam[],
model: string = "gpt-4",
stream: boolean = false
): Promise<OpenAI.Chat.Completions.ChatCompletion | AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>> {
try {
const chatCompletion = await this.openai.chat.completions.create({
messages,
model,
stream,
});
if (stream) {
return chatCompletion as AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;
} else {
return chatCompletion as OpenAI.Chat.Completions.ChatCompletion;
}
} catch (error) {
console.error("Error in OpenAI completion:", error);
throw error;
}
}
}