github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix test

Browse Source
This commit is contained in:
Toran Bruce Richards
2024-09-29 21:07:04 +01:00
parent 212e97767d
commit a788aa26cc
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
autogpt_platform/backend/backend/blocks/replicate_flux_advanced.py
View File

@@ -110,29 +110,26 @@ class ReplicateFluxAdvancedModelBlock(Block):
input_schema=ReplicateFluxAdvancedModelBlock.Input,
output_schema=ReplicateFluxAdvancedModelBlock.Output,
test_input={
"api_key": "your_test_api_key",
"api_key": "test_api_key",
"replicate_model_name": ReplicateFluxModelName.FLUX_SCHNELL,
"prompt": "A beautiful landscape painting of a serene lake at sunrise",
"seed": None,
"steps": 25,
"guidance": 3.0,
"interval": 2.0,
"aspect_ratio": "1:1",
"output_format": "png",
"output_format": ImageType.PNG,
"output_quality": 80,
"safety_tolerance": 2,
},
test_output=[
(
"result",
{
"result_url": "https://replicate.com/output/generated-image-url.jpg",
},
"https://replicate.com/output/generated-image-url.jpg",
),
],
test_mock={
"run_model": lambda *args, **kwargs: {
"result_url": "https://replicate.com/output/generated-image-url.jpg",
}
"run_model": lambda api_key, model_name, prompt, seed, steps, guidance, interval, aspect_ratio, output_format, output_quality, safety_tolerance: "https://replicate.com/output/generated-image-url.jpg",
},
)