You've already forked bilibili-subtitle
21 lines
659 B
TypeScript
21 lines
659 B
TypeScript
import {getServerUrl} from '../util/biz_util'
|
|
|
|
export const handleChatCompleteTask = async (task: Task) => {
|
|
const data = task.def.data
|
|
const serverUrl = getServerUrl(task.def.serverUrl)
|
|
const resp = await fetch(`${serverUrl}/v1/chat/completions`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
Authorization: 'Bearer ' + task.def.extra.apiKey,
|
|
},
|
|
body: JSON.stringify(data),
|
|
})
|
|
task.resp = await resp.json()
|
|
if (task.resp.usage) {
|
|
return (task.resp.usage.total_tokens??0) > 0
|
|
} else {
|
|
throw new Error(`${task.resp.error.code as string??''} ${task.resp.error.message as string ??''}`)
|
|
}
|
|
}
|