fix task cloning error

Co-authored-by: João Moura <joaomdmoura@gmail.com>

mkdocs.yml

@@ -209,7 +209,7 @@ extra:
     provider: google
     property: G-N3Q505TMQ6
   social:
-    - icon: fontawesome/brands/twitter
+    - icon: fontawesome/brands/x-twitter
       link: https://x.com/crewAIInc
     - icon: fontawesome/brands/github
       link: https://github.com/crewAIInc/crewAI

src/crewai/__init__.py

@@ -1,5 +1,4 @@
 import warnings
-
 from crewai.agent import Agent
 from crewai.crew import Crew
 from crewai.flow.flow import Flow
@@ -15,5 +14,6 @@ warnings.filterwarnings(
     category=UserWarning,
     module="pydantic.main",
 )
-
+__version__ = "0.65.2"
 __all__ = ["Agent", "Crew", "Process", "Task", "Pipeline", "Router", "LLM", "Flow"]
+

src/crewai/agent.py

@@ -1,18 +1,19 @@
 import os
 from inspect import signature
 from typing import Any, List, Optional, Union
+
 from pydantic import Field, InstanceOf, PrivateAttr, model_validator
 
 from crewai.agents import CacheHandler
-from crewai.utilities import Converter, Prompts
-from crewai.tools.agent_tools import AgentTools
-from crewai.agents.crew_agent_executor import CrewAgentExecutor
 from crewai.agents.agent_builder.base_agent import BaseAgent
-from crewai.memory.contextual.contextual_memory import ContextualMemory
-from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_FILE
-from crewai.utilities.training_handler import CrewTrainingHandler
-from crewai.utilities.token_counter_callback import TokenCalcHandler
+from crewai.agents.crew_agent_executor import CrewAgentExecutor
 from crewai.llm import LLM
+from crewai.memory.contextual.contextual_memory import ContextualMemory
+from crewai.tools.agent_tools import AgentTools
+from crewai.utilities import Converter, Prompts
+from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_FILE
+from crewai.utilities.token_counter_callback import TokenCalcHandler
+from crewai.utilities.training_handler import CrewTrainingHandler
 
 
 def mock_agent_ops_provider():
@@ -292,9 +293,9 @@ class Agent(BaseAgent):
             step_callback=self.step_callback,
             function_calling_llm=self.function_calling_llm,
             respect_context_window=self.respect_context_window,
-            request_within_rpm_limit=self._rpm_controller.check_or_wait
-            if self._rpm_controller
-            else None,
+            request_within_rpm_limit=(
+                self._rpm_controller.check_or_wait if self._rpm_controller else None
+            ),
             callbacks=[TokenCalcHandler(self._token_process)],
         )
 

src/crewai/crew.py

@@ -900,7 +900,22 @@ class Crew(BaseModel):
         }
 
         cloned_agents = [agent.copy() for agent in self.agents]
-        cloned_tasks = [task.copy(cloned_agents) for task in self.tasks]
+
+        task_mapping = {}
+
+        cloned_tasks = []
+        for task in self.tasks:
+            cloned_task = task.copy(cloned_agents, task_mapping)
+            cloned_tasks.append(cloned_task)
+            task_mapping[task.key] = cloned_task
+
+        for cloned_task, original_task in zip(cloned_tasks, self.tasks):
+            if original_task.context:
+                cloned_context = [
+                    task_mapping[context_task.key]
+                    for context_task in original_task.context
+                ]
+                cloned_task.context = cloned_context
 
         copied_data = self.model_dump(exclude=exclude)
         copied_data = {k: v for k, v in copied_data.items() if v is not None}

src/crewai/memory/storage/rag_storage.py

@@ -5,11 +5,6 @@ import os
 import shutil
 from typing import Any, Dict, List, Optional
 
-from embedchain import App
-from embedchain.llm.base import BaseLlm
-from embedchain.models.data_type import DataType
-from embedchain.vectordb.chroma import InvalidDimensionException
-
 from crewai.memory.storage.interface import Storage
 from crewai.utilities.paths import db_storage_path
 
