docs: implement feedback

2025-12-20 23:24:20 +01:00 · 2023-04-19 10:44:14 +02:00
parent f513277a05
commit 0af51e4bd8
3 changed files with 6 additions and 8 deletions
--- a/src/cli.py
+++ b/src/cli.py
@@ -65,10 +65,8 @@ def generate(

    from src.options.generate.generator import Generator
    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}")
+    generator.generate(path)
+

@main.command()
@path_param