satanicscriptures
diff --git a/‎examples/function_calling.py‎
Lines changed: 16 additions & 20 deletions b/‎examples/function_calling.py‎
Lines changed: 16 additions & 20 deletions
diff --git a/‎examples/json_format.py‎
Lines changed: 1 addition & 1 deletion b/‎examples/json_format.py‎
Lines changed: 1 addition & 1 deletion
@@ -1,8 +1,8 @@
 import functools
 import json
 import os
+from typing import Dict, List
 
-import pandas as pd
 from mistralai.client import MistralClient
 from mistralai.models.chat_completion import ChatMessage, Function
 
@@ -15,30 +15,27 @@
     "payment_status": ["Paid", "Unpaid", "Paid", "Paid", "Pending"],
 }
 
-# Create DataFrame
-df = pd.DataFrame(data)
+n_rows = len(data["transaction_id"])
 
+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"})
 
-def retrieve_payment_status(df: pd.DataFrame, transaction_id: str) -> str:
-    if transaction_id in df.transaction_id.values:
-        return json.dumps({"status": df[df.transaction_id == transaction_id].payment_status.item()})
-    else:
-        return json.dumps({"status": "error - transaction id not found."})
-
-
-def retrieve_payment_date(df: pd.DataFrame, transaction_id: str) -> str:
-    if transaction_id in df.transaction_id.values:
-        return json.dumps({"date": df[df.transaction_id == transaction_id].payment_date.item()})
-    else:
-        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"})
 
 names_to_functions = {
-    "retrieve_payment_status": functools.partial(retrieve_payment_status, df=df),
-    "retrieve_payment_date": functools.partial(retrieve_payment_date, df=df),
+        "retrieve_payment_status": functools.partial(retrieve_payment_status, data=data),
+        "retrieve_payment_date": functools.partial(retrieve_payment_date, data=data)
 }
 
-
 tools = [
     {
         "type": "function",
@@ -66,7 +63,6 @@ def retrieve_payment_date(df: pd.DataFrame, transaction_id: str) -> str:
     },
 ]
 
-
 api_key = os.environ["MISTRAL_API_KEY"]
 model = "mistral-large-latest"
 
 
@@ -15,7 +15,7 @@ def main():
     chat_response = client.chat(
         model=model,
         response_format={"type": "json_object"},
-        messages=[ChatMessage(role="user", content="What is the best French cheese?")],
+        messages=[ChatMessage(role="user", content="What is the best French cheese? Answer shortly in JSON.")],
 
     )
     print(chat_response.choices[0].message.content)
Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@ def main():`
`15`	`15`	`chat_response = client.chat(`
`16`	`16`	`model=model,`
`17`	`17`	`response_format={"type": "json_object"},`
`18`		`- messages=[ChatMessage(role="user", content="What is the best French cheese?")],`
	`18`	`+ messages=[ChatMessage(role="user", content="What is the best French cheese? Answer shortly in JSON.")],`
`19`	`19`
`20`	`20`	`)`
`21`	`21`	`print(chat_response.choices[0].message.content)`