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

Merge branch 'main' of https://github.com/kwaroran/RisuAI

Browse Source
This commit is contained in:
kwaroran
2024-03-16 19:51:55 +09:00
parent fbde71e271 e7dda4093a
commit b5a6011039
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/lib/Setting/Pages/BotSettings.svelte
View File

@@ -186,6 +186,7 @@
<OptionInput value="claude-instant-v1.1-100k">claude-instant-v1.1-100k</OptionInput>
<OptionInput value="claude-3-opus-20240229">claude-3-opus-20240229</OptionInput>
<OptionInput value="claude-3-sonnet-20240229">claude-3-sonnet-20240229</OptionInput>
<!-- <OptionInput value="claude-3-haiku-202403079">claude-3-haiku-20240307</OptionInput> -->
<OptionInput value="custom">Custom</OptionInput>
</SelectInput>
{#if $DataBase.proxyRequestModel === 'custom'}

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

@@ -1514,18 +1514,24 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
const AMZ_HOST = "bedrock-runtime.%REGION%.amazonaws.com";
const host = AMZ_HOST.replace("%REGION%", region);
const stream = false
const CLAUDE_3_COMPAT_MODEL = "anthropic.claude-3-sonnet-20240229-v1:0";
const stream = false; // todo?
// AWS currently only supports one v3 model.
const awsModel = CLAUDE_3_COMPAT_MODEL;
// https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html
const modelIDs = [
"anthropic.claude-v2",
"anthropic.claude-v2:1",
"anthropic.claude-3-haiku-20240307-v1:0",
"anthropic.claude-3-sonnet-20240229-v1:0",
];
const awsModel = raiModel.includes("haiku") ? modelIDs[2] : modelIDs[3];
const url = `https://${host}/model/${awsModel}/invoke${stream ? "-with-response-stream" : ""}`
const params = {
messages : claudeChat,
system: systemPrompt.trim(),
max_tokens: maxTokens,
// stop_sequences: ["user:", "assistant:", "system:"],
// stop_sequences: null,
temperature: temperature,
top_p: db.top_p,
top_k: db.top_k,