Fix function calling example (#116)

mistralai · Jul 18, 2024 · 9d238f8 · 9d238f8
1 parent c2bfb76
commit 9d238f8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/function_calling.py b/examples/function_calling.py
@@ -20,16 +20,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 = {