@aws-sdk/client-bedrock-agent-runtime 3.701.0 → 3.703.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (33) hide show
  1. package/README.md +16 -0
  2. package/dist-cjs/index.js +530 -3
  3. package/dist-es/BedrockAgentRuntime.js +4 -0
  4. package/dist-es/commands/RerankCommand.js +23 -0
  5. package/dist-es/commands/RetrieveAndGenerateStreamCommand.js +27 -0
  6. package/dist-es/commands/index.js +2 -0
  7. package/dist-es/models/models_0.js +180 -0
  8. package/dist-es/pagination/RerankPaginator.js +4 -0
  9. package/dist-es/pagination/index.js +1 -0
  10. package/dist-es/protocols/Aws_restJson1.js +231 -0
  11. package/dist-types/BedrockAgentRuntime.d.ts +14 -0
  12. package/dist-types/BedrockAgentRuntimeClient.d.ts +4 -2
  13. package/dist-types/commands/InvokeAgentCommand.d.ts +45 -2
  14. package/dist-types/commands/InvokeInlineAgentCommand.d.ts +45 -2
  15. package/dist-types/commands/RerankCommand.d.ts +143 -0
  16. package/dist-types/commands/RetrieveAndGenerateCommand.d.ts +42 -2
  17. package/dist-types/commands/RetrieveAndGenerateStreamCommand.d.ts +371 -0
  18. package/dist-types/commands/RetrieveCommand.d.ts +46 -1
  19. package/dist-types/commands/index.d.ts +2 -0
  20. package/dist-types/models/models_0.d.ts +1038 -166
  21. package/dist-types/pagination/RerankPaginator.d.ts +7 -0
  22. package/dist-types/pagination/index.d.ts +1 -0
  23. package/dist-types/protocols/Aws_restJson1.d.ts +18 -0
  24. package/dist-types/ts3.4/BedrockAgentRuntime.d.ts +34 -0
  25. package/dist-types/ts3.4/BedrockAgentRuntimeClient.d.ts +12 -0
  26. package/dist-types/ts3.4/commands/RerankCommand.d.ts +41 -0
  27. package/dist-types/ts3.4/commands/RetrieveAndGenerateStreamCommand.d.ts +51 -0
  28. package/dist-types/ts3.4/commands/index.d.ts +2 -0
  29. package/dist-types/ts3.4/models/models_0.d.ts +455 -0
  30. package/dist-types/ts3.4/pagination/RerankPaginator.d.ts +11 -0
  31. package/dist-types/ts3.4/pagination/index.d.ts +1 -0
  32. package/dist-types/ts3.4/protocols/Aws_restJson1.d.ts +24 -0
  33. package/package.json +1 -1
@@ -0,0 +1,7 @@
1
+ import { Paginator } from "@smithy/types";
2
+ import { RerankCommandInput, RerankCommandOutput } from "../commands/RerankCommand";
3
+ import { BedrockAgentRuntimePaginationConfiguration } from "./Interfaces";
4
+ /**
5
+ * @public
6
+ */
7
+ export declare const paginateRerank: (config: BedrockAgentRuntimePaginationConfiguration, input: RerankCommandInput, ...rest: any[]) => Paginator<RerankCommandOutput>;
@@ -1,3 +1,4 @@
1
1
  export * from "./GetAgentMemoryPaginator";
2
2
  export * from "./Interfaces";
3
+ export * from "./RerankPaginator";
3
4
  export * from "./RetrievePaginator";
@@ -6,7 +6,9 @@ import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "../commands/I
6
6
  import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "../commands/InvokeFlowCommand";
7
7
  import { InvokeInlineAgentCommandInput, InvokeInlineAgentCommandOutput } from "../commands/InvokeInlineAgentCommand";
8
8
  import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "../commands/OptimizePromptCommand";
9
+ import { RerankCommandInput, RerankCommandOutput } from "../commands/RerankCommand";
9
10
  import { RetrieveAndGenerateCommandInput, RetrieveAndGenerateCommandOutput } from "../commands/RetrieveAndGenerateCommand";
11
+ import { RetrieveAndGenerateStreamCommandInput, RetrieveAndGenerateStreamCommandOutput } from "../commands/RetrieveAndGenerateStreamCommand";
10
12
  import { RetrieveCommandInput, RetrieveCommandOutput } from "../commands/RetrieveCommand";
