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

functionManager: Forward username mapping here

Browse source
This commit is contained in:
Wroclaw 2023-09-28 11:42:01 +02:00
parent 52f16c9d0a
commit 6e0a5e72ea
2 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/funcitonManager.ts
View file

@ -43,7 +43,10 @@ export abstract class OpenAIFunction<T extends nameTypeMap = nameTypeMap> {
};
}
abstract execute(data: OpenAIFunctionRequestData<T>): string;
abstract execute(
data: OpenAIFunctionRequestData<T>,
modelUsernameMap: Map<string, string>,
): string;
}
/*
@ -64,7 +67,10 @@ export default class FunctionManager {
return rvalue;
}
public handleFunction(request: ChatCompletionFunctionCall): ChatCompletionMessageParam {
public handleFunction(
request: ChatCompletionFunctionCall,
modelUsernameMap: Map<string, string> = new Map(),
): ChatCompletionMessageParam {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
let parsedArguments: any;
@ -91,7 +97,7 @@ export default class FunctionManager {
role: "function",
name: request.name,
// eslint-disable-next-line @typescript-eslint/no-unsafe-argument
content: functionToRun.execute(parsedArguments),
content: functionToRun.execute(parsedArguments, modelUsernameMap),
};
}
}