From b483672516f4a20b20348b84b7190d289464775d Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Tue, 18 Feb 2025 15:05:23 +0800 Subject: [PATCH] support reasoning_content --- swift/llm/infer/protocol.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/swift/llm/infer/protocol.py b/swift/llm/infer/protocol.py index 59f4d8e8c8..e13a7e2e04 100644 --- a/swift/llm/infer/protocol.py +++ b/swift/llm/infer/protocol.py @@ -224,6 +224,7 @@ class ChatCompletionMessageToolCall: class ChatMessage: role: Literal['system', 'user', 'assistant'] content: Union[str, List[Dict[str, Any]]] + reasoning_content: Optional[str] = None tool_calls: Optional[List[ChatCompletionMessageToolCall]] = None @@ -278,6 +279,7 @@ class CompletionResponse: class DeltaMessage: role: Literal['system', 'user', 'assistant', None] = None content: Optional[str] = None + reasoning_content: Optional[str] = None tool_calls: Optional[List[ChatCompletionMessageToolCall]] = None