srcwalk 0.2.5

Tree-sitter indexed lookups — smart code reading for AI agents
Documentation
from tasks.base import Task, GroundTruth


class FastAPIDependencyResolutionTask(Task):
    @property
    def name(self) -> str:
        return "fastapi_dependency_resolution"

    @property
    def repo(self) -> str:
        return "fastapi"

    @property
    def prompt(self) -> str:
        return (
            "Trace how FastAPI resolves dependencies at request time. Starting from when "
            "a request hits an endpoint with Depends(), show the code path through "
            "dependency resolution. What function analyzes the callable's parameters, and "
            "what function actually executes the dependency chain?"
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=["get_dependant", "solve_dependencies", "Dependant", "analyze_param"],
        )

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


class FastAPIRequestValidationTask(Task):
    @property
    def name(self) -> str:
        return "fastapi_request_validation"

    @property
    def repo(self) -> str:
        return "fastapi"

    @property
    def prompt(self) -> str:
        return (
            "In FastAPI's routing.py, find the get_request_handler function. Show how it "
            "creates the actual request handler that processes incoming requests. Specifically, "
            "what happens when request validation fails? What exception is raised and where?"
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=[
                "get_request_handler",
                "solve_dependencies",
                "RequestValidationError",
                "async def app",
            ],
        )


class FastAPIDependsInternalsTask(Task):
    @property
    def name(self) -> str:
        return "fastapi_depends_internals"

    @property
    def repo(self) -> str:
        return "fastapi"

    @property
    def prompt(self) -> str:
        return (
            "In fastapi/param_functions.py, find the Depends function. Show its complete "
            "signature including all parameters. Then explain: is Depends a class or a function? "
            "What does it actually return, and where is the underlying implementation?"
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=["def Depends", "use_cache", "params.Depends"],
        )


class FastAPIDependsFunctionTask(Task):
    @property
    def name(self) -> str:
        return "fastapi_depends_function"

    @property
    def repo(self) -> str:
        return "fastapi"

    @property
    def prompt(self) -> str:
        return (
            "Find the Depends function in FastAPI. Show its complete signature "
            "and docstring."
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=["def Depends", "use_cache"],
        )


class FastAPIDependsProcessingTask(Task):
    @property
    def name(self) -> str:
        return "fastapi_depends_processing"

    @property
    def repo(self) -> str:
        return "fastapi"

    @property
    def prompt(self) -> str:
        return (
            "Find the Depends function in fastapi/param_functions.py and show "
            "its implementation. Then find how FastAPI processes dependencies "
            "— look for the function that walks the dependency tree and resolves "
            "each Depends() call at request time."
        )

    @property
    def ground_truth(self) -> GroundTruth:
        return GroundTruth(
            required_strings=[
                "def Depends", "solve_dependencies",
                "Dependant", "get_dependant",
            ],
        )

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