From 33c92e18611953d76772bd7bf03e601aabd6700c Mon Sep 17 00:00:00 2001
From: Marco <121761685+mlinmg@users.noreply.github.com>
Date: Mon, 13 Jan 2025 13:41:40 +0100
Subject: [PATCH] Update crew.py

Added OpenAiEmbeddingFunction as an allowed type to be in line with the documentation and to accept custom OAI-like embedder
---
 src/crewai/crew.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/crewai/crew.py b/src/crewai/crew.py
index b2857d827..650169e29 100644
--- a/src/crewai/crew.py
+++ b/src/crewai/crew.py
@@ -19,6 +19,8 @@ from pydantic import (
 )
 from pydantic_core import PydanticCustomError
 
+from chromadb.utils.embedding_functions.openai_embedding_function import OpenAIEmbeddingFunction
+
 from crewai.agent import Agent
 from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.agents.cache import CacheHandler
@@ -136,7 +138,7 @@ class Crew(BaseModel):
         default=None,
         description="An instance of the UserMemory to be used by the Crew to store/fetch memories of a specific user.",
     )
-    embedder: Optional[dict] = Field(
+    embedder: Optional[Union[dict, OpenAIEmbeddingFunction]] = Field(
         default=None,
         description="Configuration for the embedder to be used for the crew.",
     )