diff --git a/src/crewai/cli/cli.py b/src/crewai/cli/cli.py index 6f9cbd423..243502b43 100644 --- a/src/crewai/cli/cli.py +++ b/src/crewai/cli/cli.py @@ -276,7 +276,13 @@ def tool_install(handle: str): @tool.command(name="publish") -@click.option("--force", is_flag=True, show_default=True, default=False, help="Bypasses Git remote validations") +@click.option( + "--force", + is_flag=True, + show_default=True, + default=False, + help="Bypasses Git remote validations", +) @click.option("--public", "is_public", flag_value=True, default=False) @click.option("--private", "is_public", flag_value=False) def tool_publish(is_public: bool, force: bool): diff --git a/src/crewai/memory/entity/entity_memory.py b/src/crewai/memory/entity/entity_memory.py index 2a199349a..d4f8d9aae 100644 --- a/src/crewai/memory/entity/entity_memory.py +++ b/src/crewai/memory/entity/entity_memory.py @@ -15,7 +15,10 @@ class EntityMemory(Memory): storage if storage else RAGStorage( - type="entities", allow_reset=False, embedder_config=embedder_config, crew=crew + type="entities", + allow_reset=False, + embedder_config=embedder_config, + crew=crew, ) ) super().__init__(storage) diff --git a/tests/agent_test.py b/tests/agent_test.py index 815702680..a2c020f90 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -276,7 +276,9 @@ def test_cache_hitting(): "multiplier-{'first_number': 12, 'second_number': 3}": 36, } - with patch.object(CacheHandler, "read") as read, patch.object(Emitter, "emit") as emit: + with patch.object(CacheHandler, "read") as read, patch.object( + Emitter, "emit" + ) as emit: read.return_value = "0" task = Task( description="What is 2 times 6? Ignore correctness and just return the result of the multiplication tool, you must use the tool.",