Update main.py

main.py

@@ -63,8 +63,10 @@ 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 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)."""}
+        {"role": "user", "content": f"""always check user query mentioned the order number/id (example : CA000001)
+                                   if order id provided then output is only : "done"
+                                   if user not given the order number then ask the number from user (include or response to greetings)
+                                   user query : {query}""" }
     ]
     result = ask_openai(messages)
 
@@ -72,7 +74,8 @@ 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 only waybill/order number from the query. User query: {query}"}
+            {"role": "user", "content": f"""extract the number from the user query example output is only : CA000001
+                                    user query : {query}""" }
         ]
         waybill_number = ask_openai(extract_message)