Source file cli.ml

(* generated by: awso-codegen generate-all --botocore-data vendor/botocore/botocore/data -o aws --runtime-dir lib/runtime/awso --cli-dir awso-cli *)
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 create_invocation =
  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 description =
         flag "description" (optional string)
           ~doc:"STRING InvocationDescription"
       and invocationId =
         flag "invocation-id" (optional string) ~doc:"STRING Uuid"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.create_invocation
           (Values.CreateInvocationRequest.make ?description ?invocationId
              ~sessionIdentifier ())
           (Some Values.CreateInvocationResponse.to_json)
           (Some Values.CreateInvocationResponse.error_to_json)])
let create_session =
  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 encryptionKeyArn =
         flag "encryption-key-arn" (optional string) ~doc:"STRING KmsKeyArn"
       and sessionMetadata =
         flag "session-metadata" (optional json_arg)
           ~doc:"JSON SessionMetadataMap"
       and tags = flag "tags" (optional json_arg) ~doc:"JSON TagsMap" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.create_session
           (Values.CreateSessionRequest.make ?encryptionKeyArn
              ?sessionMetadata:(Option.map
                                  ~f:Values.SessionMetadataMap.of_json
                                  sessionMetadata)
              ?tags:(Option.map ~f:Values.TagsMap.of_json tags) ())
           (Some Values.CreateSessionResponse.to_json)
           (Some Values.CreateSessionResponse.error_to_json)])
let delete_agent_memory =
  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 memoryId =
         flag "memory-id" (optional string) ~doc:"STRING MemoryId"
       and sessionId =
         flag "session-id" (optional string) ~doc:"STRING SessionId"
       and agentAliasId =
         flag "agent-alias-id" (required string) ~doc:"STRING AgentAliasId"
       and agentId = flag "agent-id" (required string) ~doc:"STRING AgentId" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.delete_agent_memory
           (Values.DeleteAgentMemoryRequest.make ?memoryId ?sessionId
              ~agentAliasId ~agentId ())
           (Some Values.DeleteAgentMemoryResponse.to_json)
           (Some Values.DeleteAgentMemoryResponse.error_to_json)])
let delete_session =
  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 sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.delete_session
           (Values.DeleteSessionRequest.make ~sessionIdentifier ())
           (Some Values.DeleteSessionResponse.to_json)
           (Some Values.DeleteSessionResponse.error_to_json)])
let end_session =
  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 sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.end_session
           (Values.EndSessionRequest.make ~sessionIdentifier ())
           (Some Values.EndSessionResponse.to_json)
           (Some Values.EndSessionResponse.error_to_json)])
let generate_query =
  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 queryGenerationInput =
         flag "query-generation-input" (required json_arg)
           ~doc:"JSON QueryGenerationInput"
       and transformationConfiguration =
         flag "transformation-configuration" (required json_arg)
           ~doc:"JSON TransformationConfiguration" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.generate_query
           (Values.GenerateQueryRequest.make
              ~queryGenerationInput:(Values.QueryGenerationInput.of_json
                                       queryGenerationInput)
              ~transformationConfiguration:(Values.TransformationConfiguration.of_json
                                              transformationConfiguration) ())
           (Some Values.GenerateQueryResponse.to_json)
           (Some Values.GenerateQueryResponse.error_to_json)])
let get_agent_memory =
  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 maxItems = flag "max-items" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and agentAliasId =
         flag "agent-alias-id" (required string) ~doc:"STRING AgentAliasId"
       and agentId = flag "agent-id" (required string) ~doc:"STRING AgentId"
       and memoryId =
         flag "memory-id" (required string) ~doc:"STRING MemoryId"
       and memoryType =
         flag "memory-type" (required json_arg) ~doc:"JSON MemoryType" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_agent_memory
           (Values.GetAgentMemoryRequest.make ?maxItems ?nextToken
              ~agentAliasId ~agentId ~memoryId
              ~memoryType:(Values.MemoryType.of_json memoryType) ())
           (Some Values.GetAgentMemoryResponse.to_json)
           (Some Values.GetAgentMemoryResponse.error_to_json)])
