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

Add experimental support for Claude AWS

Browse Source
This commit is contained in:
kwaroran
2024-01-11 21:50:04 +09:00
parent 6031ecea32
commit a319eedee1
3 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1427,27 +1427,23 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}).join('') + '\n\nAssistant: '
//claude bedrock
//placeholders
const bedrock = false
const region = ''
const AMZ_HOST = "invoke-bedrock.%REGION%.amazonaws.com";
const host = AMZ_HOST.replace("%REGION%", region);
function getCredentialParts(key:string) {
const [accessKeyId, secretAccessKey, region] = key.split(":");
if (!accessKeyId || !secretAccessKey || !region) {
throw new Error("The key assigned to this request is invalid.");
}
return { accessKeyId, secretAccessKey, region };
}
const bedrock = db.claudeAws
if(bedrock){
function getCredentialParts(key:string) {
const [accessKeyId, secretAccessKey, region] = key.split(":");
if (!accessKeyId || !secretAccessKey || !region) {
throw new Error("The key assigned to this request is invalid.");
}
return { accessKeyId, secretAccessKey, region };
}
const { accessKeyId, secretAccessKey, region } = getCredentialParts(apiKey);
const AMZ_HOST = "invoke-bedrock.%REGION%.amazonaws.com";
const host = AMZ_HOST.replace("%REGION%", region);
const stream = false
const url = `https://${host}/model/${model}/invoke${stream ? "-with-response-stream" : ""}`
const params = {
@@ -1472,7 +1468,6 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
});
const { accessKeyId, secretAccessKey, region } = getCredentialParts(apiKey);
const signer = new SignatureV4({
sha256: Sha256,
credentials: { accessKeyId, secretAccessKey },

1
src/ts/storage/database.ts
View File

@@ -591,6 +591,7 @@ export interface Database{
}
},
top_k:number
claudeAws:boolean
}
export interface customscript{