yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-09-25 08:05:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

add workdir in test (#1274)

Browse Source 
* add workdir in test

* catch filenotfound

* modify print msg
This commit is contained in:
Chi Wang 2024-01-15 19:26:18 -08:00 committed by GitHub
parent e784f2b9b6
commit adbfb206dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
notebook/agentchat_auto_feedback_from_code_execution.ipynb
View File

@ -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.\")"
]
},
{

11
test/agentchat/test_assistant_agent.py
View File

@ -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 test_tsp(human_input_mode="NEVER", max_consecutive_auto_reply=10):
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)