add workdir in test (#1274)

* add workdir in test * catch filenotfound * modify print msg
microsoft · Jan 16, 2024 · adbfb20 · adbfb20
1 parent e784f2b
commit adbfb20
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/notebook/agentchat_auto_feedback_from_code_execution.ipynb b/notebook/agentchat_auto_feedback_from_code_execution.ipynb
@@ -514,7 +514,10 @@
     }
    ],
    "source": [
-    "Image(filename=\"coding/stock_price_ytd.png\")"
+    "try:\n",
+    "    Image(filename=\"coding/stock_price_ytd.png\")\n",
+    "except FileNotFoundError:\n",
+    "    print(\"Image not found. Please check the file name and modify if necessary.\")"
    ]
   },
   {

diff --git a/test/agentchat/test_assistant_agent.py b/test/agentchat/test_assistant_agent.py
@@ -142,6 +142,13 @@ def test_create_execute_script(human_input_mode="NEVER", max_consecutive_auto_re
         message="""Create and execute a script to plot a rocket without using matplotlib""",
     )
     assistant.reset()
+    user = UserProxyAgent(
+        "user",
+        human_input_mode=human_input_mode,
+        code_execution_config={"work_dir": f"{here}/test_agent_scripts"},
+        max_consecutive_auto_reply=max_consecutive_auto_reply,
+        is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
+    )
     user.initiate_chat(
         assistant,
         message="""Create a temp.py file with the following content:
@@ -197,8 +204,8 @@ def generate_init_message(self, question) -> str:
 
 if __name__ == "__main__":
     # test_gpt35()
-    # test_create_execute_script(human_input_mode="TERMINATE")
+    test_create_execute_script(human_input_mode="TERMINATE")
     # when GPT-4, i.e., the DEFAULT_MODEL, is used, conversation in the following test
     # should terminate in 2-3 rounds of interactions (because is_termination_msg should be true after 2-3 rounds)
     # although the max_consecutive_auto_reply is set to 10.
-    test_tsp(human_input_mode="NEVER", max_consecutive_auto_reply=10)
+    # test_tsp(human_input_mode="NEVER", max_consecutive_auto_reply=10)