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,11 +334,9 @@ 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
@ -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)
]) ])