From a47d91f0e710c3a5d3bb1cbc9ecefff810e18bf4 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 29 Dec 2023 00:00:58 +1100 Subject: [PATCH] feat(api): add max_prompts constraints --- invokeai/app/api/routers/utilities.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/app/api/routers/utilities.py b/invokeai/app/api/routers/utilities.py index 476d10e2c0..2a912dfacf 100644 --- a/invokeai/app/api/routers/utilities.py +++ b/invokeai/app/api/routers/utilities.py @@ -23,10 +23,11 @@ class DynamicPromptsResponse(BaseModel): ) async def parse_dynamicprompts( prompt: str = Body(description="The prompt to parse with dynamicprompts"), - max_prompts: int = Body(default=1000, description="The max number of prompts to generate"), + max_prompts: int = Body(ge=1, le=10000, default=1000, description="The max number of prompts to generate"), combinatorial: bool = Body(default=True, description="Whether to use the combinatorial generator"), ) -> DynamicPromptsResponse: """Creates a batch process""" + max_prompts = min(max_prompts, 10000) generator: Union[RandomPromptGenerator, CombinatorialPromptGenerator] try: error: Optional[str] = None