Skip to content

Commit bfd079d

Browse files
committed
fix(mypy): changed type hinting of tools
1 parent 391e792 commit bfd079d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

examples/mistral/chat/function_calling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
ToolMessage,
1313
UserMessage,
1414
)
15+
from mistralai.client.models.chatcompletionrequest import ChatCompletionRequestTool
1516

1617
# Assuming we have the following data
1718
data: dict[str, list[Any]] = {
@@ -48,7 +49,7 @@ def retrieve_payment_date(data: dict[str, list[Any]], transaction_id: str) -> st
4849
"retrieve_payment_date": functools.partial(retrieve_payment_date, data=data),
4950
}
5051

51-
tools: list[Tool] = [
52+
tools: list[ChatCompletionRequestTool] = [
5253
Tool(
5354
function=Function(
5455
name="retrieve_payment_status",

0 commit comments

Comments
 (0)