kopia lustrzana https://github.com/transitive-bullshit/chatgpt-api
chore: minor fixes
rodzic
db6e5941ab
commit
0f966dc402
|
@ -231,7 +231,6 @@ export class ChatGPTAPI {
|
||||||
}
|
}
|
||||||
|
|
||||||
result.detail = response
|
result.detail = response
|
||||||
|
|
||||||
onProgress?.(result)
|
onProgress?.(result)
|
||||||
}
|
}
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
|
@ -299,15 +298,21 @@ export class ChatGPTAPI {
|
||||||
}
|
}
|
||||||
).then(async (message) => {
|
).then(async (message) => {
|
||||||
if (message.detail && !message.detail.usage) {
|
if (message.detail && !message.detail.usage) {
|
||||||
const promptTokens = numTokens
|
try {
|
||||||
const completionTokens = await this._getTokenCount(message.text)
|
const promptTokens = numTokens
|
||||||
message.detail.usage = {
|
const completionTokens = await this._getTokenCount(message.text)
|
||||||
prompt_tokens: promptTokens,
|
message.detail.usage = {
|
||||||
completion_tokens: completionTokens,
|
prompt_tokens: promptTokens,
|
||||||
total_tokens: promptTokens + completionTokens,
|
completion_tokens: completionTokens,
|
||||||
estimated: true
|
total_tokens: promptTokens + completionTokens,
|
||||||
|
estimated: true
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
// TODO: this should really never happen, but if it does,
|
||||||
|
// we should handle notify the user gracefully
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return this._upsertMessage(message).then(() => message)
|
return this._upsertMessage(message).then(() => message)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
20
src/types.ts
20
src/types.ts
|
@ -61,16 +61,6 @@ export type SendMessageBrowserOptions = {
|
||||||
abortSignal?: AbortSignal
|
abortSignal?: AbortSignal
|
||||||
}
|
}
|
||||||
|
|
||||||
interface CreateChatCompletionStreamResponse
|
|
||||||
extends openai.CreateChatCompletionDeltaResponse {
|
|
||||||
usage: CreateCompletionStreamResponseUsage
|
|
||||||
}
|
|
||||||
|
|
||||||
interface CreateCompletionStreamResponseUsage
|
|
||||||
extends openai.CreateCompletionResponseUsage {
|
|
||||||
estimated: true
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface ChatMessage {
|
export interface ChatMessage {
|
||||||
id: string
|
id: string
|
||||||
text: string
|
text: string
|
||||||
|
@ -100,6 +90,16 @@ export type GetMessageByIdFunction = (id: string) => Promise<ChatMessage>
|
||||||
/** Upserts a chat message to a store. */
|
/** Upserts a chat message to a store. */
|
||||||
export type UpsertMessageFunction = (message: ChatMessage) => Promise<void>
|
export type UpsertMessageFunction = (message: ChatMessage) => Promise<void>
|
||||||
|
|
||||||
|
export interface CreateChatCompletionStreamResponse
|
||||||
|
extends openai.CreateChatCompletionDeltaResponse {
|
||||||
|
usage: CreateCompletionStreamResponseUsage
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface CreateCompletionStreamResponseUsage
|
||||||
|
extends openai.CreateCompletionResponseUsage {
|
||||||
|
estimated: true
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* https://chat.openapi.com/backend-api/conversation
|
* https://chat.openapi.com/backend-api/conversation
|
||||||
*/
|
*/
|
||||||
|
|
Ładowanie…
Reference in New Issue