feat: update spec

withfig · Dec 11, 2024 · b389f59 · b389f59
1 parent 6f07a24
commit b389f59
Show file tree

Hide file tree

Showing 14 changed files with 4,194 additions and 199 deletions.
diff --git a/src/aws.ts b/src/aws.ts
@@ -264,6 +264,16 @@ const completionSpec: Fig.Spec = {
         "Contains APIs related to model invocation and querying of knowledge bases",
       loadSpec: "aws/bedrock-agent-runtime",
     },
+    {
+      name: "bedrock-data-automation",
+      description: "Amazon Bedrock Keystone Build",
+      loadSpec: "aws/bedrock-data-automation",
+    },
+    {
+      name: "bedrock-data-automation-runtime",
+      description: "Amazon Bedrock Keystone Runtime",
+      loadSpec: "aws/bedrock-data-automation-runtime",
+    },
     {
       name: "bedrock-runtime",
       description:

diff --git a/src/aws/bedrock-agent-runtime.ts b/src/aws/bedrock-agent-runtime.ts
@@ -48,6 +48,46 @@ const completionSpec: Fig.Spec = {
         },
       ],
     },
+    {
+      name: "generate-query",
+      description:
+        "Generates an SQL query from a natural language query. For more information, see Generate a query for structured data in the Amazon Bedrock User Guide",
+      options: [
+        {
+          name: "--query-generation-input",
+          description:
+            "Specifies information about a natural language query to transform into SQL",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--transformation-configuration",
+          description:
+            "Specifies configurations for transforming the natural language query into SQL",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--cli-input-json",
+          description:
+            "Performs service operation based on the JSON string provided. The JSON string follows the format provided by ``--generate-cli-skeleton``. If other arguments are provided on the command line, the CLI values will override the JSON-provided values. It is not possible to pass arbitrary binary values using a JSON-provided value as the string will be taken literally",
+          args: {
+            name: "string",
+          },
+        },
+        {
+          name: "--generate-cli-skeleton",
+          description:
+            "Prints a JSON skeleton to standard output without sending an API request. If provided with no value or the value ``input``, prints a sample input JSON that can be used as an argument for ``--cli-input-json``. If provided with the value ``output``, it validates the command inputs and returns a sample output JSON for that command",
+          args: {
+            name: "string",
+            suggestions: ["input", "output"],
+          },
+        },
+      ],
+    },
     {
       name: "get-agent-memory",
       description: "Gets the sessions stored in the memory of the agent",

diff --git a/src/aws/bedrock-data-automation-runtime.ts b/src/aws/bedrock-data-automation-runtime.ts
@@ -0,0 +1,110 @@
+const completionSpec: Fig.Spec = {
+  name: "bedrock-data-automation-runtime",
+  description: "Amazon Bedrock Keystone Runtime",
+  subcommands: [
+    {
+      name: "get-data-automation-status",
+      description: "API used to get data automation status",
+      options: [
+        {
+          name: "--invocation-arn",
+          description: "Invocation arn",
+          args: {
+            name: "string",
+          },
+        },
+        {
+          name: "--cli-input-json",
+          description:
+            "Performs service operation based on the JSON string provided. The JSON string follows the format provided by ``--generate-cli-skeleton``. If other arguments are provided on the command line, the CLI values will override the JSON-provided values. It is not possible to pass arbitrary binary values using a JSON-provided value as the string will be taken literally",
+          args: {
+            name: "string",
+          },
+        },
+        {
+          name: "--generate-cli-skeleton",
+          description:
+            "Prints a JSON skeleton to standard output without sending an API request. If provided with no value or the value ``input``, prints a sample input JSON that can be used as an argument for ``--cli-input-json``. If provided with the value ``output``, it validates the command inputs and returns a sample output JSON for that command",
+          args: {
+            name: "string",
+            suggestions: ["input", "output"],
+          },
+        },
+      ],
+    },
+    {
+      name: "invoke-data-automation-async",
+      description: "Async API: Invoke data automation",
+      options: [
+        {
+          name: "--client-token",
+          description: "Idempotency token",
+          args: {
+            name: "string",
+          },
+        },
+        {
+          name: "--input-configuration",
+          description: "Input configuration",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--output-configuration",
+          description: "Output configuration",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--data-automation-configuration",
+          description: "Data automation configuration",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--encryption-configuration",
+          description: "Encryption configuration",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--notification-configuration",
+          description: "Notification configuration",
+          args: {
+            name: "structure",
+          },
+        },
+        {
+          name: "--blueprints",
+          description: "Blueprint list",
+          args: {
+            name: "list",
+          },
+        },
+        {
+          name: "--cli-input-json",
+          description:
+            "Performs service operation based on the JSON string provided. The JSON string follows the format provided by ``--generate-cli-skeleton``. If other arguments are provided on the command line, the CLI values will override the JSON-provided values. It is not possible to pass arbitrary binary values using a JSON-provided value as the string will be taken literally",
+          args: {
+            name: "string",
+          },
+        },
+        {
+          name: "--generate-cli-skeleton",
+          description:
+            "Prints a JSON skeleton to standard output without sending an API request. If provided with no value or the value ``input``, prints a sample input JSON that can be used as an argument for ``--cli-input-json``. If provided with the value ``output``, it validates the command inputs and returns a sample output JSON for that command",
+          args: {
+            name: "string",
+            suggestions: ["input", "output"],
+          },
+        },
+      ],
+    },
+  ],
+};
+
+export default completionSpec;