docs: fix documentation

This commit is contained in:
Florian Hönicke
2023-04-19 01:03:24 +02:00
parent 8fb9c3cc8e
commit b32adf501d
4 changed files with 74 additions and 79 deletions

View File

@@ -46,7 +46,7 @@ def main(ctx):
@click.option('--description', required=True, help='Description of the microservice.')
@click.option('--test', required=True, help='Test scenario for the microservice.')
@click.option('--model', default='gpt-4', help='GPT model to use (default: gpt-4).')
@click.option('--verbose', default=False, is_flag=True, help='Verbose mode.')
@click.option('--verbose', default=False, is_flag=True, help='Verbose mode.') # only for development
@path_param
def generate(
description,
@@ -67,8 +67,8 @@ def generate(
generator = Generator(description, test, model=model)
with get_openai_callback() as cb:
generator.generate(path)
print(f"Prompt/Completion/Total Tokens: {cb.prompt_tokens}/{cb.completion_tokens}/{cb.total_tokens}")
print(f"Total Cost on OpenAI (USD): ${cb.total_cost}")
# print(f"Prompt/Completion/Total Tokens: {cb.prompt_tokens}/{cb.completion_tokens}/{cb.total_tokens}")
# print(f"Total Cost on OpenAI (USD): ${cb.total_cost}")
@main.command()
@path_param