@@ -29,10 +24,6 @@ def suppress_logging(
     logger.setLevel(original_level)
 
 
-class FakeLLM(BaseLlm):
-    pass
-
-
 class RAGStorage(Storage):
     """
     Extends Storage to handle embeddings for memory entries, improving
@@ -74,9 +65,19 @@ class RAGStorage(Storage):
         if embedder_config:
             config["embedder"] = embedder_config
         self.type = type
-        self.app = App.from_config(config=config)
+        self.config = config
+        self.allow_reset = allow_reset
+
+    def _initialize_app(self):
+        from embedchain import App
+        from embedchain.llm.base import BaseLlm
+
+        class FakeLLM(BaseLlm):
+            pass
+
+        self.app = App.from_config(config=self.config)
         self.app.llm = FakeLLM()
-        if allow_reset:
+        if self.allow_reset:
             self.app.reset()
 
     def _sanitize_role(self, role: str) -> str:
@@ -86,6 +87,8 @@ class RAGStorage(Storage):
         return role.replace("\n", "").replace(" ", "_").replace("/", "_")
 
     def save(self, value: Any, metadata: Dict[str, Any]) -> None:
+        if not hasattr(self, "app"):
+            self._initialize_app()
         self._generate_embedding(value, metadata)
 
     def search(  # type: ignore # BUG?: Signature of "search" incompatible with supertype "Storage"
@@ -95,6 +98,10 @@ class RAGStorage(Storage):
         filter: Optional[dict] = None,
         score_threshold: float = 0.35,
     ) -> List[Any]:
+        if not hasattr(self, "app"):
+            self._initialize_app()
+        from embedchain.vectordb.chroma import InvalidDimensionException
+
         with suppress_logging():
             try:
                 results = (
@@ -108,6 +115,10 @@ class RAGStorage(Storage):
         return [r for r in results if r["metadata"]["score"] >= score_threshold]
 
     def _generate_embedding(self, text: str, metadata: Dict[str, Any]) -> Any:
+        if not hasattr(self, "app"):
+            self._initialize_app()
+        from embedchain.models.data_type import DataType
+
         self.app.add(text, data_type=DataType.TEXT, metadata=metadata)
 
     def reset(self) -> None:

src/crewai/task.py

@@ -276,9 +276,7 @@ class Task(BaseModel):
             content = (
                 json_output
                 if json_output
-                else pydantic_output.model_dump_json()
-                if pydantic_output
-                else result
+                else pydantic_output.model_dump_json() if pydantic_output else result
             )
             self._save_file(content)
 
@@ -319,7 +317,9 @@ class Task(BaseModel):
             self.processed_by_agents.add(agent_name)
         self.delegations += 1
 
-    def copy(self, agents: List["BaseAgent"]) -> "Task":
+    def copy(
+        self, agents: List["BaseAgent"], task_mapping: Dict[str, "Task"]
+    ) -> "Task":
         """Create a deep copy of the Task."""
         exclude = {
             "id",
@@ -332,7 +332,9 @@ class Task(BaseModel):
         copied_data = {k: v for k, v in copied_data.items() if v is not None}
 
         cloned_context = (
-            [task.copy(agents) for task in self.context] if self.context else None
+            [task_mapping[context_task.key] for context_task in self.context]
+            if self.context
+            else None
         )
 
         def get_agent_by_role(role: str) -> Union["BaseAgent", None]:

src/crewai/telemetry/telemetry.py

@@ -21,9 +21,7 @@ with suppress_warnings():
 
 
 from opentelemetry import trace  # noqa: E402
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
-    OTLPSpanExporter,  # noqa: E402
-)
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter  # noqa: E402
 from opentelemetry.sdk.resources import SERVICE_NAME, Resource  # noqa: E402
 from opentelemetry.sdk.trace import TracerProvider  # noqa: E402
 from opentelemetry.sdk.trace.export import BatchSpanProcessor  # noqa: E402

src/crewai/tools/agent_tools.py

@@ -1,4 +1,3 @@
-from langchain.tools import StructuredTool
 from crewai.agents.agent_builder.utilities.base_agent_tool import BaseAgentTools
 
 
@@ -6,6 +5,8 @@ class AgentTools(BaseAgentTools):
     """Default tools around agent delegation"""
 
     def tools(self):
+        from langchain.tools import StructuredTool
+
         coworkers = ", ".join([f"{agent.role}" for agent in self.agents])
         tools = [
             StructuredTool.from_function(

src/crewai/tools/cache_tools.py

@@ -1,4 +1,3 @@
-from langchain.tools import StructuredTool
 from pydantic import BaseModel, Field
 
 from crewai.agents.cache import CacheHandler
@@ -14,6 +13,8 @@ class CacheTools(BaseModel):
     )
 
     def tool(self):
+        from langchain.tools import StructuredTool
+
         return StructuredTool.from_function(
             func=self.hit_cache,
             name=self.name,

src/crewai/utilities/internal_instructor.py

@@ -1,8 +1,5 @@
 from typing import Any, Optional, Type
 
-import instructor
-from litellm import completion
-
 
 class InternalInstructor:
     """Class that wraps an agent llm with instructor."""
@@ -28,6 +25,10 @@ class InternalInstructor:
         if self.agent and not self.llm:
             self.llm = self.agent.function_calling_llm or self.agent.llm
 
+        # Lazy import
+        import instructor
+        from litellm import completion
+
         self._client = instructor.from_litellm(
             completion,
             mode=instructor.Mode.TOOLS,

src/crewai/utilities/token_counter_callback.py

@@ -1,4 +1,5 @@
 from litellm.integrations.custom_logger import CustomLogger
+
 from crewai.agents.agent_builder.utilities.base_token_process import TokenProcess