Merge branch 'sd_tools' into 'mgx_ops'

Sd tools

See merge request pub/MetaGPT!48

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

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: