chore: simplify some code

pull/464/head
Travis Fischer 2023-03-09 12:41:02 -06:00
rodzic 93a5094654
commit 874248d959
1 zmienionych plików z 9 dodań i 13 usunięć

Wyświetl plik

@ -334,12 +334,10 @@ export class ChatGPTAPI {
let nextMessages = text let nextMessages = text
? messages.concat([ ? messages.concat([
{ {
...{
role: 'user', role: 'user',
content: text, content: text,
name: opts.name name: opts.name
} }
}
]) ])
: messages : messages
let numTokens = 0 let numTokens = 0
@ -349,11 +347,11 @@ export class ChatGPTAPI {
.reduce((prompt, message) => { .reduce((prompt, message) => {
switch (message.role) { switch (message.role) {
case 'system': case 'system':
return [...prompt, `Instructions:\n${message.content}`] return prompt.concat([`Instructions:\n${message.content}`])
case 'user': case 'user':
return [...prompt, `${userLabel}:\n${message.content}`] return prompt.concat([`${userLabel}:\n${message.content}`])
default: default:
return [...prompt, `${assistantLabel}:\n${message.content}`] return prompt.concat([`${assistantLabel}:\n${message.content}`])
} }
}, [] as string[]) }, [] as string[])
.join('\n\n') .join('\n\n')
@ -385,11 +383,9 @@ export class ChatGPTAPI {
nextMessages = nextMessages.slice(0, systemMessageOffset).concat([ nextMessages = nextMessages.slice(0, systemMessageOffset).concat([
{ {
...{
role: parentMessageRole, role: parentMessageRole,
content: parentMessage.text, content: parentMessage.text,
name: parentMessage.name name: parentMessage.name
}
}, },
...nextMessages.slice(systemMessageOffset) ...nextMessages.slice(systemMessageOffset)
]) ])