[Feature] Add support for AWS Bedrock LLM (#1189)

Co-authored-by: Deven Patel <deven298@yahoo.com>
2024-01-21 14:09:08 +05:30
parent 751a3a4bd1
commit 069d265338
8 changed files with 226 additions and 8 deletions
--- a/embedchain/utils/misc.py
+++ b/embedchain/utils/misc.py
@@ -406,6 +406,7 @@ def validate_config(config_data):
                    "llama2",
                    "vertexai",
                    "google",
+                    "aws_bedrock",
                    "mistralai",
                ),
                Optional("config"): {
@@ -423,6 +424,7 @@ def validate_config(config_data):
                    Optional("query_type"): str,
                    Optional("api_key"): str,
                    Optional("endpoint"): str,
+                    Optional("model_kwargs"): dict,
                },
            },
            Optional("vectordb"): {