diff --git a/.gitignore b/.gitignore index b5dafc3fc..0a78c3d58 100644 --- a/.gitignore +++ b/.gitignore @@ -178,3 +178,12 @@ htmlcov.* *-structure.csv *-structure.json +/Titanic/2023_12_07_11_44_319a116fff/LLM_inout_pair/*.json +/ICR/2023_12_06_14_14_26e593d09f/LLM_inout_pair/*.json +/ICR/5cd9acb669c443fabe763e8f1ade5e86/workspace/*.txt +/ICR/5cd9acb669c443fabe763e8f1ade5e86/workspace/*.csv +/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.csv +/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.txt +/metagpt/roles/catboost_info/*.tsv +/metagpt/roles/catboost_info/*.json +/Titanic/9530b3c5550a4366ae92e5af6a74e6c3/workspace/*.md diff --git a/tests/metagpt/tools/functions/test_sd.py b/tests/metagpt/tools/functions/test_sd.py new file mode 100644 index 000000000..405ac9a32 --- /dev/null +++ b/tests/metagpt/tools/functions/test_sd.py @@ -0,0 +1,17 @@ +# -*- coding: utf-8 -*- +# @Date : 1/10/2024 10:07 PM +# @Author : stellahong (stellahong@fuzhi.ai) +# @Desc : +from metagpt.tools.sd_engine import SDEngine + +def test_sd_tools(): + engine = SDEngine() + prompt = "1boy, hansom" + engine.construct_payload(prompt) + engine.simple_run_t2i(engine.payload) + +def test_sd_construct_payload(): + engine = SDEngine() + prompt = "1boy, hansom" + engine.construct_payload(prompt) + assert "negative_prompt" in engine.payload \ No newline at end of file