11
13
  /**
12
14
  * serializeAws_restJson1DeleteAgentMemoryCommand
@@ -32,6 +34,10 @@ export declare const se_InvokeInlineAgentCommand: (input: InvokeInlineAgentComma
32
34
  * serializeAws_restJson1OptimizePromptCommand
33
35
  */
34
36
  export declare const se_OptimizePromptCommand: (input: OptimizePromptCommandInput, context: __SerdeContext) => Promise<__HttpRequest>;
37
+ /**
38
+ * serializeAws_restJson1RerankCommand
39
+ */
40
+ export declare const se_RerankCommand: (input: RerankCommandInput, context: __SerdeContext) => Promise<__HttpRequest>;
35
41
  /**
36
42
  * serializeAws_restJson1RetrieveCommand
37
43
  */
@@ -40,6 +46,10 @@ export declare const se_RetrieveCommand: (input: RetrieveCommandInput, context:
40
46
  * serializeAws_restJson1RetrieveAndGenerateCommand
41
47
  */
42
48
  export declare const se_RetrieveAndGenerateCommand: (input: RetrieveAndGenerateCommandInput, context: __SerdeContext) => Promise<__HttpRequest>;
49
+ /**
50
+ * serializeAws_restJson1RetrieveAndGenerateStreamCommand
51
+ */
52
+ export declare const se_RetrieveAndGenerateStreamCommand: (input: RetrieveAndGenerateStreamCommandInput, context: __SerdeContext) => Promise<__HttpRequest>;
43
53
  /**
44
54
  * deserializeAws_restJson1DeleteAgentMemoryCommand
45
55
  */
@@ -64,6 +74,10 @@ export declare const de_InvokeInlineAgentCommand: (output: __HttpResponse, conte
64
74
  * deserializeAws_restJson1OptimizePromptCommand
65
75
  */
66
76
  export declare const de_OptimizePromptCommand: (output: __HttpResponse, context: __SerdeContext & __EventStreamSerdeContext) => Promise<OptimizePromptCommandOutput>;
77
+ /**
78
+ * deserializeAws_restJson1RerankCommand
79
+ */
80
+ export declare const de_RerankCommand: (output: __HttpResponse, context: __SerdeContext) => Promise<RerankCommandOutput>;
67
81
  /**
68
82
  * deserializeAws_restJson1RetrieveCommand
69
83
  */
@@ -72,3 +86,7 @@ export declare const de_RetrieveCommand: (output: __HttpResponse, context: __Ser
72
86
  * deserializeAws_restJson1RetrieveAndGenerateCommand
73
87
  */
74
88
  export declare const de_RetrieveAndGenerateCommand: (output: __HttpResponse, context: __SerdeContext) => Promise<RetrieveAndGenerateCommandOutput>;
89
+ /**
90
+ * deserializeAws_restJson1RetrieveAndGenerateStreamCommand
91
+ */
92
+ export declare const de_RetrieveAndGenerateStreamCommand: (output: __HttpResponse, context: __SerdeContext & __EventStreamSerdeContext) => Promise<RetrieveAndGenerateStreamCommandOutput>;
@@ -24,10 +24,18 @@ import {
24
24
  OptimizePromptCommandInput,
25
25
  OptimizePromptCommandOutput,
26
26
  } from "./commands/OptimizePromptCommand";
27
+ import {
28
+ RerankCommandInput,
29
+ RerankCommandOutput,
30
+ } from "./commands/RerankCommand";
27
31
  import {
28
32
  RetrieveAndGenerateCommandInput,
29
33
  RetrieveAndGenerateCommandOutput,
30
34
  } from "./commands/RetrieveAndGenerateCommand";
35
+ import {
36
+ RetrieveAndGenerateStreamCommandInput,
37
+ RetrieveAndGenerateStreamCommandOutput,
38
+ } from "./commands/RetrieveAndGenerateStreamCommand";
31
39
  import {
32
40
  RetrieveCommandInput,
33
41
  RetrieveCommandOutput,
@@ -111,6 +119,19 @@ export interface BedrockAgentRuntime {
111
119
  options: __HttpHandlerOptions,
112
120
  cb: (err: any, data?: OptimizePromptCommandOutput) => void
113
121
  ): void;
122
+ rerank(
123
+ args: RerankCommandInput,
124
+ options?: __HttpHandlerOptions
125
+ ): Promise<RerankCommandOutput>;
126
+ rerank(
127
+ args: RerankCommandInput,
128
+ cb: (err: any, data?: RerankCommandOutput) => void
129
+ ): void;
130
+ rerank(
131
+ args: RerankCommandInput,
132
+ options: __HttpHandlerOptions,
133
+ cb: (err: any, data?: RerankCommandOutput) => void
134
+ ): void;
114
135
  retrieve(
115
136
  args: RetrieveCommandInput,
116
137
  options?: __HttpHandlerOptions
@@ -137,6 +158,19 @@ export interface BedrockAgentRuntime {
137
158
  options: __HttpHandlerOptions,
138
159
  cb: (err: any, data?: RetrieveAndGenerateCommandOutput) => void
139
160
  ): void;
161
+ retrieveAndGenerateStream(
162
+ args: RetrieveAndGenerateStreamCommandInput,
163
+ options?: __HttpHandlerOptions
164
+ ): Promise<RetrieveAndGenerateStreamCommandOutput>;
165
+ retrieveAndGenerateStream(
166
+ args: RetrieveAndGenerateStreamCommandInput,
167
+ cb: (err: any, data?: RetrieveAndGenerateStreamCommandOutput) => void
168
+ ): void;
169
+ retrieveAndGenerateStream(
170
+ args: RetrieveAndGenerateStreamCommandInput,
171
+ options: __HttpHandlerOptions,
172
+ cb: (err: any, data?: RetrieveAndGenerateStreamCommandOutput) => void
173
+ ): void;
140
174
  }
141
175
  export declare class BedrockAgentRuntime
142
176
  extends BedrockAgentRuntimeClient
@@ -74,10 +74,18 @@ import {
74
74
  OptimizePromptCommandInput,
75
75
  OptimizePromptCommandOutput,
76
76
  } from "./commands/OptimizePromptCommand";
77
+ import {
78
+ RerankCommandInput,
79
+ RerankCommandOutput,
80
+ } from "./commands/RerankCommand";
77
81
  import {
78
82
  RetrieveAndGenerateCommandInput,
79
83
  RetrieveAndGenerateCommandOutput,
80
84
  } from "./commands/RetrieveAndGenerateCommand";
85
+ import {
86
+ RetrieveAndGenerateStreamCommandInput,
87
+ RetrieveAndGenerateStreamCommandOutput,
88
+ } from "./commands/RetrieveAndGenerateStreamCommand";
81
89
  import {
82
90
  RetrieveCommandInput,
83
91
  RetrieveCommandOutput,
@@ -96,7 +104,9 @@ export type ServiceInputTypes =
96
104
  | InvokeFlowCommandInput
97
105
  | InvokeInlineAgentCommandInput
98
106
  | OptimizePromptCommandInput
107
+ | RerankCommandInput
99
108
  | RetrieveAndGenerateCommandInput
109
+ | RetrieveAndGenerateStreamCommandInput
100
110
  | RetrieveCommandInput;
101
111
  export type ServiceOutputTypes =
102
112
  | DeleteAgentMemoryCommandOutput
@@ -105,7 +115,9 @@ export type ServiceOutputTypes =
105
115
  | InvokeFlowCommandOutput
106
116
  | InvokeInlineAgentCommandOutput
107
117
  | OptimizePromptCommandOutput
118
+ | RerankCommandOutput
108
119
  | RetrieveAndGenerateCommandOutput
120
+ | RetrieveAndGenerateStreamCommandOutput
109
121
  | RetrieveCommandOutput;
110
122
  export interface ClientDefaults
111
123
  extends Partial<__SmithyConfiguration<__HttpHandlerOptions>> {
@@ -0,0 +1,41 @@
1
+ import { Command as $Command } from "@smithy/smithy-client";
2
+ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
+ import {
4
+ BedrockAgentRuntimeClientResolvedConfig,
5
+ ServiceInputTypes,
6
+ ServiceOutputTypes,
7
+ } from "../BedrockAgentRuntimeClient";
8
+ import { RerankRequest, RerankResponse } from "../models/models_0";
9
+ export { __MetadataBearer };
10
+ export { $Command };
11
+ export interface RerankCommandInput extends RerankRequest {}
12
+ export interface RerankCommandOutput extends RerankResponse, __MetadataBearer {}
13
+ declare const RerankCommand_base: {
14
+ new (input: RerankCommandInput): import("@smithy/smithy-client").CommandImpl<
15
+ RerankCommandInput,
16
+ RerankCommandOutput,
17
+ BedrockAgentRuntimeClientResolvedConfig,
18
+ ServiceInputTypes,
19
+ ServiceOutputTypes
20
+ >;
21
+ new (__0_0: RerankCommandInput): import("@smithy/smithy-client").CommandImpl<
22
+ RerankCommandInput,
23
+ RerankCommandOutput,
24
+ BedrockAgentRuntimeClientResolvedConfig,
25
+ ServiceInputTypes,
26
+ ServiceOutputTypes
27
+ >;
28
+ getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
29
+ };
30
+ export declare class RerankCommand extends RerankCommand_base {
31
+ protected static __types: {
32
+ api: {
33
+ input: RerankRequest;
34
+ output: RerankResponse;
35
+ };
36
+ sdk: {
37
+ input: RerankCommandInput;
38
+ output: RerankCommandOutput;
39
+ };
40
+ };
41
+ }
@@ -0,0 +1,51 @@
1
+ import { Command as $Command } from "@smithy/smithy-client";
2
+ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
+ import {
4
+ BedrockAgentRuntimeClientResolvedConfig,
5
+ ServiceInputTypes,
6
+ ServiceOutputTypes,
7
+ } from "../BedrockAgentRuntimeClient";
8
+ import {
9
+ RetrieveAndGenerateStreamRequest,
10
+ RetrieveAndGenerateStreamResponse,
11
+ } from "../models/models_0";
12
+ export { __MetadataBearer };
13
+ export { $Command };
14
+ export interface RetrieveAndGenerateStreamCommandInput
15
+ extends RetrieveAndGenerateStreamRequest {}
16
+ export interface RetrieveAndGenerateStreamCommandOutput
17
+ extends RetrieveAndGenerateStreamResponse,
18
+ __MetadataBearer {}
19
+ declare const RetrieveAndGenerateStreamCommand_base: {
20
+ new (
21
+ input: RetrieveAndGenerateStreamCommandInput
22
+ ): import("@smithy/smithy-client").CommandImpl<
23
+ RetrieveAndGenerateStreamCommandInput,
24
+ RetrieveAndGenerateStreamCommandOutput,
25
+ BedrockAgentRuntimeClientResolvedConfig,
26
+ ServiceInputTypes,
27
+ ServiceOutputTypes
28
+ >;
29
+ new (
30
+ __0_0: RetrieveAndGenerateStreamCommandInput
31
+ ): import("@smithy/smithy-client").CommandImpl<
32
+ RetrieveAndGenerateStreamCommandInput,
33
+ RetrieveAndGenerateStreamCommandOutput,
34
+ BedrockAgentRuntimeClientResolvedConfig,
35
+ ServiceInputTypes,
36
+ ServiceOutputTypes
37
+ >;
38
+ getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
39
+ };
40
+ export declare class RetrieveAndGenerateStreamCommand extends RetrieveAndGenerateStreamCommand_base {
41
+ protected static __types: {
42
+ api: {
43
+ input: RetrieveAndGenerateStreamRequest;
44
+ output: RetrieveAndGenerateStreamResponse;
45
+ };
46
+ sdk: {
47
+ input: RetrieveAndGenerateStreamCommandInput;
48
+ output: RetrieveAndGenerateStreamCommandOutput;
49
+ };
50
+ };
51
+ }
@@ -4,5 +4,7 @@ export * from "./InvokeAgentCommand";
4
4
  export * from "./InvokeFlowCommand";
5
5
  export * from "./InvokeInlineAgentCommand";
6
6
  export * from "./OptimizePromptCommand";
7
+ export * from "./RerankCommand";
7
8
  export * from "./RetrieveAndGenerateCommand";
9
+ export * from "./RetrieveAndGenerateStreamCommand";
8
10
  export * from "./RetrieveCommand";