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

feat: support auth type like basic bearer and custom (#2613)

Browse source
This commit is contained in:
Yeuoly 2024-02-28 23:19:08 +08:00 • committed by GitHub
commit d44b05a9e5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 122 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/core/tools/tool/api_tool.py
View file

@ -62,6 +62,17 @@ class ApiTool(Tool):
if 'api_key_value' not in credentials:
raise ToolProviderCredentialValidationError('Missing api_key_value')
elif not isinstance(credentials['api_key_value'], str):
raise ToolProviderCredentialValidationError('api_key_value must be a string')
if 'api_key_header_prefix' in credentials:
api_key_header_prefix = credentials['api_key_header_prefix']
if api_key_header_prefix == 'basic':
credentials['api_key_value'] = f'Basic {credentials["api_key_value"]}'
elif api_key_header_prefix == 'bearer':
credentials['api_key_value'] = f'Bearer {credentials["api_key_value"]}'
elif api_key_header_prefix == 'custom':
pass
headers[api_key_header] = credentials['api_key_value']