File tree Expand file tree Collapse file tree 2 files changed +26
-0
lines changed
tests/metagpt/tools/functions Expand file tree Collapse file tree 2 files changed +26
-0
lines changed Original file line number Diff line number Diff line change @@ -178,3 +178,12 @@ htmlcov.*
178178* -structure.csv
179179* -structure.json
180180
181+ /Titanic /2023_12_07_11_44_319a116fff /LLM_inout_pair /* .json
182+ /ICR /2023_12_06_14_14_26e593d09f /LLM_inout_pair /* .json
183+ /ICR /5cd9acb669c443fabe763e8f1ade5e86 /workspace /* .txt
184+ /ICR /5cd9acb669c443fabe763e8f1ade5e86 /workspace /* .csv
185+ /Titanic /9530b3c5550a4366ae92e5af6a74e6c3 /workspace /* .csv
186+ /Titanic /9530b3c5550a4366ae92e5af6a74e6c3 /workspace /* .txt
187+ /metagpt /roles /catboost_info /* .tsv
188+ /metagpt /roles /catboost_info /* .json
189+ /Titanic /9530b3c5550a4366ae92e5af6a74e6c3 /workspace /* .md
Original file line number Diff line number Diff line change 1+ # -*- coding: utf-8 -*-
2+ # @Date : 1/10/2024 10:07 PM
3+ # @Author : stellahong (stellahong@fuzhi.ai)
4+ # @Desc :
5+ from metagpt .tools .sd_engine import SDEngine
6+
7+ def test_sd_tools ():
8+ engine = SDEngine ()
9+ prompt = "1boy, hansom"
10+ engine .construct_payload (prompt )
11+ engine .simple_run_t2i (engine .payload )
12+
13+ def test_sd_construct_payload ():
14+ engine = SDEngine ()
15+ prompt = "1boy, hansom"
16+ engine .construct_payload (prompt )
17+ assert "negative_prompt" in engine .payload
You can’t perform that action at this time.
0 commit comments