fix(deepseek): move _ensure_reasoning_content before @overload stubs to fix mypy errors

This commit is contained in:
Ishaan Jaffer 2026-04-27 18:59:15 -07:00
parent dc4b206bf2
commit c54ff38646
No known key found for this signature in database

View file

@ -62,19 +62,6 @@ class DeepSeekChatConfig(OpenAIGPTConfig):
return optional_params
@overload
def _transform_messages(
self, messages: List[AllMessageValues], model: str, is_async: Literal[True]
) -> Coroutine[Any, Any, List[AllMessageValues]]: ...
@overload
def _transform_messages(
self,
messages: List[AllMessageValues],
model: str,
is_async: Literal[False] = False,
) -> List[AllMessageValues]: ...
def _ensure_reasoning_content_on_assistant_messages(
self, messages: List[AllMessageValues]
) -> List[AllMessageValues]:
@ -99,6 +86,19 @@ class DeepSeekChatConfig(OpenAIGPTConfig):
message["reasoning_content"] = "" # type: ignore[typeddict-unknown-key]
return messages
@overload
def _transform_messages(
self, messages: List[AllMessageValues], model: str, is_async: Literal[True]
) -> Coroutine[Any, Any, List[AllMessageValues]]: ...
@overload
def _transform_messages(
self,
messages: List[AllMessageValues],
model: str,
is_async: Literal[False] = False,
) -> List[AllMessageValues]: ...
def _transform_messages(
self, messages: List[AllMessageValues], model: str, is_async: bool = False
) -> Union[List[AllMessageValues], Coroutine[Any, Any, List[AllMessageValues]]]: