Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
radbrt committed Mar 4, 2024
1 parent acf1035 commit 3862b4e
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions dbtai/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,10 +125,11 @@ def unit(model, instructions, write):
click.echo(test)
click.echo(explanation)

@dbtai.command(help="Write dbt constraints given the uniqueness tests in the model")
@dbtai.command(help="Not yet implemented. Write dbt constraints given the uniqueness tests in the model")
def constraints():
raise NotImplementedError("Not yet implemented")


@dbtai.command(help="Generate model code")
@click.argument("model_name", required=True)
@click.argument("description", required=True)
Expand All @@ -139,6 +140,7 @@ def gen(model_name, description, input):
click.echo(model["code"])
click.echo(f"\n\n{model['explanation']}")


@dbtai.command(help="Make a change to a dbt model")
@click.argument("model_name", required=True)
@click.argument("description", required=True)
Expand Down Expand Up @@ -177,6 +179,7 @@ def fluff(model, write, rewrite):
with open(write_path, "w") as f:
f.write(result['code'])


@dbtai.command(help="Explain the dbt code")
@click.argument("model", required=True)
def explain(model):
Expand All @@ -198,8 +201,6 @@ def chat(model):





@dbtai.command(help="Show logo")
def hello():
greeting = r"""
Expand Down

0 comments on commit 3862b4e

Please sign in to comment.