Wroclaw/GPTcord
1
0
Fork 0
Code Issues 7 Pull requests 1 Releases Activity

Use Iterable when converting to OpenAI messages

Browse source
This commit is contained in:
Wroclaw 2023-08-20 15:12:26 +02:00
parent 0c0cbceb46
commit 32dd705498
2 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/execution.ts
View file

@ -274,7 +274,9 @@ async function executeFromQueue(channel: string) {
await message.deferReply();
}
OpenAImessages = toOpenAIMessages(messages);
messages.sort((a, b) => b.createdTimestamp - a.createdTimestamp);
OpenAImessages = toOpenAIMessages(messages.values());
let generatedMessage: ChatCompletionResponseMessage | undefined = undefined;
let answer: Awaited<ReturnType<typeof openai.createChatCompletion>>;

13
src/toOpenAIMessages.ts
View file

@ -92,17 +92,18 @@ function getAuthorUsername(message: DiscordMessage): string | undefined {
}
/**
* Converts the Collection of Discord Messages to array of OpenAI Messages to send
* @param messages the collection to convert
* Converts the Iterable of Discord Messages to array of OpenAI Messages to send
* first message in the interable will be the last of the returned array (reverse order)
* @param messages the iterable to convert
* @returns the converted messages
*/
export default function toOpenAIMessages(messages: Collection<string, DiscordMessage>): OpenAIMessage[] {
export default function toOpenAIMessages(
messages: Iterable<DiscordMessage>,
): OpenAIMessage[] {
const rvalue: OpenAIMessage[] = [];
let tokenCount = 0;
messages.sort((a, b) => b.createdTimestamp - a.createdTimestamp);
for (const message of messages.values()) {
for (const message of messages) {
const content = formatMessage(message);
// FIXME: tokens are not being counted properly (it's lower than it is) but it's enough for me for now.
tokenCount += countTokens(content);