support for tokenids logprobs

[shreymodi1]

---

name: Pull Request
about: Propose changes to the codebase
title: "Brief description of changes"
labels: ''
assignees: ''

---

Description

Please include a summary of the change and which issue is fixed or feature is implemented. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes # (issue)
Implements # (issue)

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• This change requires a documentation update
• Refactoring/Code cleanup
• Build/CI/CD related changes
• Other (please describe):

How Has This Been Tested?

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration.

• Test A
• Test B

Test Configuration:

• Firmware version:
• Hardware:
• Toolchain:
• SDK:

Checklist:

• My code follows the style guidelines of this project (ran black ., isort ., flake8 .)
• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes
• Any dependent changes have been merged and published in downstream modules
• I have checked my code and corrected any misspellings

Screenshots (if applicable)

If applicable, add screenshots to help showcase your changes.

Additional context

Add any other context about the PR here.

---

Note

Extracts token IDs, routing matrix, and logprobs in single-turn rollout and exposes them via execution metadata; adds a pytest validating this behavior.

• Rollout processing (SingleTurn):
  • Extracts token_ids, routing_matrix, and logprobs from response.choices[0].logprobs and stores them in row.execution_metadata.extra.
  • Logs extraction counts; adds guarded parsing for dict/object shapes with error handling.
• Tests:
  • Adds test_single_turn_with_logprobs_and_routing_matrix to validate metadata extraction with logprobs=True parameters.

^{Written by Cursor Bugbot for commit 12fcae3. This will update automatically on new commits. Configure here.}