MurasameNoa/chatgpt-plugin
1
0
Fork 0
mirror of https://github.com/ikechan8370/chatgpt-plugin.git synced 2025-12-17 13:57:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cherry-pick: 1

Browse source
This commit is contained in:
ikechan8370 2024-03-10 17:27:41 +08:00
parent 8e73a28dae
commit fd2d976686
5 changed files with 69 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

14
utils/openai/chatgpt-api.ts
View file

@ -7,7 +7,7 @@ import * as tokenizer from './tokenizer'
import * as types from './types'
import globalFetch from 'node-fetch'
import { fetchSSE } from './fetch-sse'
import {openai, Role} from "./types";
import {ChatCompletionRequestMessage, openai, Role} from "./types";
const CHATGPT_MODEL = 'gpt-3.5-turbo-0613'
@ -176,16 +176,17 @@ export class ChatGPTAPI {
completionParams
)
console.log(`maxTokens: ${maxTokens}, numTokens: ${numTokens}`)
const result: types.ChatMessage = {
const result: types.ChatMessage & { conversation: openai.ChatCompletionRequestMessage[] }= {
role: 'assistant',
id: uuidv4(),
conversationId,
parentMessageId: messageId,
text: undefined,
functionCall: undefined
functionCall: undefined,
conversation: []
}
const responseP = new Promise<types.ChatMessage>(
const responseP = new Promise<types.ChatMessage & { conversation: openai.ChatCompletionRequestMessage[] }>(
async (resolve, reject) => {
const url = `${this._apiBaseUrl}/chat/completions`
const headers = {
@ -223,6 +224,7 @@ export class ChatGPTAPI {
onMessage: (data: string) => {
if (data === '[DONE]') {
result.text = result.text.trim()
result.conversation = messages
return resolve(result)
}
@ -318,7 +320,7 @@ export class ChatGPTAPI {
}
result.detail = response
result.conversation = messages
return resolve(result)
} catch (err) {
return reject(err)
@ -548,4 +550,4 @@ export class ChatGPTAPI {
): Promise<void> {
await this._messageStore.set(message.id, message)
}
}
}