Add in missing triple quote and execution time to resume agent functionality. (#1025)

* Add in missing triple quote and execution time to resume agent functionality

* Fixing broken kwargs and other issues causing our tests to fail

src/crewai/agent.py

@@ -55,6 +55,9 @@ class Agent(BaseAgent):
             tools: Tools at agents disposal
             step_callback: Callback to be executed after each step of the agent execution.
             callbacks: A list of callback functions from the langchain library that are triggered during the agent's execution process
+    """
+
+    _times_executed: int = PrivateAttr(default=0)
     max_execution_time: Optional[int] = Field(
         default=None,
         description="Maximum execution time for an agent to execute a task",
@@ -186,6 +189,20 @@ class Agent(BaseAgent):
         else:
             task_prompt = self._use_trained_data(task_prompt=task_prompt)
 
+        try:
+            result = self.agent_executor.invoke(
+                {
+                    "input": task_prompt,
+                    "tool_names": self.agent_executor.tools_names,
+                    "tools": self.agent_executor.tools_description,
+                }
+            )["output"]
+        except Exception as e:
+            self._times_executed += 1
+            if self._times_executed > self.max_retry_limit:
+                raise e
+            result = self.execute_task(task, context, tools)
+
         if self.max_rpm:
             self._rpm_controller.stop_rpm_counter()
 

src/crewai/memory/short_term/short_term_memory.py

@@ -19,7 +19,7 @@ class ShortTermMemory(Memory):
         super().__init__(storage)
 
     def save(self, item: ShortTermMemoryItem) -> None:
-        super().save(item.data, item.metadata, item.agent)
+        super().save(value=item.data, metadata=item.metadata, agent=item.agent)
 
     def search(self, query: str, score_threshold: float = 0.35):
         return self.storage.search(query=query, score_threshold=score_threshold)  # type: ignore # BUG? The reference is to the parent class, but the parent class does not have this parameters

src/crewai/task.py

@@ -355,6 +355,9 @@ class Task(BaseModel):
         return OutputFormat.RAW
 
     def _save_file(self, result: Any) -> None:
+        if self.output_file is None:
+            raise ValueError("output_file is not set.")
+
         directory = os.path.dirname(self.output_file)  # type: ignore # Value of type variable "AnyOrLiteralStr" of "dirname" cannot be "str | None"
 
         if directory and not os.path.exists(directory):
@@ -363,6 +366,7 @@ class Task(BaseModel):
         with open(self.output_file, "w", encoding="utf-8") as file:
             if isinstance(result, dict):
                 import json
+
                 json.dump(result, file, ensure_ascii=False, indent=2)
             else:
                 file.write(str(result))