From fcd049eae87b64422dd8430f13112055ee55938d Mon Sep 17 00:00:00 2001 From: Kwaroran Date: Thu, 12 Dec 2024 12:56:17 +0900 Subject: [PATCH] Refactor parameter handling in requestGoogleCloudVertex function to filter parameters based on modelInfo --- src/ts/process/request.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts index 588280ef..0bd0c650 100644 --- a/src/ts/process/request.ts +++ b/src/ts/process/request.ts @@ -1494,12 +1494,17 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise }, ] + let para:Parameter[] = ['temperature', 'top_p', 'top_k', 'presence_penalty', 'frequency_penalty'] + + para = para.filter((v) => { + return arg.modelInfo.parameters.includes(v) + }) const body = { contents: reformatedChat, generation_config: applyParameters({ "maxOutputTokens": maxTokens, - }, ['temperature', 'top_p', 'top_k', 'presence_penalty', 'frequency_penalty'], { + }, para, { 'top_p': "topP", 'top_k': "topK", 'presence_penalty': "presencePenalty",