apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add sd ut

Browse source
This commit is contained in:
stellahsr 2024-01-11 21:56:38 +08:00
parent af26fe06cf
commit 3be26cf94f
2 changed files with 26 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
.gitignore vendored
View file

@ -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

17
tests/metagpt/tools/functions/test_sd.py Normal file
View file

@ -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