mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
ab39753a75
* feat: emit events about memory usage * test: add tests about memory events usage * fixed linter issues * test: use scoped_handlers while listener Memory events
29 lines
1007 B
Python
29 lines
1007 B
Python
from datetime import datetime
|
|
from typing import Any, Dict, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from crewai.utilities.serialization import to_serializable
|
|
|
|
|
|
class BaseEvent(BaseModel):
|
|
"""Base class for all events"""
|
|
|
|
timestamp: datetime = Field(default_factory=datetime.now)
|
|
type: str
|
|
source_fingerprint: Optional[str] = None # UUID string of the source entity
|
|
source_type: Optional[str] = None # "agent", "task", "crew", "memory", "entity_memory", "short_term_memory", "long_term_memory", "external_memory"
|
|
fingerprint_metadata: Optional[Dict[str, Any]] = None # Any relevant metadata
|
|
|
|
def to_json(self, exclude: set[str] | None = None):
|
|
"""
|
|
Converts the event to a JSON-serializable dictionary.
|
|
|
|
Args:
|
|
exclude (set[str], optional): Set of keys to exclude from the result. Defaults to None.
|
|
|
|
Returns:
|
|
dict: A JSON-serializable dictionary.
|
|
"""
|
|
return to_serializable(self, exclude=exclude)
|