jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: check result should be string. (#1959)

Browse source
This commit is contained in:
Garfield Dai 2024-01-05 22:11:51 +08:00 • committed by GitHub
commit a8cedea15a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
api/core/external_data_tool/api/api.py
View file

@ -58,7 +58,7 @@ class ApiExternalDataTool(ExternalDataTool):
if not api_based_extension: if not api_based_extension:
raise ValueError("[External data tool] API query failed, variable: {}, " raise ValueError("[External data tool] API query failed, variable: {}, "
"error: api_based_extension_id is invalid" "error: api_based_extension_id is invalid"
.format(self.config.get('variable'))) .format(self.variable))
# decrypt api_key # decrypt api_key
api_key = encrypter.decrypt_token( api_key = encrypter.decrypt_token(
@ -74,7 +74,7 @@ class ApiExternalDataTool(ExternalDataTool):
) )
except Exception as e: except Exception as e:
raise ValueError("[External data tool] API query failed, variable: {}, error: {}".format( raise ValueError("[External data tool] API query failed, variable: {}, error: {}".format(
self.config.get('variable'), self.variable,
e e
)) ))
@ -87,6 +87,10 @@ class ApiExternalDataTool(ExternalDataTool):
if 'result' not in response_json: if 'result' not in response_json:
raise ValueError("[External data tool] API query failed, variable: {}, error: result not found in response" raise ValueError("[External data tool] API query failed, variable: {}, error: result not found in response"
.format(self.config.get('variable'))) .format(self.variable))
if not isinstance(response_json['result'], str):
raise ValueError("[External data tool] API query failed, variable: {}, error: result is not string"
.format(self.variable))
return response_json['result'] return response_json['result']