chore: fixed issues with grade result prompt

.pre-commit-config.yaml

@@ -32,7 +32,7 @@ repos:
   - id: pretty-format-yaml
     args: [--autofix, --indent, '2']
 - repo: https://github.com/jsh9/pydoclint
-  rev: 0.5.4
+  rev: 0.5.5
   hooks:
   - id: pydoclint
     args: [--style=google, --check-return-types=False, --arg-type-hints-in-docstring=False]

gitmind/cli/commands/commit.py

@@ -5,7 +5,9 @@
 
 from gitmind.cli._utils import debug_echo, get_or_set_cli_context
 from gitmind.prompts import DescribeCommitHandler
-from gitmind.utils.commit import extract_commit_data, get_commit
+from gitmind.prompts.describe_commit import CommitDescriptionResult
+from gitmind.prompts.grade_commit import CommitGradingResult, GradeCommitHandler
+from gitmind.utils.commit import extract_commit_data
 from gitmind.utils.sync import run_as_sync
 
 
@@ -14,24 +16,39 @@ def commit() -> None:
     """Commit commands."""
 
 
-@commit.command()
-@option("--commit-hash", required=True, type=str)
-@pass_context
-@run_as_sync
-async def grade(ctx: Context, commit_hash: str) -> None:
-    """Grade a commit."""
+async def handle_describe(ctx: Context, commit_hash: str) -> CommitDescriptionResult:
+    """Describe a commit."""
     cli_ctx = get_or_set_cli_context(ctx)
     cli_ctx["commit_hash"] = commit_hash
-    commit = get_commit(repo=cli_ctx["repo"], commit_hex=commit_hash)
-    echo(f"Commit {commit_hash}: {commit.message}")
+
+    commit_statistics, commit_metadata, diff = extract_commit_data(repo=cli_ctx["repo"], commit_hex=commit_hash)
+    debug_echo(
+        cli_ctx,
+        f"Retrieved commit {commit_hash}: {commit_metadata["message"]}\n\ncommit_data: {dumps(commit_statistics, indent=2)}",
+    )
+    handler = DescribeCommitHandler(
+        client=cli_ctx["settings"].llm_client,
+    )
+    description = await handler(
+        statistics=commit_statistics,
+        metadata=commit_metadata,
+        diff=diff,
+    )
+    cli_ctx["commit_description"] = description
+    return description
 
 
 @commit.command()
 @option("--commit-hash", required=True, type=str)
 @pass_context
-@run_as_sync
-async def describe(ctx: Context, commit_hash: str) -> None:
+def describe(ctx: Context, commit_hash: str) -> None:
     """Describe a commit."""
+    description_result = run_as_sync(handle_describe)(ctx, commit_hash)
+    echo(dumps(description_result, indent=2))
+
+
+async def handle_grade(ctx: Context, commit_hash: str) -> dict[str, CommitGradingResult]:
+    """Grade a commit."""
     cli_ctx = get_or_set_cli_context(ctx)
     cli_ctx["commit_hash"] = commit_hash
 
@@ -40,14 +57,21 @@ async def describe(ctx: Context, commit_hash: str) -> None:
         cli_ctx,
         f"Retrieved commit {commit_hash}: {commit_metadata["message"]}\n\ncommit_data: {dumps(commit_statistics, indent=2)}",
     )
-    llm_client = cli_ctx["settings"].llm_client
-    handler = DescribeCommitHandler(
-        client=llm_client,
+
+    handler = GradeCommitHandler(
+        client=cli_ctx["settings"].llm_client,
     )
-    description = await handler(
-        statistics=commit_statistics,
+
+    return await handler(
         metadata=commit_metadata,
         diff=diff,
     )
-    cli_ctx["commit_description"] = description
-    echo(dumps(description, indent=2))
+
+
+@commit.command()
+@option("--commit-hash", required=True, type=str)
+@pass_context
+def grade(ctx: Context, commit_hash: str) -> None:
+    """Grade a commit."""
+    grading_results = run_as_sync(handle_grade)(ctx, commit_hash)
+    echo(dumps(grading_results, indent=2))