fix: update EmbedderConfig type definition

Author: greysonlalonde

src/crewai/agent.py

@@ -174,8 +174,7 @@ class Agent(BaseAgent):
     )
 
     @model_validator(mode="before")
-    @classmethod
-    def validate_from_repository(cls, v):
+    def validate_from_repository(cls, v): # noqa: N805
         if v is not None and (from_repository := v.get("from_repository")):
             return load_agent_from_repository(from_repository) | v
         return v
@@ -686,7 +685,7 @@ def _inject_date_to_task(self, task):
                 if not is_valid:
                     raise ValueError(f"Invalid date format: {self.date_format}")
 
-                current_date: str = datetime.now().strftime(self.date_format)
+                current_date = datetime.now().strftime(self.date_format)
                 task.description += f"\n\nCurrent Date: {current_date}"
             except Exception as e:
                 if hasattr(self, "_logger"):

src/crewai/rag/embeddings/providers/aws/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Any, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class BedrockProviderConfig(TypedDict, total=False):
     """Configuration for Bedrock provider."""
@@ -10,8 +12,8 @@ class BedrockProviderConfig(TypedDict, total=False):
     session: Any
 
 
-class BedrockProviderSpec(TypedDict):
+class BedrockProviderSpec(TypedDict, total=False):
     """Bedrock provider specification."""
 
-    provider: Literal["amazon-bedrock"]
+    provider: Required[Literal["amazon-bedrock"]]
     config: BedrockProviderConfig

src/crewai/rag/embeddings/providers/cohere/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class CohereProviderConfig(TypedDict, total=False):
     """Configuration for Cohere provider."""
@@ -10,8 +12,8 @@ class CohereProviderConfig(TypedDict, total=False):
     model_name: Annotated[str, "large"]
 
 
-class CohereProviderSpec(TypedDict):
+class CohereProviderSpec(TypedDict, total=False):
     """Cohere provider specification."""
 
-    provider: Literal["cohere"]
+    provider: Required[Literal["cohere"]]
     config: CohereProviderConfig

src/crewai/rag/embeddings/providers/custom/types.py

@@ -3,6 +3,7 @@
 from typing import Literal, TypedDict
 
 from chromadb.api.types import EmbeddingFunction
+from typing_extensions import Required
 
 
 class CustomProviderConfig(TypedDict, total=False):
@@ -11,8 +12,8 @@ class CustomProviderConfig(TypedDict, total=False):
     embedding_callable: type[EmbeddingFunction]
 
 
-class CustomProviderSpec(TypedDict):
+class CustomProviderSpec(TypedDict, total=False):
     """Custom provider specification."""
 
-    provider: Literal["custom"]
+    provider: Required[Literal["custom"]]
     config: CustomProviderConfig

src/crewai/rag/embeddings/providers/google/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class GenerativeAiProviderConfig(TypedDict, total=False):
     """Configuration for Google Generative AI provider."""
@@ -27,8 +29,8 @@ class VertexAIProviderConfig(TypedDict, total=False):
     region: Annotated[str, "us-central1"]
 
 
-class VertexAIProviderSpec(TypedDict):
+class VertexAIProviderSpec(TypedDict, total=False):
     """Vertex AI provider specification."""
 
-    provider: Literal["google-vertex"]
+    provider: Required[Literal["google-vertex"]]
     config: VertexAIProviderConfig

src/crewai/rag/embeddings/providers/huggingface/types.py

@@ -2,15 +2,17 @@
 
 from typing import Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class HuggingFaceProviderConfig(TypedDict, total=False):
     """Configuration for HuggingFace provider."""
 
     url: str
 
 
-class HuggingFaceProviderSpec(TypedDict):
+class HuggingFaceProviderSpec(TypedDict, total=False):
     """HuggingFace provider specification."""
 
-    provider: Literal["huggingface"]
+    provider: Required[Literal["huggingface"]]
     config: HuggingFaceProviderConfig

src/crewai/rag/embeddings/providers/ibm/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Any, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class WatsonProviderConfig(TypedDict, total=False):
     """Configuration for Watson provider."""
@@ -35,8 +37,8 @@ class WatsonProviderConfig(TypedDict, total=False):
     proxies: dict
 
 
-class WatsonProviderSpec(TypedDict):
+class WatsonProviderSpec(TypedDict, total=False):
     """Watson provider specification."""
 
-    provider: Literal["watson"]
+    provider: Required[Literal["watson"]]
     config: WatsonProviderConfig

src/crewai/rag/embeddings/providers/instructor/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class InstructorProviderConfig(TypedDict, total=False):
     """Configuration for Instructor provider."""
@@ -11,8 +13,8 @@ class InstructorProviderConfig(TypedDict, total=False):
     instruction: str
 
 
-class InstructorProviderSpec(TypedDict):
+class InstructorProviderSpec(TypedDict, total=False):
     """Instructor provider specification."""
 
-    provider: Literal["instructor"]
+    provider: Required[Literal["instructor"]]
     config: InstructorProviderConfig

src/crewai/rag/embeddings/providers/jina/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class JinaProviderConfig(TypedDict, total=False):
     """Configuration for Jina provider."""
@@ -10,8 +12,8 @@ class JinaProviderConfig(TypedDict, total=False):
     model_name: Annotated[str, "jina-embeddings-v2-base-en"]
 
 
-class JinaProviderSpec(TypedDict):
+class JinaProviderSpec(TypedDict, total=False):
     """Jina provider specification."""
 
-    provider: Literal["jina"]
+    provider: Required[Literal["jina"]]
     config: JinaProviderConfig

src/crewai/rag/embeddings/providers/microsoft/types.py

@@ -2,6 +2,8 @@
 
 from typing import Annotated, Any, Literal, TypedDict
 
+from typing_extensions import Required
+
 
 class AzureProviderConfig(TypedDict, total=False):
     """Configuration for Azure provider."""
@@ -17,8 +19,8 @@ class AzureProviderConfig(TypedDict, total=False):
     organization_id: str
 
 
-class AzureProviderSpec(TypedDict):
+class AzureProviderSpec(TypedDict, total=False):
     """Azure provider specification."""
 
-    provider: Literal["azure"]
+    provider: Required[Literal["azure"]]
     config: AzureProviderConfig