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

fix: embedding job fails using IAM role (#5252)

Browse source
This commit is contained in:
Masashi Tomooka 2024-06-15 19:57:54 +09:00 • committed by GitHub
commit d9bee03ff6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/model_runtime/model_providers/bedrock/bedrock.yaml
View file

@ -23,7 +23,7 @@ provider_credential_schema:
- variable: aws_access_key_id - variable: aws_access_key_id
required: false required: false
label: label:
en_US: Access Key (If not provided, credentials are obtained from your running environment. e.g. IAM role) en_US: Access Key (If not provided, credentials are obtained from the running environment.)
zh_Hans: Access Key zh_Hans: Access Key
type: secret-input type: secret-input
placeholder: placeholder:

4
api/core/model_runtime/model_providers/bedrock/text_embedding/text_embedding.py
View file

@ -49,8 +49,8 @@ class BedrockTextEmbeddingModel(TextEmbeddingModel):
bedrock_runtime = boto3.client( bedrock_runtime = boto3.client(
service_name='bedrock-runtime', service_name='bedrock-runtime',
config=client_config, config=client_config,
aws_access_key_id=credentials["aws_access_key_id"], aws_access_key_id=credentials.get("aws_access_key_id", None),
aws_secret_access_key=credentials["aws_secret_access_key"] aws_secret_access_key=credentials.get("aws_secret_access_key", None)
) )
embeddings = [] embeddings = []