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

add aws haiku

Browse Source
This commit is contained in:
bangonicdd
2024-03-16 09:07:11 +09:00
parent c46744f4e4
commit 1907d4f229
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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("sonnet") ? modelIDs[3] : modelIDs[2];
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,