Update utility.py

utility.py

@@ -134,7 +134,7 @@ def generateResponse(prompt,model='Meta-Llama-3.1-70B-Instruct'):
         model = model,
         # model="gpt-4o",
         messages=[
-              {"role": "system", "content": f"You are a helpful assistant that classifies transactions written in natural language into CRUD operations (Create, Read, Update, and Delete) and extracts relevant information. You should be able to recognize the currency being used and any quantity units into separate fields. Format the relevant information extracted from the transaction text in this format: {relevant_info_template}. A sample response for a single transaction could look like this: {sample_single_transaction_template}, while multiple transactions could look like this: {sample_multi_transaction_template}. There should be only one intent even in the case of multiple transactions."},
+              {"role": "system", "content": f"You are a helpful assistant that classifies transactions written in natural language into CRUD operations (Create, Read, Update, and Delete) and extracts relevant information. For update and delete queries, the transaction type should either be sales or inventory. You should be able to recognize the currency being used and any quantity units into separate fields. Format the relevant information extracted from the transaction text in this format: {relevant_info_template}. A sample response for a single transaction could look like this: {sample_single_transaction_template}, while multiple transactions could look like this: {sample_multi_transaction_template}. There should be only one intent even in the case of multiple transactions."},
               {"role": "user", "content": prompt}
         ]
     )
@@ -356,6 +356,18 @@ def delete_transaction(user_phone, transaction_id):
     print("Transaction deleted successfully!")
 
 
+# Delete an entire transaction
+def delete_transaction(user_phone, transaction_data):
+    # the transaction id currently defaults to the item name
+    transaction_id = transaction_data[0]['details']['item']
+    transaction_type = transaction_data[0]['transaction_type'].lower()
+    doc_ref = db.collection("users").document(user_phone).collection(transaction_type).document(transaction_id)
+    item_doc = doc_ref.get()
+    if item_doc.exists:
+        doc_ref.delete()
+        return True
+    else:
+        return False
 
 
 # Example usage