srcwalk 0.2.5

Tree-sitter indexed lookups — smart code reading for AI agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

from .base import Task, GroundTruth


class EditTask(Task):
    @property
    def name(self) -> str:
        return "edit_task"

    @property
    def prompt(self) -> str:
        return (
            "In src/database/connection.py, change the return type annotation of the "
            "`get_pool` function to `Optional[ConnectionPool]`. Add the necessary import for Optional."
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=["Optional"],
            file_path="src/database/connection.py",
            expected_diff_contains=["Optional[ConnectionPool]", "Optional"]
        )

    @property
    def task_type(self) -> str:
        return "edit"