Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(aws): update spec #2525

Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions src/aws.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
40 changes: 40 additions & 0 deletions src/aws/bedrock-agent-runtime.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
110 changes: 110 additions & 0 deletions src/aws/bedrock-data-automation-runtime.ts
Original file line number Diff line number Diff line change
@@ -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;
Loading
Loading