This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-31 15:22:26 -03:00
commit ddf14638e5
6 changed files with 6 additions and 9 deletions

View file

@ -26,10 +26,7 @@ class AmazonBedrockComponent(CustomComponent):
],
},
"credentials_profile_name": {"display_name": "Credentials Profile Name"},
"streaming": {
"display_name": "Streaming",
"field_type": "bool"
},
"streaming": {"display_name": "Streaming", "field_type": "bool"},
"code": {"show": False},
}

View file

@ -41,7 +41,8 @@ class AmazonKendraRetrieverComponent(CustomComponent):
region_name=region_name,
credentials_profile_name=credentials_profile_name,
attribute_filter=attribute_filter,
user_context=user_context) # type: ignore
user_context=user_context,
) # type: ignore
except Exception as e:
raise ValueError("Could not connect to AmazonKendra API.") from e
return output