aws-sdk-bedrockagentruntime 1.5.0 → 1.7.0

Sign up to get free protection for your applications and to get access to all the features.
@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
53
53
  # @!group service
54
54
  module Aws::BedrockAgentRuntime
55
55
 
56
- GEM_VERSION = '1.5.0'
56
+ GEM_VERSION = '1.7.0'
57
57
 
58
58
  end
data/sig/client.rbs CHANGED
@@ -87,10 +87,33 @@ module Aws
87
87
  agent_id: ::String,
88
88
  ?enable_trace: bool,
89
89
  ?end_session: bool,
90
- input_text: ::String,
90
+ ?input_text: ::String,
91
91
  session_id: ::String,
92
92
  ?session_state: {
93
+ invocation_id: ::String?,
93
94
  prompt_session_attributes: Hash[::String, ::String]?,
95
+ return_control_invocation_results: Array[
96
+ {
97
+ api_result: {
98
+ action_group: ::String,
99
+ api_path: ::String?,
100
+ http_method: ::String?,
101
+ http_status_code: ::Integer?,
102
+ response_body: Hash[::String, {
103
+ body: ::String?
104
+ }]?,
105
+ response_state: ("FAILURE" | "REPROMPT")?
106
+ }?,
107
+ function_result: {
108
+ action_group: ::String,
109
+ function: ::String?,
110
+ response_body: Hash[::String, {
111
+ body: ::String?
112
+ }]?,
113
+ response_state: ("FAILURE" | "REPROMPT")?
114
+ }?
115
+ },
116
+ ]?,
94
117
  session_attributes: Hash[::String, ::String]?
95
118
  }
96
119
  ) ?{ (*untyped) -> void } -> _InvokeAgentResponseSuccess
@@ -182,6 +205,27 @@ module Aws
182
205
  text: ::String
183
206
  },
184
207
  ?retrieve_and_generate_configuration: {
208
+ external_sources_configuration: {
209
+ generation_configuration: {
210
+ prompt_template: {
211
+ text_prompt_template: ::String?
212
+ }?
213
+ }?,
214
+ model_arn: ::String,
215
+ sources: Array[
216
+ {
217
+ byte_content: {
218
+ content_type: ::String,
219
+ data: ::String,
220
+ identifier: ::String
221
+ }?,
222
+ s3_location: {
223
+ uri: ::String
224
+ }?,
225
+ source_type: ("S3" | "BYTE_CONTENT")
226
+ },
227
+ ]
228
+ }?,
185
229
  knowledge_base_configuration: {
186
230
  generation_configuration: {
187
231
  prompt_template: {
@@ -250,7 +294,7 @@ module Aws
250
294
  }
251
295
  }?
252
296
  }?,
253
- type: ("KNOWLEDGE_BASE")
297
+ type: ("KNOWLEDGE_BASE" | "EXTERNAL_SOURCES")
254
298
  },
