nemo_eval.adapters.interceptors.reasoning_interceptor#

Module Contents#

Classes#

ResponseReasoningInterceptor

Intercepts responses to clean up reasoning tokens from the content.

Functions#

_clean_reasoning_tokens

Clean up reasoning tokens from the response.

API#

nemo_eval.adapters.interceptors.reasoning_interceptor._clean_reasoning_tokens(
response: requests.Response,
end_reasoning_token: str,
) → requests.Response[source]#

Clean up reasoning tokens from the response.

Parameters:

  • response – The API response object from requests

  • end_reasoning_token – Token that marks the end of reasoning section

Returns:

Response with reasoning tokens removed

class nemo_eval.adapters.interceptors.reasoning_interceptor.ResponseReasoningInterceptor(end_reasoning_token: str)[source]#

Bases: nemo_eval.adapters.interceptors.types.ResponseInterceptor

Intercepts responses to clean up reasoning tokens from the content.

Initialization

_end_reasoning_token: str#

None

intercept_response(
ar: nemo_eval.adapters.interceptors.types.AdapterResponse,
) → nemo_eval.adapters.interceptors.types.AdapterResponse[source]#