From 9a2a53d633cd6579ee8e1288497ab740588271ea Mon Sep 17 00:00:00 2001 From: oli5679 Date: Sun, 16 Nov 2025 13:49:34 +0000 Subject: [PATCH] Fix function_calling.py to check all transactions before returning error --- examples/mistral/chat/function_calling.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/examples/mistral/chat/function_calling.py b/examples/mistral/chat/function_calling.py index aba7d671..498cfa5f 100644 --- a/examples/mistral/chat/function_calling.py +++ b/examples/mistral/chat/function_calling.py @@ -29,16 +29,14 @@ def retrieve_payment_status(data: Dict[str, List], transaction_id: str) -> str: for i, r in enumerate(data["transaction_id"]): if r == transaction_id: return json.dumps({"status": data["payment_status"][i]}) - else: - return json.dumps({"status": "Error - transaction id not found"}) + return json.dumps({"status": "Error - transaction id not found"}) def retrieve_payment_date(data: Dict[str, List], transaction_id: str) -> str: for i, r in enumerate(data["transaction_id"]): if r == transaction_id: return json.dumps({"date": data["payment_date"][i]}) - else: - return json.dumps({"status": "Error - transaction id not found"}) + return json.dumps({"status": "Error - transaction id not found"}) names_to_functions = {