Skip to content

Commit a269cf9

Browse files
Enhance ToolCall message to support all argument types (#1823)
Fixes - #1788
1 parent c0dc689 commit a269cf9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/ragas/messages.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,13 @@ class ToolCall(BaseModel):
2727
----------
2828
name : str
2929
The name of the tool being called.
30-
args : Dict[str, Union[str, int, float]]
30+
args : Dict[str, Any]
3131
A dictionary of arguments for the tool call, where keys are argument names
3232
and values can be strings, integers, or floats.
3333
"""
3434

3535
name: str
36-
args: t.Dict[str, t.Union[str, int, float]]
36+
args: t.Dict[str, t.Any]
3737

3838

3939
class HumanMessage(Message):

0 commit comments

Comments
 (0)