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

fix: 修复openai client由于计算token引起的错误

Browse source
This commit is contained in:
ikechan8370 2023-07-01 21:50:26 +08:00
parent 285afad993
commit 80b320ff3a
2 changed files with 123 additions and 102 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -415,44 +415,47 @@ export class ChatGPTAPI {
let functionToken = 0
let numTokens = functionToken
for (const func of completionParams.functions) {
functionToken += await this._getTokenCount(func.name)
functionToken += await this._getTokenCount(func.description)
if (func.parameters.properties) {
for (let key of Object.keys(func.parameters.properties)) {
functionToken += await this._getTokenCount(key)
let property = func.parameters.properties[key]
for (let field of Object.keys(property)) {
switch (field) {
case 'type': {
functionToken += 2
functionToken += await this._getTokenCount(property.type)
break
}
case 'description': {
functionToken += 2
functionToken += await this._getTokenCount(property.description)
break
}
case 'field': {
functionToken -= 3
for (let enumElement of property.enum) {
functionToken += 3
functionToken += await this._getTokenCount(enumElement)
if (completionParams.functions) {
for (const func of completionParams.functions) {
functionToken += await this._getTokenCount(func?.name)
functionToken += await this._getTokenCount(func?.description)
if (func?.parameters?.properties) {
for (let key of Object.keys(func.parameters.properties)) {
functionToken += await this._getTokenCount(key)
let property = func.parameters.properties[key]
for (let field of Object.keys(property)) {
switch (field) {
case 'type': {
functionToken += 2
functionToken += await this._getTokenCount(property?.type)
break
}
case 'description': {
functionToken += 2
functionToken += await this._getTokenCount(property?.description)
break
}
case 'enum': {
functionToken -= 3
for (let enumElement of property?.enum) {
functionToken += 3
functionToken += await this._getTokenCount(enumElement)
}
break
}
break
}
}
}
}
}
if (func.parameters.required) {
for (let string of func.parameters.required) {
functionToken += 2
functionToken += await this._getTokenCount(string)
if (func?.parameters?.required) {
for (let string of func.parameters.required) {
functionToken += 2
functionToken += await this._getTokenCount(string)
}
}
}
}
do {
const prompt = nextMessages
.reduce((prompt, message) => {