hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 20:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 4f0488abb5
commit d9bee03ff6
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 = []