summaryrefslogtreecommitdiff
path: root/server/continuedev/plugins/steps/comment_code.py
blob: 1eee791d116dceaa1b1c112aeb9081e897a203e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from ...core.main import ContinueSDK, Models, Step
from .main import EditHighlightedCodeStep


class CommentCodeStep(Step):
    hide: bool = True

    async def describe(self, models: Models):
        return "Writing comments"

    async def run(self, sdk: ContinueSDK):
        await sdk.run_step(
            EditHighlightedCodeStep(
                user_input="Write comprehensive comments in the canonical format for every class and function"
            )
        )