255
299
  ?session_configuration: {
256
300
  kms_key_arn: ::String
data/sig/types.rbs CHANGED
@@ -17,10 +17,11 @@ module Aws::BedrockAgentRuntime
17
17
  class ActionGroupInvocationInput
18
18
  attr_accessor action_group_name: ::String
19
19
  attr_accessor api_path: ::String
20
+ attr_accessor function: ::String
20
21
  attr_accessor parameters: ::Array[Types::Parameter]
21
22
  attr_accessor request_body: Types::RequestBody
22
23
  attr_accessor verb: ::String
23
- SENSITIVE: [:action_group_name, :api_path, :verb]
24
+ SENSITIVE: [:action_group_name, :api_path, :function, :verb]
24
25
  end
25
26
 
26
27
  class ActionGroupInvocationOutput
@@ -28,6 +29,37 @@ module Aws::BedrockAgentRuntime
28
29
  SENSITIVE: [:text]
29
30
  end
30
31
 
32
+ class ApiInvocationInput
33
+ attr_accessor action_group: ::String
34
+ attr_accessor api_path: ::String
35
+ attr_accessor http_method: ::String
36
+ attr_accessor parameters: ::Array[Types::ApiParameter]
37
+ attr_accessor request_body: Types::ApiRequestBody
38
+ SENSITIVE: [:api_path]
39
+ end
40
+
41
+ class ApiParameter
42
+ attr_accessor name: ::String
43
+ attr_accessor type: ::String
44
+ attr_accessor value: ::String
45
+ SENSITIVE: []
46
+ end
47
+
48
+ class ApiRequestBody
49
+ attr_accessor content: ::Hash[::String, Types::PropertyParameters]
50
+ SENSITIVE: []
51
+ end
52
+
53
+ class ApiResult
54
+ attr_accessor action_group: ::String
55
+ attr_accessor api_path: ::String
56
+ attr_accessor http_method: ::String
57
+ attr_accessor http_status_code: ::Integer
58
+ attr_accessor response_body: ::Hash[::String, Types::ContentBody]
59
+ attr_accessor response_state: ("FAILURE" | "REPROMPT")
60
+ SENSITIVE: [:api_path]
61
+ end
62
+
31
63
  class Attribution
32
64
  attr_accessor citations: ::Array[Types::Citation]
33
65
  SENSITIVE: []
@@ -40,6 +72,13 @@ module Aws::BedrockAgentRuntime
40
72
  SENSITIVE: []
41
73
  end
42
74
 
75
+ class ByteContentDoc
76
+ attr_accessor content_type: ::String
77
+ attr_accessor data: ::String
78
+ attr_accessor identifier: ::String
79
+ SENSITIVE: [:data, :identifier]
80
+ end
81
+
43
82
  class Citation
44
83
  attr_accessor generated_response_part: Types::GeneratedResponsePart
45
84
  attr_accessor retrieved_references: ::Array[Types::RetrievedReference]
@@ -52,6 +91,11 @@ module Aws::BedrockAgentRuntime
52
91
  SENSITIVE: []
53
92
  end
54
93
 
94
+ class ContentBody
95
+ attr_accessor body: ::String
96
+ SENSITIVE: []
97
+ end
98
+
55
99
  class DependencyFailedException
56
100
  attr_accessor message: ::String
57
101
  attr_accessor resource_name: ::String
@@ -59,6 +103,25 @@ module Aws::BedrockAgentRuntime
59
103
  SENSITIVE: []
60
104
  end
61
105
 
106
+ class ExternalSource
107
+ attr_accessor byte_content: Types::ByteContentDoc
108
+ attr_accessor s3_location: Types::S3ObjectDoc
109
+ attr_accessor source_type: ("S3" | "BYTE_CONTENT")
110
+ SENSITIVE: []
111
+ end
112
+
113
+ class ExternalSourcesGenerationConfiguration
114
+ attr_accessor prompt_template: Types::PromptTemplate
115
+ SENSITIVE: []
116
+ end
117
+
118
+ class ExternalSourcesRetrieveAndGenerateConfiguration
119
+ attr_accessor generation_configuration: Types::ExternalSourcesGenerationConfiguration
120
+ attr_accessor model_arn: ::String
121
+ attr_accessor sources: ::Array[Types::ExternalSource]
122
+ SENSITIVE: []
123
+ end
124
+
62
125
  class FailureTrace
63
126
  attr_accessor failure_reason: ::String
64
127
  attr_accessor trace_id: ::String
@@ -76,6 +139,28 @@ module Aws::BedrockAgentRuntime
76
139
  SENSITIVE: [:text]
77
140
  end
78
141
 
142
+ class FunctionInvocationInput
143
+ attr_accessor action_group: ::String
144
+ attr_accessor function: ::String
145
+ attr_accessor parameters: ::Array[Types::FunctionParameter]
146
+ SENSITIVE: []
147
+ end
148
+
149
+ class FunctionParameter
150
+ attr_accessor name: ::String
151
+ attr_accessor type: ::String
152
+ attr_accessor value: ::String
153
+ SENSITIVE: []
154
+ end
155
+
156
+ class FunctionResult
157
+ attr_accessor action_group: ::String
158
+ attr_accessor function: ::String
159
+ attr_accessor response_body: ::Hash[::String, Types::ContentBody]
160
+ attr_accessor response_state: ("FAILURE" | "REPROMPT")
161
+ SENSITIVE: []
162
+ end
163
+
79
164
  class GeneratedResponsePart
80
165
  attr_accessor text_response_part: Types::TextResponsePart
81
166
  SENSITIVE: [:text_response_part]
@@ -109,6 +194,34 @@ module Aws::BedrockAgentRuntime
109
194
  SENSITIVE: []
110
195
  end
111
196
 
197
+ class InvocationInputMember
198
+ attr_accessor api_invocation_input: Types::ApiInvocationInput
199
+ attr_accessor function_invocation_input: Types::FunctionInvocationInput
200
+ attr_accessor unknown: untyped
201
+ SENSITIVE: []
202
+
203
+ class ApiInvocationInput < InvocationInputMember
204
+ end
205
+ class FunctionInvocationInput < InvocationInputMember
206
+ end
207
+ class Unknown < InvocationInputMember
208
+ end
209
+ end
210
+
211
+ class InvocationResultMember
212
+ attr_accessor api_result: Types::ApiResult
213
+ attr_accessor function_result: Types::FunctionResult
214
+ attr_accessor unknown: untyped
215
+ SENSITIVE: []
216
+
217
+ class ApiResult < InvocationResultMember
218
+ end
219
+ class FunctionResult < InvocationResultMember
220
+ end
221
+ class Unknown < InvocationResultMember
222
+ end
223
+ end
224
+
112
225
  class InvokeAgentRequest
113
226
  attr_accessor agent_alias_id: ::String
114
227
  attr_accessor agent_id: ::String
@@ -282,6 +395,11 @@ module Aws::BedrockAgentRuntime
282
395
  SENSITIVE: [:text_prompt_template]
283
396
  end
284
397
 
398
+ class PropertyParameters
399
+ attr_accessor properties: ::Array[Types::Parameter]
400
+ SENSITIVE: []
401
+ end
402
+
285
403
  class Rationale
286
404
  attr_accessor text: ::String
287
405
  attr_accessor trace_id: ::String
@@ -363,8 +481,9 @@ module Aws::BedrockAgentRuntime
363
481
  end
364
482
 
365
483
  class RetrieveAndGenerateConfiguration
484
+ attr_accessor external_sources_configuration: Types::ExternalSourcesRetrieveAndGenerateConfiguration
366
485
  attr_accessor knowledge_base_configuration: Types::KnowledgeBaseRetrieveAndGenerateConfiguration
367
- attr_accessor type: ("KNOWLEDGE_BASE")
486
+ attr_accessor type: ("KNOWLEDGE_BASE" | "EXTERNAL_SOURCES")
368
487
  SENSITIVE: []
369
488
  end
370
489
 
@@ -419,6 +538,18 @@ module Aws::BedrockAgentRuntime
419
538
  SENSITIVE: [:content, :location, :metadata]
420
539
  end
421
540
 
541
+ class ReturnControlPayload
542
+ attr_accessor invocation_id: ::String
543
+ attr_accessor invocation_inputs: ::Array[Types::InvocationInputMember]
544
+ attr_accessor event_type: untyped
545
+ SENSITIVE: []
546
+ end
547
+
548
+ class S3ObjectDoc
549
+ attr_accessor uri: ::String
550
+ SENSITIVE: []
551
+ end
552
+
422
553
  class ServiceQuotaExceededException
423
554
  attr_accessor message: ::String
424
555
  attr_accessor event_type: untyped
@@ -426,7 +557,9 @@ module Aws::BedrockAgentRuntime
426
557
  end
427
558
 
428
559
  class SessionState
560
+ attr_accessor invocation_id: ::String
429
561
  attr_accessor prompt_session_attributes: ::Hash[::String, ::String]
562
+ attr_accessor return_control_invocation_results: ::Array[Types::InvocationResultMember]
430
563
  attr_accessor session_attributes: ::Hash[::String, ::String]
431
564
  SENSITIVE: []
432
565
  end
@@ -472,6 +605,7 @@ module Aws::BedrockAgentRuntime
472
605
  class TracePart
473
606
  attr_accessor agent_alias_id: ::String
474
607
  attr_accessor agent_id: ::String
608
+ attr_accessor agent_version: ::String
475
609
  attr_accessor session_id: ::String
476
610
  attr_accessor trace: Types::Trace
477
611
  attr_accessor event_type: untyped
@@ -485,7 +619,7 @@ module Aws::BedrockAgentRuntime
485
619
  end
486
620
 
487
621
  class ResponseStream < Enumerator[untyped, untyped]
488
- def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
622
+ def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
489
623
  end
490
624
  end
491
625
  end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: aws-sdk-bedrockagentruntime
3
3
  version: !ruby/object:Gem::Version
4
- version: 1.5.0
4
+ version: 1.7.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - Amazon Web Services
8
8
  autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2024-03-27 00:00:00.000000000 Z
11
+ date: 2024-04-25 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: aws-sdk-core
@@ -19,7 +19,7 @@ dependencies:
19
19
  version: '3'
20
20
  - - ">="
21
21
  - !ruby/object:Gem::Version
22
- version: 3.191.0
22
+ version: 3.193.0
23
23
  type: :runtime
24
24
  prerelease: false
25
25
  version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
29
29
  version: '3'
30
30
  - - ">="
31
31
  - !ruby/object:Gem::Version
32
- version: 3.191.0
32
+ version: 3.193.0
33
33
  - !ruby/object:Gem::Dependency
34
34
  name: aws-sigv4
35
35
  requirement: !ruby/object:Gem::Requirement