From 276f661e6c90894958eb2c89921aa46af7616820 Mon Sep 17 00:00:00 2001
From: Nick Fujita <nickfujita@gmail.com>
Date: Thu, 20 Feb 2025 18:02:36 +0900
Subject: [PATCH] 'add specific providers to provider type'

---
 src/crewai/utilities/embedding_configurator.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py
index 78704e954..e6c773717 100644
--- a/src/crewai/utilities/embedding_configurator.py
+++ b/src/crewai/utilities/embedding_configurator.py
@@ -1,5 +1,5 @@
 import os
-from typing import Any, Callable, cast
+from typing import Any, Callable, Literal, cast
 
 from chromadb import Documents, EmbeddingFunction, Embeddings
 from chromadb.api.types import validate_embedding_function
@@ -26,7 +26,19 @@ class EmbeddingProviderConfig(BaseModel):
 
 
 class EmbeddingConfig(BaseModel):
-    provider: str
+    provider: Literal[
+        "openai",
+        "azure",
+        "ollama",
+        "vertexai",
+        "google",
+        "cohere",
+        "voyageai",
+        "bedrock",
+        "huggingface",
+        "watson",
+        "custom",
+    ]
     config: EmbeddingProviderConfig | None = None