mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Gemini parameters

Browse Source
This commit is contained in:
Kwaroran
2024-12-12 07:57:01 +09:00
parent 2f1e14d0d7
commit 61763deaa7
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/ts/process/index.svelte.ts
View File

@@ -41,7 +41,7 @@ export interface OpenAIChat{
}
export interface MultiModal{
type:'image'|'video'
type:'image'|'video'|'audio'
base64:string,
height?:number,
width?:number

13
src/ts/process/request.ts
View File

@@ -1406,7 +1406,11 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
});
for (const modal of chat.multimodals) {
if (modal.type === "image") {
if (
(modal.type === "image" && arg.modelInfo.flags.includes(LLMFlags.hasImageInput)) ||
(modal.type === "audio" && arg.modelInfo.flags.includes(LLMFlags.hasAudioInput)) ||
(modal.type === "video" && arg.modelInfo.flags.includes(LLMFlags.hasVideoInput))
) {
const dataurl = modal.base64;
const base64 = dataurl.split(",")[1];
const mediaType = dataurl.split(";")[0].split(":")[1];
@@ -1486,8 +1490,11 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
contents: reformatedChat,
generation_config: applyParameters({
"maxOutputTokens": maxTokens,
}, ['temperature', 'top_p'], {
'top_p': "topP"
}, ['temperature', 'top_p', 'top_k', 'presence_penalty', 'frequency_penalty'], {
'top_p': "topP",
'top_k': "topK",
'presence_penalty': "presencePenalty",
'frequency_penalty': "frequencyPenalty"
}, arg.mode),
safetySettings: uncensoredCatagory,
systemInstruction: {