let get_execution_flow_snapshot =
  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 executionIdentifier =
         flag "execution-identifier" (required string)
           ~doc:"STRING FlowExecutionIdentifier"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_execution_flow_snapshot
           (Values.GetExecutionFlowSnapshotRequest.make ~executionIdentifier
              ~flowAliasIdentifier ~flowIdentifier ())
           (Some Values.GetExecutionFlowSnapshotResponse.to_json)
           (Some Values.GetExecutionFlowSnapshotResponse.error_to_json)])
let get_flow_execution =
  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 executionIdentifier =
         flag "execution-identifier" (required string)
           ~doc:"STRING FlowExecutionIdentifier"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_flow_execution
           (Values.GetFlowExecutionRequest.make ~executionIdentifier
              ~flowAliasIdentifier ~flowIdentifier ())
           (Some Values.GetFlowExecutionResponse.to_json)
           (Some Values.GetFlowExecutionResponse.error_to_json)])
let get_invocation_step =
  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 invocationIdentifier =
         flag "invocation-identifier" (required string)
           ~doc:"STRING InvocationIdentifier"
       and invocationStepId =
         flag "invocation-step-id" (required string) ~doc:"STRING Uuid"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_invocation_step
           (Values.GetInvocationStepRequest.make ~invocationIdentifier
              ~invocationStepId ~sessionIdentifier ())
           (Some Values.GetInvocationStepResponse.to_json)
           (Some Values.GetInvocationStepResponse.error_to_json)])
let get_session =
  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 sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_session
           (Values.GetSessionRequest.make ~sessionIdentifier ())
           (Some Values.GetSessionResponse.to_json)
           (Some Values.GetSessionResponse.error_to_json)])
let invoke_agent =
  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 bedrockModelConfigurations =
         flag "bedrock-model-configurations" (optional json_arg)
           ~doc:"JSON BedrockModelConfigurations"
       and enableTrace =
         flag "enable-trace" (optional bool) ~doc:"BOOL Boolean"
       and endSession =
         flag "end-session" (optional bool) ~doc:"BOOL Boolean"
       and inputText =
         flag "input-text" (optional string) ~doc:"STRING InputText"
       and memoryId =
         flag "memory-id" (optional string) ~doc:"STRING MemoryId"
       and promptCreationConfigurations =
         flag "prompt-creation-configurations" (optional json_arg)
           ~doc:"JSON PromptCreationConfigurations"
       and sessionState =
         flag "session-state" (optional json_arg) ~doc:"JSON SessionState"
       and sourceArn =
         flag "source-arn" (optional string) ~doc:"STRING AWSResourceARN"
       and streamingConfigurations =
         flag "streaming-configurations" (optional json_arg)
           ~doc:"JSON StreamingConfigurations"
       and agentAliasId =
         flag "agent-alias-id" (required string) ~doc:"STRING AgentAliasId"
       and agentId = flag "agent-id" (required string) ~doc:"STRING AgentId"
       and sessionId =
         flag "session-id" (required string) ~doc:"STRING SessionId" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.invoke_agent
           (Values.InvokeAgentRequest.make
              ?bedrockModelConfigurations:(Option.map
                                             ~f:Values.BedrockModelConfigurations.of_json
                                             bedrockModelConfigurations)
              ?enableTrace ?endSession ?inputText ?memoryId
              ?promptCreationConfigurations:(Option.map
                                               ~f:Values.PromptCreationConfigurations.of_json
                                               promptCreationConfigurations)
              ?sessionState:(Option.map ~f:Values.SessionState.of_json
                               sessionState) ?sourceArn
              ?streamingConfigurations:(Option.map
                                          ~f:Values.StreamingConfigurations.of_json
                                          streamingConfigurations)
              ~agentAliasId ~agentId ~sessionId ())
           (Some Values.InvokeAgentResponse.to_json)
           (Some Values.InvokeAgentResponse.error_to_json)])
