chore: skip unnecessary key checks prior to accessing a dictionary (#4497)

2024-05-19 18:30:45 +08:00
parent aa13d14019
commit 04ad46dd31
30 changed files with 45 additions and 44 deletions
--- a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
@@ -70,7 +70,7 @@ class StableDiffusionTool(BuiltinTool):
        if not base_url:
            return self.create_text_message('Please input base_url')

-        if 'model' in tool_parameters and tool_parameters['model']:
+        if tool_parameters.get('model'):
            self.runtime.credentials['model'] = tool_parameters['model']

        model = self.runtime.credentials.get('model', None)