vibe-client 0.1.0__py3-none-any.whl

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 (11) hide show
  1. vibe_client/__init__.py +3 -0
  2. vibe_client/_private/__init__.py +1 -0
  3. vibe_client/_private/schemas.py +352 -0
  4. vibe_client/client.py +498 -0
  5. vibe_client/config.py +74 -0
  6. vibe_client/deserialize.py +115 -0
  7. vibe_client/models.py +710 -0
  8. vibe_client/serialize.py +53 -0
  9. vibe_client-0.1.0.dist-info/METADATA +128 -0
  10. vibe_client-0.1.0.dist-info/RECORD +11 -0
  11. vibe_client-0.1.0.dist-info/WHEEL +4 -0
vibe_client/__init__.py ADDED
@@ -0,0 +1,3 @@
1
+ # Code generated by smithy-python-codegen DO NOT EDIT.
2
+
3
+ __version__: str = "0.1.0"
vibe_client/_private/__init__.py ADDED
@@ -0,0 +1 @@
1
+ # Code generated by smithy-python-codegen DO NOT EDIT.
vibe_client/_private/schemas.py ADDED
@@ -0,0 +1,352 @@
1
+ # Code generated by smithy-python-codegen DO NOT EDIT.
2
+
3
+ from types import MappingProxyType
4
+
5
+ from smithy_core.prelude import BOOLEAN, FLOAT, INTEGER, STRING
6
+ from smithy_core.schemas import Schema
7
+ from smithy_core.shapes import ShapeID, ShapeType
8
+ from smithy_core.traits import Trait
9
+
10
+
11
+ VALIDATION_EXCEPTION_FIELD = Schema.collection(
12
+ id=ShapeID("smithy.framework#ValidationExceptionField"),
13
+
14
+ members={
15
+ "path": {
16
+ "target": STRING,
17
+ "index": 0,
18
+ "traits": [
19
+ Trait.new(id=ShapeID("smithy.api#required")),
20
+
21
+ ],
22
+ },
23
+
24
+ "message": {
25
+ "target": STRING,
26
+ "index": 1,
27
+ "traits": [
28
+ Trait.new(id=ShapeID("smithy.api#required")),
29
+
30
+ ],
31
+ },
32
+
33
+ }
34
+ )
35
+
36
+ VALIDATION_EXCEPTION_FIELD_LIST = Schema.collection(
37
+ id=ShapeID("smithy.framework#ValidationExceptionFieldList"),
38
+ shape_type=ShapeType.LIST,
39
+ members={
40
+ "member": {
41
+ "target": VALIDATION_EXCEPTION_FIELD,
42
+ "index": 0,
43
+ },
44
+
45
+ }
46
+ )
47
+
48
+ VALIDATION_EXCEPTION = Schema.collection(
49
+ id=ShapeID("smithy.framework#ValidationException"),
50
+
51
+ traits=[
52
+ Trait.new(id=ShapeID("smithy.api#error"), value="client"),
53
+
54
+ ],
55
+ members={
56
+ "message": {
57
+ "target": STRING,
58
+ "index": 0,
59
+ "traits": [
60
+ Trait.new(id=ShapeID("smithy.api#required")),
61
+
62
+ ],
63
+ },
64
+
65
+ "fieldList": {
66
+ "target": VALIDATION_EXCEPTION_FIELD_LIST,
67
+ "index": 1,
68
+ },
69
+
70
+ }
71
+ )
72
+
73
+ EXPERIMENT_IDENTIFIER = Schema(
74
+ id=ShapeID("vibe.astar.public.api.shared#ExperimentIdentifier"),
75
+ shape_type=ShapeType.STRING,
76
+ traits=[
77
+ Trait.new(id=ShapeID("smithy.api#pattern"), value="^[A-Za-z0-9-]+$"),
78
+
79
+ ],
80
+
81
+ )
82
+
83
+ FLOAT_LIST = Schema.collection(
84
+ id=ShapeID("vibe.astar.public.api.operations#FloatList"),
85
+ shape_type=ShapeType.LIST,
86
+ members={
87
+ "member": {
88
+ "target": FLOAT,
89
+ "index": 0,
90
+ },
91
+
92
+ }
93
+ )
94
+
95
+ FLOAT_LIST_LIST = Schema.collection(
96
+ id=ShapeID("vibe.astar.public.api.operations#FloatListList"),
97
+ shape_type=ShapeType.LIST,
98
+ members={
99
+ "member": {
100
+ "target": FLOAT_LIST,
101
+ "index": 0,
102
+ },
103
+
104
+ }
105
+ )
106
+
107
+ BOOLEAN_LIST = Schema.collection(
108
+ id=ShapeID("vibe.astar.public.api.operations#BooleanList"),
109
+ shape_type=ShapeType.LIST,
110
+ members={
111
+ "member": {
112
+ "target": BOOLEAN,
113
+ "index": 0,
114
+ },
115
+
116
+ }
117
+ )
118
+
119
+ INTEGER_LIST = Schema.collection(
120
+ id=ShapeID("vibe.astar.public.api.shared#IntegerList"),
121
+ shape_type=ShapeType.LIST,
122
+ members={
123
+ "member": {
124
+ "target": INTEGER,
125
+ "index": 0,
126
+ },
127
+
128
+ }
129
+ )
130
+
131
+ UNAUTHORIZED_EXCEPTION = Schema.collection(
132
+ id=ShapeID("vibe.astar.public.api.shared#UnauthorizedException"),
133
+
134
+ traits=[
135
+ Trait.new(id=ShapeID("smithy.api#error"), value="client"),
136
+ Trait.new(id=ShapeID("smithy.api#httpError"), value=401),
137
+
138
+ ],
139
+ members={
140
+ "message": {
141
+ "target": STRING,
142
+ "index": 0,
143
+ "traits": [
144
+ Trait.new(id=ShapeID("smithy.api#required")),
145
+
146
+ ],
147
+ },
148
+
149
+ }
150
+ )
151
+
152
+ VIBE_VALIDATION_EXCEPTION = Schema.collection(
153
+ id=ShapeID("vibe.astar.public.api.shared#VibeValidationException"),
154
+
155
+ traits=[
156
+ Trait.new(id=ShapeID("smithy.api#error"), value="client"),
157
+ Trait.new(id=ShapeID("smithy.api#httpError"), value=400),
158
+
159
+ ],
160
+ members={
161
+ "message": {
162
+ "target": STRING,
163
+ "index": 0,
164
+ "traits": [
165
+ Trait.new(id=ShapeID("smithy.api#required")),
166
+
167
+ ],
168
+ },
169
+
170
+ }
171
+ )
172
+
173
+ OBSERVATION_VALUE = Schema.collection(
174
+ id=ShapeID("vibe.astar.public.api.operations#ObservationValue"),
175
+ shape_type=ShapeType.UNION,
176
+ members={
177
+ "discrete": {
178
+ "target": INTEGER,
179
+ "index": 0,
180
+ },
181
+
182
+ "continuous": {
183
+ "target": FLOAT_LIST,
184
+ "index": 1,
185
+ },
186
+
187
+ "multiDiscrete": {
188
+ "target": INTEGER_LIST,
189
+ "index": 2,
190
+ },
191
+
192
+ "multiBinary": {
193
+ "target": BOOLEAN_LIST,
194
+ "index": 3,
195
+ },
196
+
197
+ "box": {
198
+ "target": FLOAT_LIST_LIST,
199
+ "index": 4,
200
+ },
201
+
202
+ }
203
+ )
204
+
205
+ VALUE_MAP = Schema.collection(
206
+ id=ShapeID("vibe.astar.public.api.operations#ValueMap"),
207
+ shape_type=ShapeType.MAP,
208
+ members={
209
+ "key": {
210
+ "target": STRING,
211
+ "index": 0,
212
+ },
213
+
214
+ "value": {
215
+ "target": OBSERVATION_VALUE,
216
+ "index": 1,
217
+ },
218
+
219
+ }
220
+ )
221
+
222
+ ACTION_VALUE = Schema.collection(
223
+ id=ShapeID("vibe.astar.public.api.operations#ActionValue"),
224
+ shape_type=ShapeType.UNION,
225
+ members={
226
+ "discrete": {
227
+ "target": INTEGER,
228
+ "index": 0,
229
+ },
230
+
231
+ "continuous": {
232
+ "target": FLOAT_LIST,
233
+ "index": 1,
234
+ },
235
+
236
+ "multiDiscrete": {
237
+ "target": INTEGER_LIST,
238
+ "index": 2,
239
+ },
240
+
241
+ "multiBinary": {
242
+ "target": BOOLEAN_LIST,
243
+ "index": 3,
244
+ },
245
+
246
+ "box": {
247
+ "target": FLOAT_LIST_LIST,
248
+ "index": 4,
249
+ },
250
+
251
+ "dict": {
252
+ "target": VALUE_MAP,
253
+ "index": 5,
254
+ },
255
+
256
+ }
257
+ )
258
+
259
+ QUERY_AGENT_INPUT = Schema.collection(
260
+ id=ShapeID("vibe.astar.public.api.operations#QueryAgentInput"),
261
+
262
+ traits=[
263
+ Trait.new(id=ShapeID("smithy.api#input")),
264
+
265
+ ],
266
+ members={
267
+ "experimentId": {
268
+ "target": EXPERIMENT_IDENTIFIER,
269
+ "index": 0,
270
+ "traits": [
271
+ Trait.new(id=ShapeID("smithy.api#required")),
272
+ Trait.new(id=ShapeID("smithy.api#httpLabel")),
273
+
274
+ ],
275
+ },
276
+
277
+ "observations": {
278
+ "target": OBSERVATION_VALUE,
279
+ "index": 1,
280
+ "traits": [
281
+ Trait.new(id=ShapeID("smithy.api#required")),
282
+
283
+ ],
284
+ },
285
+
286
+ }
287
+ )
288
+
289
+ QUERY_AGENT_OUTPUT = Schema.collection(
290
+ id=ShapeID("vibe.astar.public.api.operations#QueryAgentOutput"),
291
+
292
+ traits=[
293
+ Trait.new(id=ShapeID("smithy.api#output")),
294
+
295
+ ],
296
+ members={
297
+ "actions": {
298
+ "target": ACTION_VALUE,
299
+ "index": 0,
300
+ "traits": [
301
+ Trait.new(id=ShapeID("smithy.api#required")),
302
+
303
+ ],
304
+ },
305
+
306
+ }
307
+ )
308
+
309
+ QUERY_AGENT = Schema(
310
+ id=ShapeID("vibe.astar.public.api.operations#QueryAgent"),
311
+ shape_type=ShapeType.OPERATION,
312
+ traits=[
313
+ Trait.new(id=ShapeID("smithy.api#http"), value=MappingProxyType({
314
+ "code": 200,
315
+ "method": "POST",
316
+ "uri": "/agents/{experimentId}/query",
317
+ })),
318
+
319
+ ],
320
+
321
+ )
322
+
323
+ VIBE = Schema(
324
+ id=ShapeID("vibe.astar.public.api#Vibe"),
325
+ shape_type=ShapeType.SERVICE,
326
+ traits=[
327
+ Trait.new(id=ShapeID("smithy.api#title"), value="A* tech Vibe's publicly facing API model"),
328
+ Trait.new(id=ShapeID("smithy.api#cors"), value=MappingProxyType({
329
+ "additionalAllowedHeaders": (
330
+ "Content-Type",
331
+ "X-Api-Key",
332
+ "X-Amz-User-Agent",
333
+ "Amz-Sdk-Invocation-Id",
334
+ "Amz-Sdk-Request",
335
+ ),
336
+ "additionalExposedHeaders": (
337
+ "Content-Type",
338
+ "X-Api-Key",
339
+ "X-Amz-User-Agent",
340
+ ),
341
+ })),
342
+ Trait.new(id=ShapeID("aws.protocols#restJson1")),
343
+
344
+ ],
345
+
346
+ )
347
+
348
+ OBSERVATION_VALUE.members["dict"] = Schema.member(
349
+ id=OBSERVATION_VALUE.id.with_member("dict"),
350
+ target=VALUE_MAP,
351
+ index=5,
352
+ )