let invoke_flow =
  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 enableTrace =
         flag "enable-trace" (optional bool) ~doc:"BOOL Boolean"
       and executionId =
         flag "execution-id" (optional string) ~doc:"STRING FlowExecutionId"
       and modelPerformanceConfiguration =
         flag "model-performance-configuration" (optional json_arg)
           ~doc:"JSON ModelPerformanceConfiguration"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier"
       and inputs = flag "inputs" (required json_arg) ~doc:"JSON FlowInputs" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.invoke_flow
           (Values.InvokeFlowRequest.make ?enableTrace ?executionId
              ?modelPerformanceConfiguration:(Option.map
                                                ~f:Values.ModelPerformanceConfiguration.of_json
                                                modelPerformanceConfiguration)
              ~flowAliasIdentifier ~flowIdentifier
              ~inputs:(Values.FlowInputs.of_json inputs) ())
           (Some Values.InvokeFlowResponse.to_json)
           (Some Values.InvokeFlowResponse.error_to_json)])
let invoke_inline_agent =
  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 actionGroups =
         flag "action-groups" (optional json_arg)
           ~doc:"JSON AgentActionGroups"
       and agentCollaboration =
         flag "agent-collaboration" (optional json_arg)
           ~doc:"JSON AgentCollaboration"
       and agentName = flag "agent-name" (optional string) ~doc:"STRING Name"
       and bedrockModelConfigurations =
         flag "bedrock-model-configurations" (optional json_arg)
           ~doc:"JSON InlineBedrockModelConfigurations"
       and collaboratorConfigurations =
         flag "collaborator-configurations" (optional json_arg)
           ~doc:"JSON CollaboratorConfigurations"
       and collaborators =
         flag "collaborators" (optional json_arg) ~doc:"JSON Collaborators"
       and customOrchestration =
         flag "custom-orchestration" (optional json_arg)
           ~doc:"JSON CustomOrchestration"
       and customerEncryptionKeyArn =
         flag "customer-encryption-key-arn" (optional string)
           ~doc:"STRING KmsKeyArn"
       and enableTrace =
         flag "enable-trace" (optional bool) ~doc:"BOOL Boolean"
       and endSession =
         flag "end-session" (optional bool) ~doc:"BOOL Boolean"
       and guardrailConfiguration =
         flag "guardrail-configuration" (optional json_arg)
           ~doc:"JSON GuardrailConfigurationWithArn"
       and idleSessionTTLInSeconds =
         flag "idle-session-t-t-l-in-seconds" (optional int)
           ~doc:"INT SessionTTL"
       and inlineSessionState =
         flag "inline-session-state" (optional json_arg)
           ~doc:"JSON InlineSessionState"
       and inputText =
         flag "input-text" (optional string) ~doc:"STRING InputText"
       and knowledgeBases =
         flag "knowledge-bases" (optional json_arg)
           ~doc:"JSON KnowledgeBases"
       and orchestrationType =
         flag "orchestration-type" (optional json_arg)
           ~doc:"JSON OrchestrationType"
       and promptCreationConfigurations =
         flag "prompt-creation-configurations" (optional json_arg)
           ~doc:"JSON PromptCreationConfigurations"
       and promptOverrideConfiguration =
         flag "prompt-override-configuration" (optional json_arg)
           ~doc:"JSON PromptOverrideConfiguration"
       and streamingConfigurations =
         flag "streaming-configurations" (optional json_arg)
           ~doc:"JSON StreamingConfigurations"
       and foundationModel =
         flag "foundation-model" (required string)
           ~doc:"STRING ModelIdentifier"
       and instruction =
         flag "instruction" (required string) ~doc:"STRING Instruction"
       and sessionId =
         flag "session-id" (required string) ~doc:"STRING SessionId" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.invoke_inline_agent
           (Values.InvokeInlineAgentRequest.make
              ?actionGroups:(Option.map ~f:Values.AgentActionGroups.of_json
                               actionGroups)
              ?agentCollaboration:(Option.map
                                     ~f:Values.AgentCollaboration.of_json
                                     agentCollaboration) ?agentName
              ?bedrockModelConfigurations:(Option.map
                                             ~f:Values.InlineBedrockModelConfigurations.of_json
                                             bedrockModelConfigurations)
              ?collaboratorConfigurations:(Option.map
                                             ~f:Values.CollaboratorConfigurations.of_json
                                             collaboratorConfigurations)
              ?collaborators:(Option.map ~f:Values.Collaborators.of_json
                                collaborators)
              ?customOrchestration:(Option.map
                                      ~f:Values.CustomOrchestration.of_json
                                      customOrchestration)
              ?customerEncryptionKeyArn ?enableTrace ?endSession
              ?guardrailConfiguration:(Option.map
                                         ~f:Values.GuardrailConfigurationWithArn.of_json
                                         guardrailConfiguration)
              ?idleSessionTTLInSeconds
              ?inlineSessionState:(Option.map
                                     ~f:Values.InlineSessionState.of_json
                                     inlineSessionState) ?inputText
              ?knowledgeBases:(Option.map ~f:Values.KnowledgeBases.of_json
                                 knowledgeBases)
              ?orchestrationType:(Option.map
                                    ~f:Values.OrchestrationType.of_json
                                    orchestrationType)
              ?promptCreationConfigurations:(Option.map
                                               ~f:Values.PromptCreationConfigurations.of_json
                                               promptCreationConfigurations)
              ?promptOverrideConfiguration:(Option.map
                                              ~f:Values.PromptOverrideConfiguration.of_json
                                              promptOverrideConfiguration)
              ?streamingConfigurations:(Option.map
                                          ~f:Values.StreamingConfigurations.of_json
                                          streamingConfigurations)
              ~foundationModel ~instruction ~sessionId ())
           (Some Values.InvokeInlineAgentResponse.to_json)
           (Some Values.InvokeInlineAgentResponse.error_to_json)])
