Skip to content

column_manifest_agent

SensAI domain agent for signal columns - read-only, forced to answer with the current ColumnManifest.

return_column_manifest

Python
return_column_manifest(
    ctx: RunContext[SensAIDeps],
) -> ColumnManifest

Call this once you've gathered whatever you need to answer the request, to finish with the current column manifest.

Returns it directly rather than asking you to reconstruct it, since re-typing every column name yourself risks dropping or inventing one.

Source code in src/mujoco_mojo/utils/layers/dojo/sensai/column_manifest_agent.py
Python
22
23
24
25
26
27
28
29
def return_column_manifest(ctx: RunContext[SensAIDeps]) -> ColumnManifest:
    """
    Call this once you've gathered whatever you need to answer the request, to finish with the current column manifest.

    Returns it directly rather than asking you to reconstruct it, since re-typing every column name yourself risks dropping or inventing one.

    """
    return ctx.deps.column_manifest

get_available_signals async

Python
get_available_signals(
    ctx: RunContext[SensAIDeps],
) -> list[str]

Returns the list of signal column names available in the current trial's telemetry data.

Source code in src/mujoco_mojo/utils/layers/dojo/sensai/column_manifest_agent.py
Python
39
40
41
42
@column_manifest_agent.tool
async def get_available_signals(ctx: RunContext[SensAIDeps]) -> list[str]:
    """Returns the list of signal column names available in the current trial's telemetry data."""
    return ctx.deps.column_manifest["all"]

get_rotatable_signals async

Python
get_rotatable_signals(
    ctx: RunContext[SensAIDeps],
) -> list[str]

Returns the list of signal column names which can be rotated by get_quat_signals.

Source code in src/mujoco_mojo/utils/layers/dojo/sensai/column_manifest_agent.py
Python
45
46
47
48
@column_manifest_agent.tool
async def get_rotatable_signals(ctx: RunContext[SensAIDeps]) -> list[str]:
    """Returns the list of signal column names which can be rotated by get_quat_signals."""
    return ctx.deps.column_manifest["rotatable_vectors"]

get_quat_signals async

Python
get_quat_signals(ctx: RunContext[SensAIDeps]) -> list[str]

Returns the list of signal column names which can be used to rotate get_rotatable_signals.

Source code in src/mujoco_mojo/utils/layers/dojo/sensai/column_manifest_agent.py
Python
51
52
53
54
@column_manifest_agent.tool
async def get_quat_signals(ctx: RunContext[SensAIDeps]) -> list[str]:
    """Returns the list of signal column names which can be used to rotate get_rotatable_signals."""
    return ctx.deps.column_manifest["available_quats"]