support reasoning_content

Jintao-Huang · Feb 18, 2025 · b483672 · b483672
1 parent 3a41cca
commit b483672
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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