andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-07-27 09:45:16 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

feat: add memory save event handling to activity log

Browse Source 
Implemented event handlers for MemorySaveStartedEvent, MemorySaveCompletedEvent, and MemorySaveFailedEvent in the crew_run_tui module. This allows the application to log memory save operations, capturing their status and details in the activity log. Added corresponding tests to verify the correct logging behavior for successful and failed memory saves.
This commit is contained in:
Joao Moura
2026-06-17 23:39:14 -07:00
parent 6c82b0c735
commit f9301fad02
4 changed files with 227 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
lib/cli/src/crewai_cli/crew_run_tui.py
View File

@@ -1923,8 +1923,113 @@ FooterKey .footer-key--key {
MemoryRetrievalCompletedEvent,
MemoryRetrievalFailedEvent,
MemoryRetrievalStartedEvent,
MemorySaveCompletedEvent,
MemorySaveFailedEvent,
MemorySaveStartedEvent,
)
@crewai_event_bus.on(MemorySaveStartedEvent)
def on_memory_save_started(
source: Any, event: MemorySaveStartedEvent
) -> None:
with self._lock:
for entry in reversed(self._log_entries):
if (
entry["tool_name"] == "memory_save"
and entry.get("started_event_id") == event.event_id
):
entry["args"] = event.value
return
self._log_entries.append(
{
"tool_name": "memory_save",
"status": "running",
"args": event.value,
"result": None,
"error": None,
"start_time": time.time(),
"duration": None,
"task_idx": self._current_task_idx,
"event_id": event.event_id,
}
)
self._register_handler(MemorySaveStartedEvent, on_memory_save_started)
@crewai_event_bus.on(MemorySaveCompletedEvent)
def on_memory_save_completed(
source: Any, event: MemorySaveCompletedEvent
) -> None:
with self._lock:
for entry in reversed(self._log_entries):
if (
entry["tool_name"] == "memory_save"
and entry["status"] == "running"
and (
event.started_event_id is None
or entry.get("event_id") == event.started_event_id
)
):
entry["status"] = "success"
entry["duration"] = event.save_time_ms / 1000
entry["result"] = event.value
entry["started_event_id"] = event.started_event_id
break
else:
self._log_entries.append(
{
"tool_name": "memory_save",
"status": "success",
"args": None,
"result": event.value,
"error": None,
"start_time": time.time(),
"duration": event.save_time_ms / 1000,
"task_idx": self._current_task_idx,
"started_event_id": event.started_event_id,
}
)
self._register_handler(MemorySaveCompletedEvent, on_memory_save_completed)
@crewai_event_bus.on(MemorySaveFailedEvent)
def on_memory_save_failed(
source: Any, event: MemorySaveFailedEvent
) -> None:
with self._lock:
for idx, entry in reversed(list(enumerate(self._log_entries))):
if (
entry["tool_name"] == "memory_save"
and entry["status"] == "running"
and (
event.started_event_id is None
or entry.get("event_id") == event.started_event_id
)
):
entry["status"] = "error"
entry["error"] = event.error
entry["duration"] = time.time() - entry["start_time"]
entry["started_event_id"] = event.started_event_id
self._log_expanded.add(idx)
break
else:
self._log_entries.append(
{
"tool_name": "memory_save",
"status": "error",
"args": event.value,
"result": None,
"error": event.error,
"start_time": time.time(),
"duration": 0,
"task_idx": self._current_task_idx,
"started_event_id": event.started_event_id,
}
)
self._log_expanded.add(len(self._log_entries) - 1)
self._register_handler(MemorySaveFailedEvent, on_memory_save_failed)
@crewai_event_bus.on(MemoryRetrievalStartedEvent)
def on_memory_retrieval_started(
source: Any, event: MemoryRetrievalStartedEvent