phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): handle dict object for chatcompletionmessagetoolcall

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-05 18:10:43 -08:00
parent e8c475b925
commit f95458dad8
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -227,13 +227,16 @@ class ChatCompletionDeltaToolCall(OpenAIObject):
class ChatCompletionMessageToolCall(OpenAIObject): class ChatCompletionMessageToolCall(OpenAIObject):
def __init__( def __init__(
self, self,
function: Function, function: Union[Dict, Function],
id: Optional[str] = None, id: Optional[str] = None,
type: Optional[str] = None, type: Optional[str] = None,
**params, **params,
): ):
super(ChatCompletionMessageToolCall, self).__init__(**params) super(ChatCompletionMessageToolCall, self).__init__(**params)
self.function = function if isinstance(function, Dict):
self.function = Function(**function)
else:
self.function = function
if id is not None: if id is not None:
self.id = id self.id = id