refactor(tools/cogview3): syntax style

langgenius · Sep 13, 2024 · 70c1378 · 70c1378
1 parent 3952d13
commit 70c1378
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/api/core/tools/provider/builtin/cogview/tools/cogview3.py b/api/core/tools/provider/builtin/cogview/tools/cogview3.py
@@ -26,7 +26,7 @@ def _invoke(
             "horizontal_1344": "1344x768",
             "horizontal_1152": "1152x864",
             "widescreen_1440": "1440x720",
-            "tallscreen_720": "720x1440"
+            "tallscreen_720": "720x1440",
         }
         # prompt
         prompt = tool_parameters.get("prompt", "")
@@ -36,7 +36,7 @@ def _invoke(
         size_key = tool_parameters.get("size", "square")
         # cogview-3-plus get size
         if size_key != "cogview_3":
-            size = size_mapping[size_key] 
+            size = size_mapping[size_key]
         # get n
         n = tool_parameters.get("n", 1)
         # get quality
@@ -50,7 +50,7 @@ def _invoke(
         # set extra body
         seed_id = tool_parameters.get("seed_id", self._generate_random_id(8))
         extra_body = {"seed": seed_id}
-        # cogview-3-plus        
+        # cogview-3-plus
         if size_key != "cogview_3":
             response = client.images.generations(
                 prompt=prompt,
@@ -60,7 +60,7 @@ def _invoke(
                 extra_body=extra_body,
                 style=style,
                 quality=quality,
-                response_format="b64_json"
+                response_format="b64_json",
             )
         # cogview-3
         else:
@@ -71,7 +71,7 @@ def _invoke(
                 extra_body=extra_body,
                 style=style,
                 quality=quality,
-                response_format="b64_json"
+                response_format="b64_json",
             )
         print(response.data)
         result = []