let list_flow_execution_events =
  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 maxResults =
         flag "max-results" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and eventType =
         flag "event-type" (required json_arg)
           ~doc:"JSON FlowExecutionEventType"
       and executionIdentifier =
         flag "execution-identifier" (required string)
           ~doc:"STRING FlowExecutionIdentifier"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_flow_execution_events
           (Values.ListFlowExecutionEventsRequest.make ?maxResults ?nextToken
              ~eventType:(Values.FlowExecutionEventType.of_json eventType)
              ~executionIdentifier ~flowAliasIdentifier ~flowIdentifier ())
           (Some Values.ListFlowExecutionEventsResponse.to_json)
           (Some Values.ListFlowExecutionEventsResponse.error_to_json)])
let list_flow_executions =
  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 flowAliasIdentifier =
         flag "flow-alias-identifier" (optional string)
           ~doc:"STRING FlowAliasIdentifier"
       and maxResults =
         flag "max-results" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_flow_executions
           (Values.ListFlowExecutionsRequest.make ?flowAliasIdentifier
              ?maxResults ?nextToken ~flowIdentifier ())
           (Some Values.ListFlowExecutionsResponse.to_json)
           (Some Values.ListFlowExecutionsResponse.error_to_json)])
let list_invocation_steps =
  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 invocationIdentifier =
         flag "invocation-identifier" (optional string)
           ~doc:"STRING InvocationIdentifier"
       and maxResults =
         flag "max-results" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_invocation_steps
           (Values.ListInvocationStepsRequest.make ?invocationIdentifier
              ?maxResults ?nextToken ~sessionIdentifier ())
           (Some Values.ListInvocationStepsResponse.to_json)
           (Some Values.ListInvocationStepsResponse.error_to_json)])
let list_invocations =
  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 maxResults =
         flag "max-results" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_invocations
           (Values.ListInvocationsRequest.make ?maxResults ?nextToken
              ~sessionIdentifier ())
           (Some Values.ListInvocationsResponse.to_json)
           (Some Values.ListInvocationsResponse.error_to_json)])
let list_sessions =
  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 maxResults =
         flag "max-results" (optional int) ~doc:"INT MaxResults"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_sessions
           (Values.ListSessionsRequest.make ?maxResults ?nextToken ())
           (Some Values.ListSessionsResponse.to_json)
           (Some Values.ListSessionsResponse.error_to_json)])
let list_tags_for_resource =
  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 resourceArn =
         flag "resource-arn" (required string)
           ~doc:"STRING TaggableResourcesArn" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_tags_for_resource
           (Values.ListTagsForResourceRequest.make ~resourceArn ())
           (Some Values.ListTagsForResourceResponse.to_json)
           (Some Values.ListTagsForResourceResponse.error_to_json)])
let optimize_prompt =
  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 input = flag "input" (required json_arg) ~doc:"JSON InputPrompt"
       and targetModelId =
         flag "target-model-id" (required string)
           ~doc:"STRING OptimizePromptRequestTargetModelIdString" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.optimize_prompt
           (Values.OptimizePromptRequest.make
              ~input:(Values.InputPrompt.of_json input) ~targetModelId ())
           (Some Values.OptimizePromptResponse.to_json)
           (Some Values.OptimizePromptResponse.error_to_json)])
let put_invocation_step =
  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 invocationStepId =
         flag "invocation-step-id" (optional string) ~doc:"STRING Uuid"
       and invocationIdentifier =
         flag "invocation-identifier" (required string)
           ~doc:"STRING InvocationIdentifier"
       and invocationStepTime =
         flag "invocation-step-time" (required json_arg)
           ~doc:"JSON DateTimestamp"
       and payload =
         flag "payload" (required json_arg) ~doc:"JSON InvocationStepPayload"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.put_invocation_step
           (Values.PutInvocationStepRequest.make ?invocationStepId
              ~invocationIdentifier
              ~invocationStepTime:(Values.DateTimestamp.of_json
                                     invocationStepTime)
              ~payload:(Values.InvocationStepPayload.of_json payload)
              ~sessionIdentifier ())
           (Some Values.PutInvocationStepResponse.to_json)
           (Some Values.PutInvocationStepResponse.error_to_json)])
let rerank =
  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 nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and queries =
         flag "queries" (required json_arg) ~doc:"JSON RerankQueriesList"
       and rerankingConfiguration =
         flag "reranking-configuration" (required json_arg)
           ~doc:"JSON RerankingConfiguration"
       and sources =
         flag "sources" (required json_arg) ~doc:"JSON RerankSourcesList" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region Io.rerank
           (Values.RerankRequest.make ?nextToken
              ~queries:(Values.RerankQueriesList.of_json queries)
              ~rerankingConfiguration:(Values.RerankingConfiguration.of_json
                                         rerankingConfiguration)
              ~sources:(Values.RerankSourcesList.of_json sources) ())
           (Some Values.RerankResponse.to_json)
           (Some Values.RerankResponse.error_to_json)])
let retrieve =
  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 guardrailConfiguration =
         flag "guardrail-configuration" (optional json_arg)
           ~doc:"JSON GuardrailConfiguration"
       and nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and retrievalConfiguration =
         flag "retrieval-configuration" (optional json_arg)
           ~doc:"JSON KnowledgeBaseRetrievalConfiguration"
       and knowledgeBaseId =
         flag "knowledge-base-id" (required string)
           ~doc:"STRING KnowledgeBaseId"
       and retrievalQuery =
         flag "retrieval-query" (required json_arg)
           ~doc:"JSON KnowledgeBaseQuery" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.retrieve
           (Values.RetrieveRequest.make
              ?guardrailConfiguration:(Option.map
                                         ~f:Values.GuardrailConfiguration.of_json
                                         guardrailConfiguration) ?nextToken
              ?retrievalConfiguration:(Option.map
                                         ~f:Values.KnowledgeBaseRetrievalConfiguration.of_json
                                         retrievalConfiguration)
              ~knowledgeBaseId
              ~retrievalQuery:(Values.KnowledgeBaseQuery.of_json
                                 retrievalQuery) ())
           (Some Values.RetrieveResponse.to_json)
           (Some Values.RetrieveResponse.error_to_json)])
let retrieve_and_generate =
  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 retrieveAndGenerateConfiguration =
         flag "retrieve-and-generate-configuration" (optional json_arg)
           ~doc:"JSON RetrieveAndGenerateConfiguration"
       and sessionConfiguration =
         flag "session-configuration" (optional json_arg)
           ~doc:"JSON RetrieveAndGenerateSessionConfiguration"
       and sessionId =
         flag "session-id" (optional string) ~doc:"STRING SessionId"
       and input =
         flag "input" (required json_arg)
           ~doc:"JSON RetrieveAndGenerateInput" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.retrieve_and_generate
           (Values.RetrieveAndGenerateRequest.make
              ?retrieveAndGenerateConfiguration:(Option.map
                                                   ~f:Values.RetrieveAndGenerateConfiguration.of_json
                                                   retrieveAndGenerateConfiguration)
              ?sessionConfiguration:(Option.map
                                       ~f:Values.RetrieveAndGenerateSessionConfiguration.of_json
                                       sessionConfiguration) ?sessionId
              ~input:(Values.RetrieveAndGenerateInput.of_json input) ())
           (Some Values.RetrieveAndGenerateResponse.to_json)
           (Some Values.RetrieveAndGenerateResponse.error_to_json)])
let retrieve_and_generate_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 retrieveAndGenerateConfiguration =
         flag "retrieve-and-generate-configuration" (optional json_arg)
           ~doc:"JSON RetrieveAndGenerateConfiguration"
       and sessionConfiguration =
         flag "session-configuration" (optional json_arg)
           ~doc:"JSON RetrieveAndGenerateSessionConfiguration"
       and sessionId =
         flag "session-id" (optional string) ~doc:"STRING SessionId"
       and input =
         flag "input" (required json_arg)
           ~doc:"JSON RetrieveAndGenerateInput" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.retrieve_and_generate_stream
           (Values.RetrieveAndGenerateStreamRequest.make
              ?retrieveAndGenerateConfiguration:(Option.map
                                                   ~f:Values.RetrieveAndGenerateConfiguration.of_json
                                                   retrieveAndGenerateConfiguration)
              ?sessionConfiguration:(Option.map
                                       ~f:Values.RetrieveAndGenerateSessionConfiguration.of_json
                                       sessionConfiguration) ?sessionId
              ~input:(Values.RetrieveAndGenerateInput.of_json input) ())
           (Some Values.RetrieveAndGenerateStreamResponse.to_json)
           (Some Values.RetrieveAndGenerateStreamResponse.error_to_json)])
let start_flow_execution =
  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 flowExecutionName =
         flag "flow-execution-name" (optional string)
           ~doc:"STRING FlowExecutionName"
       and modelPerformanceConfiguration =
         flag "model-performance-configuration" (optional json_arg)
           ~doc:"JSON ModelPerformanceConfiguration"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier"
       and inputs = flag "inputs" (required json_arg) ~doc:"JSON FlowInputs" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.start_flow_execution
           (Values.StartFlowExecutionRequest.make ?flowExecutionName
              ?modelPerformanceConfiguration:(Option.map
                                                ~f:Values.ModelPerformanceConfiguration.of_json
                                                modelPerformanceConfiguration)
              ~flowAliasIdentifier ~flowIdentifier
              ~inputs:(Values.FlowInputs.of_json inputs) ())
           (Some Values.StartFlowExecutionResponse.to_json)
           (Some Values.StartFlowExecutionResponse.error_to_json)])
