andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 00:02:36 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

'add specific providers to provider type'

Browse Source
This commit is contained in:
Nick Fujita
2025-02-20 18:02:36 +09:00
parent 8f99caf61b
commit 276f661e6c
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/crewai/utilities/embedding_configurator.py
View File

@@ -1,5 +1,5 @@
import os import os
from typing import Any, Callable, cast from typing import Any, Callable, Literal, cast
from chromadb import Documents, EmbeddingFunction, Embeddings from chromadb import Documents, EmbeddingFunction, Embeddings
from chromadb.api.types import validate_embedding_function from chromadb.api.types import validate_embedding_function
@@ -26,7 +26,19 @@ class EmbeddingProviderConfig(BaseModel):
class EmbeddingConfig(BaseModel): class EmbeddingConfig(BaseModel):
provider: str provider: Literal[
"openai",
"azure",
"ollama",
"vertexai",
"google",
"cohere",
"voyageai",
"bedrock",
"huggingface",
"watson",
"custom",
]
config: EmbeddingProviderConfig | None = None config: EmbeddingProviderConfig | None = None