Update main.py

main.py

@@ -63,8 +63,8 @@ async def process_query(query: str):
     # Initial message to check for waybill number in the query
     messages = [
         {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
-        {"role": "user", "content": f"""Check if the order number is present in the query (e.g., CA000001). 
-                                        Respond 'done' if present, otherwise prompt the user to provide the number."""}
+        {"role": "user", "content": f"""Check if the waybill/order number is present in the query (e.g., CA000001). 
+                                        Respond 'done' if present, otherwise prompt the user to provide the number/waybill number (include or response to greetings)."""}
     ]
     result = ask_openai(messages)
 
@@ -72,7 +72,7 @@ async def process_query(query: str):
         # Extract the waybill number
         extract_message = [
             {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
-            {"role": "user", "content": f"Extract the waybill number from the query. User query: {query}"}
+            {"role": "user", "content": f"Extract the only waybill/order number from the query. User query: {query}"}
         ]
         waybill_number = ask_openai(extract_message)