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

Handle unregistered function calls

Browse source
This commit is contained in:
Wroclaw 2023-07-30 01:51:40 +02:00
parent 000641bcfc
commit 0e3962e110
2 changed files with 23 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/execution.ts
View file

@ -260,12 +260,10 @@ async function executeFromQueue(channel: string) {
// handle function calls // handle function calls
if (generatedMessage.function_call) { if (generatedMessage.function_call) {
OpenAImessages.push(generatedMessage); OpenAImessages.push(generatedMessage);
OpenAImessages.push({
role: "function",
name: generatedMessage.function_call.name,
// FIXME: don't use new instance of FunctionManager // FIXME: don't use new instance of FunctionManager
content: new FunctionManager().handleFunction(generatedMessage.function_call), OpenAImessages.push(
}); new FunctionManager().handleFunction(generatedMessage.function_call)
);
} }
} while (generatedMessage.function_call); } while (generatedMessage.function_call);

22
src/funcitonManager.ts
View file

@ -1,4 +1,5 @@
import { ChatCompletionFunctions, ChatCompletionRequestMessageFunctionCall } from "openai"; import { ChatCompletionFunctions, ChatCompletionRequestMessage, ChatCompletionRequestMessageFunctionCall } from "openai";
import config from "./config"; import config from "./config";
type parameterMap = { type parameterMap = {
@ -60,9 +61,20 @@ export default class FunctionManager {
return rvalue; return rvalue;
} }
public handleFunction(request: ChatCompletionRequestMessageFunctionCall) { public handleFunction(request: ChatCompletionRequestMessageFunctionCall): ChatCompletionRequestMessage {
// eslint-disable-next-line @typescript-eslint/no-explicit-any // eslint-disable-next-line @typescript-eslint/no-explicit-any
let parsedArguments: any; let parsedArguments: any;
const functionToRun = this.store.get(request.name ?? "");
// check if the function is registered
if (!functionToRun) {
return {
role: "system",
content: "Only use functions that were provided to you",
};
}
try { try {
parsedArguments = JSON.parse(request.arguments ?? ""); parsedArguments = JSON.parse(request.arguments ?? "");
} }
@ -70,7 +82,11 @@ export default class FunctionManager {
console.error("Function arguments raw: " + request.arguments); console.error("Function arguments raw: " + request.arguments);
throw new Error(`Failed to parse the function JSON arguments when running function [${request.name}]`, {cause: e}); throw new Error(`Failed to parse the function JSON arguments when running function [${request.name}]`, {cause: e});
} }
return this.store.get(request.name ?? "")?.execute(parsedArguments); return {
role: "function",
name: request.name,
content: functionToRun.execute(parsedArguments),
};
} }
} }