@@ -23,11 +23,26 @@ class ChatCompletionInputMessageChunk(BaseInferenceType):
2323 text : Optional [str ] = None
2424
2525
26+ @dataclass_with_extra
27+ class ChatCompletionInputFunctionDefinition (BaseInferenceType ):
28+ arguments : Any
29+ name : str
30+ description : Optional [str ] = None
31+
32+
33+ @dataclass_with_extra
34+ class ChatCompletionInputToolCall (BaseInferenceType ):
35+ function : ChatCompletionInputFunctionDefinition
36+ id : str
37+ type : str
38+
39+
2640@dataclass_with_extra
2741class ChatCompletionInputMessage (BaseInferenceType ):
28- content : Union [List [ChatCompletionInputMessageChunk ], str ]
2942 role : str
43+ content : Optional [Union [List [ChatCompletionInputMessageChunk ], str ]] = None
3044 name : Optional [str ] = None
45+ tool_calls : Optional [List [ChatCompletionInputToolCall ]] = None
3146
3247
3348ChatCompletionInputGrammarTypeType = Literal ["json" , "regex" ]
@@ -45,7 +60,7 @@ class ChatCompletionInputGrammarType(BaseInferenceType):
4560
4661@dataclass_with_extra
4762class ChatCompletionInputStreamOptions (BaseInferenceType ):
48- include_usage : bool
63+ include_usage : Optional [ bool ] = None
4964 """If set, an additional chunk will be streamed before the data: [DONE] message. The usage
5065 field on this chunk shows the token usage statistics for the entire request, and the
5166 choices field will always be an empty array. All other chunks will also include a usage
@@ -66,13 +81,6 @@ class ChatCompletionInputToolChoiceClass(BaseInferenceType):
6681ChatCompletionInputToolChoiceEnum = Literal ["auto" , "none" , "required" ]
6782
6883
69- @dataclass_with_extra
70- class ChatCompletionInputFunctionDefinition (BaseInferenceType ):
71- arguments : Any
72- name : str
73- description : Optional [str ] = None
74-
75-
7684@dataclass_with_extra
7785class ChatCompletionInputTool (BaseInferenceType ):
7886 function : ChatCompletionInputFunctionDefinition
@@ -197,6 +205,7 @@ class ChatCompletionOutputToolCall(BaseInferenceType):
197205class ChatCompletionOutputMessage (BaseInferenceType ):
198206 role : str
199207 content : Optional [str ] = None
208+ tool_call_id : Optional [str ] = None
200209 tool_calls : Optional [List [ChatCompletionOutputToolCall ]] = None
201210
202211
@@ -249,7 +258,8 @@ class ChatCompletionStreamOutputDeltaToolCall(BaseInferenceType):
249258class ChatCompletionStreamOutputDelta (BaseInferenceType ):
250259 role : str
251260 content : Optional [str ] = None
252- tool_calls : Optional [ChatCompletionStreamOutputDeltaToolCall ] = None
261+ tool_call_id : Optional [str ] = None
262+ tool_calls : Optional [List [ChatCompletionStreamOutputDeltaToolCall ]] = None
253263
254264
255265@dataclass_with_extra
0 commit comments