feat: fix tests

docs/getting-started/Start-a-New-CrewAI-Project-Template-Method.md

@@ -17,7 +17,7 @@ Beforre we start there are a couple of things to note:
 Before getting started with CrewAI, make sure that you have installed it via pip:
 
 ```shell
-$ pip install crewai crewai-tools
+$ pip install crewai crewi-tools
 ```
 
 ### Virtual Environemnts

src/crewai/agents/agent_builder/base_agent_executor_mixin.py

@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING, Optional
 
 from crewai.memory.entity.entity_memory_item import EntityMemoryItem
 from crewai.memory.long_term.long_term_memory_item import LongTermMemoryItem
+from crewai.memory.short_term.short_term_memory_item import ShortTermMemoryItem
 from crewai.utilities.converter import ConverterError
 from crewai.utilities.evaluators.task_evaluator import TaskEvaluator
 from crewai.utilities import I18N
@@ -38,17 +39,18 @@ class CrewAgentExecutorMixin:
             and "Action: Delegate work to coworker" not in output.log
         ):
             try:
+                memory = ShortTermMemoryItem(
+                    data=output.log,
+                    agent=self.crew_agent.role,
+                    metadata={
+                        "observation": self.task.description,
+                    },
+                )
                 if (
                     hasattr(self.crew, "_short_term_memory")
                     and self.crew._short_term_memory
                 ):
-                    self.crew._short_term_memory.save(
-                        value=output.log,
-                        metadata={
-                            "observation": self.task.description,
-                        },
-                        agent=self.crew_agent.role,
-                    )
+                    self.crew._short_term_memory.save(memory)
             except Exception as e:
                 print(f"Failed to add to short term memory: {e}")
                 pass

src/crewai/memory/short_term/short_term_memory.py

@@ -1,4 +1,3 @@
-from typing import Any, Dict, Optional
 from crewai.memory.memory import Memory
 from crewai.memory.short_term.short_term_memory_item import ShortTermMemoryItem
 from crewai.memory.storage.rag_storage import RAGStorage
@@ -19,14 +18,7 @@ class ShortTermMemory(Memory):
         )
         super().__init__(storage)
 
-    def save(
-        self,
-        value: Any,
-        metadata: Optional[Dict[str, Any]] = None,
-        agent: Optional[str] = None,
-    ) -> None:
-        item = ShortTermMemoryItem(data=value, metadata=metadata, agent=agent)
-
+    def save(self, item: ShortTermMemoryItem) -> None:
         super().save(value=item.data, metadata=item.metadata, agent=item.agent)
 
     def search(self, query: str, score_threshold: float = 0.35):

src/crewai/memory/short_term/short_term_memory_item.py

@@ -3,10 +3,7 @@ from typing import Any, Dict, Optional
 
 class ShortTermMemoryItem:
     def __init__(
-        self,
-        data: Any,
-        agent: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        self, data: Any, agent: str, metadata: Optional[Dict[str, Any]] = None
     ):
         self.data = data
         self.agent = agent

src/crewai/memory/storage/interface.py

@@ -4,7 +4,7 @@ from typing import Any, Dict
 class Storage:
     """Abstract base class defining the storage interface"""
 
-    def save(self, value: Any, metadata: Dict[str, Any]) -> None:
+    def save(self, key: str, value: Any, metadata: Dict[str, Any]) -> None:
         pass
 
     def search(self, key: str) -> Dict[str, Any]:  # type: ignore

src/crewai/telemetry/telemetry.py

@@ -40,7 +40,7 @@ class Telemetry:
     - Roles of agents in a crew
     - Tools names available
 
-    Users can opt-in to sharing more complete data using the `share_crew`
+    Users can opt-in to sharing more complete data suing the `share_crew`
     attribute in the Crew class.
     """
 

tests/memory/short_term_memory_test.py

@@ -23,7 +23,10 @@ def short_term_memory():
         expected_output="A list of relevant URLs based on the search query.",
         agent=agent,
     )
-    return ShortTermMemory(crew=Crew(agents=[agent], tasks=[task]))
+    return ShortTermMemory(crew=Crew(
+        agents=[agent],
+        tasks=[task]
+    ))
 
 
 @pytest.mark.vcr(filter_headers=["authorization"])
@@ -35,11 +38,7 @@ def test_save_and_search(short_term_memory):
         agent="test_agent",
         metadata={"task": "test_task"},
     )
-    short_term_memory.save(
-        value=memory.data,
-        metadata=memory.metadata,
-        agent=memory.agent,
-    )
+    short_term_memory.save(memory)
 
     find = short_term_memory.search("test value", score_threshold=0.01)[0]
     assert find["context"] == memory.data, "Data value mismatch."