diff --git a/litellm/types/router.py b/litellm/types/router.py index fc19efa4b58..7fa49352a9c 100644 --- a/litellm/types/router.py +++ b/litellm/types/router.py @@ -425,11 +425,6 @@ class GenericLiteLLMParams(CredentialLiteLLMParams, CustomPricingLiteLLMParams): ) model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True) merge_reasoning_content_in_choices: bool | None = False - forward_reasoning_content: bool | None = None - reasoning_content_field: str | None = Field( - default=None, - description="Historical assistant reasoning field: reasoning_content (default) or reasoning.", - ) model_info: dict | None = None mock_response: str | ModelResponse | Exception | object | None = None @@ -531,6 +526,11 @@ class LiteLLM_Params(GenericLiteLLMParams): model: str model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True) + forward_reasoning_content: bool | None = None + reasoning_content_field: str | None = Field( + default=None, + description="Historical assistant reasoning field: reasoning_content (default) or reasoning.", + ) def __contains__(self, key) -> bool: # Define custom behavior for the 'in' operator @@ -553,6 +553,11 @@ class updateLiteLLMParams(GenericLiteLLMParams): # This class is used to update the LiteLLM_Params # only differece is model is optional model: str | None = None + forward_reasoning_content: bool | None = None + reasoning_content_field: str | None = Field( + default=None, + description="Historical assistant reasoning field: reasoning_content (default) or reasoning.", + ) class updateDeployment(BaseModel):