[Feature] Add support for NVIDIA AI LLMs and embedding models (#1293)

2024-02-29 23:56:25 -08:00
parent 6518c0c06b
commit c77a75dfb5
18 changed files with 195 additions and 22 deletions
--- a/embedchain/utils/misc.py
+++ b/embedchain/utils/misc.py
@@ -407,6 +407,7 @@ def validate_config(config_data):
                    "mistralai",
                    "vllm",
                    "groq",
+                    "nvidia",
                ),
                Optional("config"): {
                    Optional("model"): str,
@@ -443,6 +444,7 @@ def validate_config(config_data):
                    "azure_openai",
                    "google",
                    "mistralai",
+                    "nvidia",
                ),
                Optional("config"): {
                    Optional("model"): Optional(str),
@@ -462,6 +464,7 @@ def validate_config(config_data):
                    "azure_openai",
                    "google",
                    "mistralai",
+                    "nvidia",
                ),
                Optional("config"): {
                    Optional("model"): str,