1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
open Core
open Async
let json_arg = Command.Arg_type.create Yojson.Safe.from_string
let call ?endpoint_url ?profile ?region f m result_to_json error_to_json =
let region =
match region with
| Some region -> Some (Awso.Region.of_string region)
| None -> None in
(Awso_async.Cfg.get_exn ?profile ?region ()) >>=
(fun cfg ->
(f ?endpoint_url ?cfg:(Some cfg) m) >>=
(fun result ->
match result with
| Error err ->
(match error_to_json with
| None ->
failwithf
"endpoint error, but no error values defined in boto"
()
| Some to_json ->
let s = (err |> to_json) |> Yojson.Safe.to_string in
failwithf "AWS error: %s" s ())
| Ok result ->
((match result_to_json with
| None -> print_endline "ok response from endpoint"
| Some to_json ->
((result |> to_json) |> Yojson.Safe.to_string) |>
print_endline);
return ())))
let invoke_endpoint =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and contentType =
flag "content-type" (optional string) ~doc:"STRING Header"
and accept = flag "accept" (optional string) ~doc:"STRING Header"
and customAttributes =
flag "custom-attributes" (optional string)
~doc:"STRING CustomAttributesHeader"
and targetModel =
flag "target-model" (optional string)
~doc:"STRING TargetModelHeader"
and targetVariant =
flag "target-variant" (optional string)
~doc:"STRING TargetVariantHeader"
and targetContainerHostname =
flag "target-container-hostname" (optional string)
~doc:"STRING TargetContainerHostnameHeader"
and inferenceId =
flag "inference-id" (optional string) ~doc:"STRING InferenceId"
and enableExplanations =
flag "enable-explanations" (optional string)
~doc:"STRING EnableExplanationsHeader"
and inferenceComponentName =
flag "inference-component-name" (optional string)
~doc:"STRING InferenceComponentHeader"
and sessionId =
flag "session-id" (optional string)
~doc:"STRING SessionIdOrNewSessionConstantHeader"
and endpointName =
flag "endpoint-name" (required string) ~doc:"STRING EndpointName"
and body = flag "body" (required json_arg) ~doc:"JSON BodyBlob" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.invoke_endpoint
(Values.InvokeEndpointInput.make ?contentType ?accept
?customAttributes ?targetModel ?targetVariant
?targetContainerHostname ?inferenceId ?enableExplanations
?inferenceComponentName ?sessionId ~endpointName
~body:(Values.BodyBlob.of_json body) ())
(Some Values.InvokeEndpointOutput.to_json)
(Some Values.InvokeEndpointOutput.error_to_json)])
let invoke_endpoint_async =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and contentType =
flag "content-type" (optional string) ~doc:"STRING Header"
and accept = flag "accept" (optional string) ~doc:"STRING Header"
and customAttributes =
flag "custom-attributes" (optional string)
~doc:"STRING CustomAttributesHeader"
and inferenceId =
flag "inference-id" (optional string) ~doc:"STRING InferenceId"
and s3OutputPathExtension =
flag "s3-output-path-extension" (optional string)
~doc:"STRING S3OutputPathExtensionHeader"
and filename =
flag "filename" (optional string) ~doc:"STRING FilenameHeader"
and requestTTLSeconds =
flag "request-t-t-l-seconds" (optional int)
~doc:"INT RequestTTLSecondsHeader"
and invocationTimeoutSeconds =
flag "invocation-timeout-seconds" (optional int)
~doc:"INT InvocationTimeoutSecondsHeader"
and endpointName =
flag "endpoint-name" (required string) ~doc:"STRING EndpointName"
and inputLocation =
flag "input-location" (required string)
~doc:"STRING InputLocationHeader" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.invoke_endpoint_async
(Values.InvokeEndpointAsyncInput.make ?contentType ?accept
?customAttributes ?inferenceId ?s3OutputPathExtension ?filename
?requestTTLSeconds ?invocationTimeoutSeconds ~endpointName
~inputLocation ())
(Some Values.InvokeEndpointAsyncOutput.to_json)
(Some Values.InvokeEndpointAsyncOutput.error_to_json)])
let invoke_endpoint_with_response_stream =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and contentType =
flag "content-type" (optional string) ~doc:"STRING Header"
and accept = flag "accept" (optional string) ~doc:"STRING Header"
and customAttributes =
flag "custom-attributes" (optional string)
~doc:"STRING CustomAttributesHeader"
and targetVariant =
flag "target-variant" (optional string)
~doc:"STRING TargetVariantHeader"
and targetContainerHostname =
flag "target-container-hostname" (optional string)
~doc:"STRING TargetContainerHostnameHeader"
and inferenceId =
flag "inference-id" (optional string) ~doc:"STRING InferenceId"
and inferenceComponentName =
flag "inference-component-name" (optional string)
~doc:"STRING InferenceComponentHeader"
and sessionId =
flag "session-id" (optional string) ~doc:"STRING SessionIdHeader"
and endpointName =
flag "endpoint-name" (required string) ~doc:"STRING EndpointName"
and body = flag "body" (required json_arg) ~doc:"JSON BodyBlob" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.invoke_endpoint_with_response_stream
(Values.InvokeEndpointWithResponseStreamInput.make ?contentType
?accept ?customAttributes ?targetVariant
?targetContainerHostname ?inferenceId ?inferenceComponentName
?sessionId ~endpointName ~body:(Values.BodyBlob.of_json body)
())
(Some Values.InvokeEndpointWithResponseStreamOutput.to_json)
(Some Values.InvokeEndpointWithResponseStreamOutput.error_to_json)])
let main =
Command.group
~summary:((Awso.Service.to_string Values.service) ^ " commands")
[("invoke-endpoint", invoke_endpoint);
("invoke-endpoint-async", invoke_endpoint_async);
("invoke-endpoint-with-response-stream",
invoke_endpoint_with_response_stream)]