let stop_flow_execution =
  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 executionIdentifier =
         flag "execution-identifier" (required string)
           ~doc:"STRING FlowExecutionIdentifier"
       and flowAliasIdentifier =
         flag "flow-alias-identifier" (required string)
           ~doc:"STRING FlowAliasIdentifier"
       and flowIdentifier =
         flag "flow-identifier" (required string)
           ~doc:"STRING FlowIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.stop_flow_execution
           (Values.StopFlowExecutionRequest.make ~executionIdentifier
              ~flowAliasIdentifier ~flowIdentifier ())
           (Some Values.StopFlowExecutionResponse.to_json)
           (Some Values.StopFlowExecutionResponse.error_to_json)])
let tag_resource =
  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 resourceArn =
         flag "resource-arn" (required string)
           ~doc:"STRING TaggableResourcesArn"
       and tags = flag "tags" (required json_arg) ~doc:"JSON TagsMap" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.tag_resource
           (Values.TagResourceRequest.make ~resourceArn
              ~tags:(Values.TagsMap.of_json tags) ())
           (Some Values.TagResourceResponse.to_json)
           (Some Values.TagResourceResponse.error_to_json)])
let untag_resource =
  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 resourceArn =
         flag "resource-arn" (required string)
           ~doc:"STRING TaggableResourcesArn"
       and tagKeys =
         flag "tag-keys" (required json_arg) ~doc:"JSON TagKeyList" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.untag_resource
           (Values.UntagResourceRequest.make ~resourceArn
              ~tagKeys:(Values.TagKeyList.of_json tagKeys) ())
           (Some Values.UntagResourceResponse.to_json)
           (Some Values.UntagResourceResponse.error_to_json)])
let update_session =
  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 sessionMetadata =
         flag "session-metadata" (optional json_arg)
           ~doc:"JSON SessionMetadataMap"
       and sessionIdentifier =
         flag "session-identifier" (required string)
           ~doc:"STRING SessionIdentifier" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.update_session
           (Values.UpdateSessionRequest.make
              ?sessionMetadata:(Option.map
                                  ~f:Values.SessionMetadataMap.of_json
                                  sessionMetadata) ~sessionIdentifier ())
           (Some Values.UpdateSessionResponse.to_json)
           (Some Values.UpdateSessionResponse.error_to_json)])
let main =
  Command.group
    ~summary:((Awso.Service.to_string Values.service) ^ " commands")
    [("create-invocation", create_invocation);
    ("create-session", create_session);
    ("delete-agent-memory", delete_agent_memory);
    ("delete-session", delete_session);
    ("end-session", end_session);
    ("generate-query", generate_query);
    ("get-agent-memory", get_agent_memory);
    ("get-execution-flow-snapshot", get_execution_flow_snapshot);
    ("get-flow-execution", get_flow_execution);
    ("get-invocation-step", get_invocation_step);
    ("get-session", get_session);
    ("invoke-agent", invoke_agent);
    ("invoke-flow", invoke_flow);
    ("invoke-inline-agent", invoke_inline_agent);
    ("list-flow-execution-events", list_flow_execution_events);
    ("list-flow-executions", list_flow_executions);
    ("list-invocation-steps", list_invocation_steps);
    ("list-invocations", list_invocations);
    ("list-sessions", list_sessions);
    ("list-tags-for-resource", list_tags_for_resource);
    ("optimize-prompt", optimize_prompt);
    ("put-invocation-step", put_invocation_step);
    ("rerank", rerank);
    ("retrieve", retrieve);
    ("retrieve-and-generate", retrieve_and_generate);
    ("retrieve-and-generate-stream", retrieve_and_generate_stream);
    ("start-flow-execution", start_flow_execution);
    ("stop-flow-execution", stop_flow_execution);
    ("tag-resource", tag_resource);
    ("untag-resource", untag_resource);
    ("update-session", update_session)]