fix: fix typing check

This commit is contained in:
lucasgomide
2025-04-01 18:51:28 -03:00
committed by Lucas Gomide
parent 135596a0dc
commit a9a329a57e
2 changed files with 10 additions and 6 deletions

View File

@@ -53,7 +53,8 @@ class CrewAgentExecutorMixin:
self.crew
and self.agent
and self.task
and getattr(self.crew, "_external_memory", None)
and hasattr(self.crew, "_external_memory")
and self.crew._external_memory
):
try:
self.crew._external_memory.save(

View File

@@ -1,32 +1,35 @@
from typing import Any, Dict, Optional, Self
from typing import TYPE_CHECKING, Any, Dict, Optional, Self
from crewai.memory.external.external_memory_item import ExternalMemoryItem
from crewai.memory.memory import Memory
from crewai.memory.storage.interface import Storage
if TYPE_CHECKING:
from crewai.memory.storage.mem0_storage import Mem0Storage
class ExternalMemory(Memory):
def __init__(self, storage: Optional[Storage] = None, **data: Any):
super().__init__(storage=storage, **data)
@staticmethod
def _configure_mem0(crew, config) -> "Mem0Storage":
def _configure_mem0(crew: Any, config: Dict[str, Any]) -> "Mem0Storage":
from crewai.memory.storage.mem0_storage import Mem0Storage
return Mem0Storage(type="external", crew=crew, config=config)
@staticmethod
def external_supported_storages():
def external_supported_storages() -> Dict[str, Any]:
return {
"mem0": ExternalMemory._configure_mem0,
}
@staticmethod
def create_storage(crew, embedder_config):
def create_storage(crew: Any, embedder_config: Optional[Dict[str, Any]]) -> Storage:
if not embedder_config:
raise ValueError("embedder_config is required")
if embedder_config and "provider" not in embedder_config:
if "provider" not in embedder_config:
raise ValueError("embedder_config must include a 'provider' key")
provider = embedder_config["provider"]