diff --git a/metagpt/const.py b/metagpt/const.py index 484987a03..c01f92adc 100644 --- a/metagpt/const.py +++ b/metagpt/const.py @@ -103,12 +103,13 @@ TEST_OUTPUTS_FILE_REPO = "test_outputs" CODE_SUMMARIES_FILE_REPO = "docs/code_summary" CODE_SUMMARIES_PDF_FILE_REPO = "resources/code_summary" RESOURCES_FILE_REPO = "resources" -SD_OUTPUT_FILE_REPO = "resources/sd_output" +SD_OUTPUT_FILE_REPO = DEFAULT_WORKSPACE_ROOT GRAPH_REPO_FILE_REPO = "docs/graph_repo" VISUAL_GRAPH_REPO_FILE_REPO = "resources/graph_db" CLASS_VIEW_FILE_REPO = "docs/class_view" YAPI_URL = "http://yapi.deepwisdomai.com/" +SD_URL = "http://172.31.0.51:49094" DEFAULT_LANGUAGE = "English" DEFAULT_MAX_TOKENS = 1500 diff --git a/metagpt/tools/libs/sd_engine.py b/metagpt/tools/libs/sd_engine.py index b62e39db8..4cf7d2310 100644 --- a/metagpt/tools/libs/sd_engine.py +++ b/metagpt/tools/libs/sd_engine.py @@ -14,7 +14,7 @@ import requests from aiohttp import ClientSession from PIL import Image, PngImagePlugin -from metagpt.const import SD_OUTPUT_FILE_REPO, SOURCE_ROOT +from metagpt.const import SD_OUTPUT_FILE_REPO, SD_URL, SOURCE_ROOT from metagpt.logs import logger from metagpt.tools.tool_registry import register_tool @@ -68,7 +68,7 @@ class SDEngine: Args: sd_url (str, optional): URL of the stable diffusion service. Defaults to "". """ - self.sd_url = sd_url + self.sd_url = SD_URL if not sd_url else sd_url self.sd_t2i_url = f"{self.sd_url}/sdapi/v1/txt2img" # Define default payload settings for SD API self.payload = payload @@ -76,12 +76,12 @@ class SDEngine: def construct_payload( self, - prompt, - negtive_prompt=default_negative_prompt, - width=512, - height=512, - sd_model="galaxytimemachinesGTM_photoV20", - ): + prompt: object, + negtive_prompt: object = default_negative_prompt, + width: object = 512, + height: object = 512, + sd_model: object = "galaxytimemachinesGTM_photoV20", + ) -> object: """Modify and set the API parameters for image generation. Args: