add aws claude haiku (#310)

# PR Checklist
- [x] Did you check if it works normally in all models? *ignore this
when it dosen't uses models*
- [x] Did you check if it works normally in all of web, local and node
hosted versions? if it dosen't, did you blocked it in those versions?
- [x] Did you added a type def?

# Description

![image](https://github.com/kwaroran/RisuAI/assets/133192207/092e3aee-e93d-4f3c-90f6-f7ec40b6d4a3)

![image](https://github.com/kwaroran/RisuAI/assets/133192207/7665bd0a-28e1-4506-83bd-78e5da5b3da5)
This commit is contained in:
kwaroran
2024-03-16 15:34:10 +09:00
committed by GitHub
2 changed files with 12 additions and 5 deletions

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,