Skip to content

Commit d77e808

Browse files
committed
feat: allow other providers
1 parent a35b170 commit d77e808

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

packages/utils-ai/src/chat.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@ export function defineMessages(message: Message[]): Message[] {
66
}
77

88
export async function fetchCompletion(messages: Message[], options: AI): Promise<Completion> {
9-
return ofetch<Completion>('https://api.openai.com/v1/chat/completions', {
9+
return ofetch<Completion>(options.endpoint, {
1010
method: 'POST',
1111
headers: {
1212
'Content-Type': 'application/json',
1313
'Authorization': `Bearer ${options.accessKey}`,
1414
},
1515
body: {
16-
model: 'gpt-3.5-turbo',
16+
model: options.model,
1717
messages,
1818
max_tokens: options.maxTokens ?? null,
1919
},

packages/utils-ai/src/config.ts

+2
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,7 @@ export const defaultConfig = defineUtilsAIConfig<Config>({
1414
accessKey: null,
1515
maxTokens: 2048,
1616
temperature: 0.7,
17+
endpoint: 'https://api.openai.com/v1/chat/completions',
18+
model: 'gpt-3.5-turbo',
1719
},
1820
})

packages/utils-ai/src/types.ts

+2
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ export interface AI {
77
accessKey: string | null
88
maxTokens?: number
99
temperature?: number
10+
endpoint: string
11+
model: string
1012
}
1113

1214
export type Language = 'fr' | 'en'

0 commit comments

Comments
 (0)