From f2122ed696e78ea867eb990e20f7dbb4c1856425 Mon Sep 17 00:00:00 2001
From: Abhishek Sharma <76608878+being-abhi@users.noreply.github.com>
Date: Thu, 14 Mar 2024 12:31:30 +0530
Subject: [PATCH] [Fix] Added missing provider for 'vllm' (#1316)

---
 embedchain/factory.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/embedchain/factory.py b/embedchain/factory.py
index 9c0dea3e..fa46f3e9 100644
--- a/embedchain/factory.py
+++ b/embedchain/factory.py
@@ -25,6 +25,7 @@ class LlmFactory:
         "mistralai": "embedchain.llm.mistralai.MistralAILlm",
         "groq": "embedchain.llm.groq.GroqLlm",
         "nvidia": "embedchain.llm.nvidia.NvidiaLlm",
+        "vllm": "embedchain.llm.vllm.VLLM",
     }
     provider_to_config_class = {
         "embedchain": "embedchain.config.llm.base.BaseLlmConfig",