Skip to content

data

agentic_ml.research.tools.data

The profile_dataset tool.

make_profile_dataset 

make_profile_dataset(ctx: ResearchContext)

Build a profile_dataset tool bound to ctx.

Source code in src/agentic_ml/research/tools/data.py 
13
14
15
16
17
18
19
20
21
22
23
24
def make_profile_dataset(ctx: ResearchContext):
    """Build a ``profile_dataset`` tool bound to ``ctx``."""

    @tool
    def profile_dataset() -> str:
        """Return an aggregate description of the dataset (schema, dtypes, missing values).

        Only summary statistics are exposed; the raw rows are never revealed.
        """
        return ctx.profile_text

    return profile_dataset