UNPKG

wickr-bedrock-bot

Version:

AWS Wickr's own Bedrock Bot

58 lines (57 loc) 2.83 kB
import { createAggregatedClient } from "@smithy/smithy-client"; import { BedrockAgentRuntimeClient } from "./BedrockAgentRuntimeClient"; import { CreateInvocationCommand, } from "./commands/CreateInvocationCommand"; import { CreateSessionCommand, } from "./commands/CreateSessionCommand"; import { DeleteAgentMemoryCommand, } from "./commands/DeleteAgentMemoryCommand"; import { DeleteSessionCommand, } from "./commands/DeleteSessionCommand"; import { EndSessionCommand } from "./commands/EndSessionCommand"; import { GenerateQueryCommand, } from "./commands/GenerateQueryCommand"; import { GetAgentMemoryCommand, } from "./commands/GetAgentMemoryCommand"; import { GetInvocationStepCommand, } from "./commands/GetInvocationStepCommand"; import { GetSessionCommand } from "./commands/GetSessionCommand"; import { InvokeAgentCommand } from "./commands/InvokeAgentCommand"; import { InvokeFlowCommand } from "./commands/InvokeFlowCommand"; import { InvokeInlineAgentCommand, } from "./commands/InvokeInlineAgentCommand"; import { ListInvocationsCommand, } from "./commands/ListInvocationsCommand"; import { ListInvocationStepsCommand, } from "./commands/ListInvocationStepsCommand"; import { ListSessionsCommand, } from "./commands/ListSessionsCommand"; import { ListTagsForResourceCommand, } from "./commands/ListTagsForResourceCommand"; import { OptimizePromptCommand, } from "./commands/OptimizePromptCommand"; import { PutInvocationStepCommand, } from "./commands/PutInvocationStepCommand"; import { RerankCommand } from "./commands/RerankCommand"; import { RetrieveAndGenerateCommand, } from "./commands/RetrieveAndGenerateCommand"; import { RetrieveAndGenerateStreamCommand, } from "./commands/RetrieveAndGenerateStreamCommand"; import { RetrieveCommand } from "./commands/RetrieveCommand"; import { TagResourceCommand } from "./commands/TagResourceCommand"; import { UntagResourceCommand, } from "./commands/UntagResourceCommand"; import { UpdateSessionCommand, } from "./commands/UpdateSessionCommand"; const commands = { CreateInvocationCommand, CreateSessionCommand, DeleteAgentMemoryCommand, DeleteSessionCommand, EndSessionCommand, GenerateQueryCommand, GetAgentMemoryCommand, GetInvocationStepCommand, GetSessionCommand, InvokeAgentCommand, InvokeFlowCommand, InvokeInlineAgentCommand, ListInvocationsCommand, ListInvocationStepsCommand, ListSessionsCommand, ListTagsForResourceCommand, OptimizePromptCommand, PutInvocationStepCommand, RerankCommand, RetrieveCommand, RetrieveAndGenerateCommand, RetrieveAndGenerateStreamCommand, TagResourceCommand, UntagResourceCommand, UpdateSessionCommand, }; export class BedrockAgentRuntime extends BedrockAgentRuntimeClient { } createAggregatedClient(commands, BedrockAgentRuntime);