📦 EqualifyEverything / equalify

📄 getBedrockModels.ts · 22 lines
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22import { BedrockClient, ListFoundationModelsCommand } from "@aws-sdk/client-bedrock";

const bedrockMgmt = new BedrockClient({ region: process.env.AWS_REGION || "us-east-1" });

export const getBedrockModels = async () => {
    const response = await bedrockMgmt.send(new ListFoundationModelsCommand({}));

    const models = (response.modelSummaries ?? [])
        .filter(m =>
            m.inferenceTypesSupported?.includes("ON_DEMAND") &&
            m.outputModalities?.includes("TEXT") &&
            m.modelLifecycle?.status === "ACTIVE"
        )
        .map(m => ({
            modelId: m.modelId,
            modelName: m.modelName,
            providerName: m.providerName,
        }));

    return { models };
};