update prompt

chatbot_simulator.py

@@ -95,30 +95,30 @@ Rules:
 
         # Update user state using GPT's response
         update_prompt = f"""
-        Update the user state based on user input: '{user_input}' and task: {self.task}
-
-        Current user state (JSON format): {self.user_state}
-
-        Sitemap: {self.sitemap}
-
-        Instructions:
-        1. If the 'current_page' has changed, update it to a page from the sitemap.
-        2. If the task is finished, update 'task_completed' to 1. Otherwise, leave it unchanged.
-        3. If no updates are needed, return the user state exactly as provided, without modification.
-        4. Preserve the **exact JSON structure** and **format** of the provided user state.
-        5. The output **must be a single JSON dictionary** representing the updated user state—do not wrap it in a list.
-        6. Do not change any other fields unless explicitly required by the instructions.
-
-        Important:
-        - Ensure 'current_page' and 'task_completed' are keys in the returned dictionary.
-        - Return **only the JSON object** without additional output or wrapping.
-        - **AVOID OUTPUT A LIST**, must be JSON!
-        
-        Example Format:
-        {{
-            'current_page': 'Home',
-            'task_completed': 0,
-        }}
+Update the user state based on user input: '{user_input}' and task: {self.task}
+
+Current user state (JSON format): {self.user_state}
+
+Sitemap: {self.sitemap}
+
+Instructions:
+1. If the 'current_page' has changed, update it to a page from the sitemap.
+2. If the task is finished, update 'task_completed' to 1. Otherwise, leave it unchanged.
+3. If no updates are needed, return the user state exactly as provided, without modification.
+4. Preserve the **exact JSON structure** and **format** of the provided user state.
+5. The output **must be a single JSON dictionary** representing the updated user state—do not wrap it in a list.
+6. Do not change any other fields unless explicitly required by the instructions.
+
+Important:
+- Ensure 'current_page' and 'task_completed' are keys in the returned dictionary.
+- Return **only the JSON object** without additional output or wrapping.
+- **AVOID OUTPUT A LIST**, must be JSON!
+
+Example Format:
+{{
+    'current_page': 'Home',
+    'task_completed': 0,
+}}
         """
 
         self.conversation.append({"role": "assistant", "content": update_prompt})
@@ -129,7 +129,12 @@ Rules:
 
         if isinstance(updated_state, list):
             reformat_prompt = f'''
-            Given the {updated_state}, reformat it into a proper JSON.
+Given the {updated_state}, reformat it into a proper JSON.
+Make sure follow the format:
+{{
+    'current_page': 'Home',
+    'task_completed': 0,
+}}
             '''
             reformat_state = self._get_openai_response([{"role": "assistant", "content": reformat_prompt}])
             updated_state = json_repair.loads(reformat_state)
@@ -153,12 +158,12 @@ Rules:
         Simulate the agent generating input based on the conversation state.
         """
         agent_prompt = f"""
-        Imagine you are an agent navigate through the Uber environment.
-        Your overarching task is: {self.task}. You may have done some part of the task, or none at all.
-        You will have access to all of your previous actions in the environment, as well as the last message from the assistant giving the current state of the environment.
-        The last message from the assistant was: {self.conversation[-1]['content']}
-        Respond first with a brief "Plan" which suggests what steps you are going to take to accomplish the task, and what your immediate. 
-        Then generate an "Action" which is the immediate next step you can take.
+Imagine you are an agent navigate through the Uber environment.
+Your overarching task is: {self.task}. You may have done some part of the task, or none at all.
+You will have access to all of your previous actions in the environment, as well as the last message from the assistant giving the current state of the environment.
+The last message from the assistant was: {self.conversation[-1]['content']}
+Respond first with a brief "Plan" which suggests what steps you are going to take to accomplish the task, and what your immediate. 
+Then generate an "Action" which is the immediate next step you can take.
         """
 
         messages = [{"role": "system", "content": agent_prompt}]