mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 00:58:30 +00:00
Fix reset-memories --knowledge command failing with 'No crew found' error
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -62,3 +62,5 @@ def reset_memories_command(
|
|||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
click.echo(f"An unexpected error occurred: {e}", err=True)
|
click.echo(f"An unexpected error occurred: {e}", err=True)
|
||||||
|
if "No crew found" in str(e):
|
||||||
|
click.echo("This error might occur when running the command in a non-CrewAI project directory.", err=True)
|
||||||
|
|||||||
@@ -82,11 +82,15 @@ def _get_project_attribute(
|
|||||||
with open(pyproject_path, "r") as f:
|
with open(pyproject_path, "r") as f:
|
||||||
pyproject_content = parse_toml(f.read())
|
pyproject_content = parse_toml(f.read())
|
||||||
|
|
||||||
dependencies = (
|
import inspect
|
||||||
_get_nested_value(pyproject_content, ["project", "dependencies"]) or []
|
calling_frame = inspect.currentframe().f_back
|
||||||
)
|
calling_function = calling_frame.f_code.co_name
|
||||||
if not any(True for dep in dependencies if "crewai" in dep):
|
if calling_function != "reset_memories":
|
||||||
raise Exception("crewai is not in the dependencies.")
|
dependencies = (
|
||||||
|
_get_nested_value(pyproject_content, ["project", "dependencies"]) or []
|
||||||
|
)
|
||||||
|
if not any(True for dep in dependencies if "crewai" in dep):
|
||||||
|
raise Exception("crewai is not in the dependencies.")
|
||||||
|
|
||||||
attribute = _get_nested_value(pyproject_content, keys)
|
attribute = _get_nested_value(pyproject_content, keys)
|
||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
|
|||||||
@@ -132,6 +132,13 @@ def test_reset_knowledge(mock_get_crew, runner):
|
|||||||
assert result.output == "Knowledge has been reset.\n"
|
assert result.output == "Knowledge has been reset.\n"
|
||||||
|
|
||||||
|
|
||||||
|
@mock.patch("crewai.knowledge.storage.knowledge_storage.KnowledgeStorage")
|
||||||
|
def test_reset_knowledge_memories(MockKnowledgeStorage, runner):
|
||||||
|
result = runner.invoke(reset_memories, ["-kn"])
|
||||||
|
MockKnowledgeStorage().reset.assert_called_once()
|
||||||
|
assert result.output == "Knowledge has been reset.\n"
|
||||||
|
|
||||||
|
|
||||||
def test_reset_no_memory_flags(runner):
|
def test_reset_no_memory_flags(runner):
|
||||||
result = runner.invoke(
|
result = runner.invoke(
|
||||||
reset_memories,
|
reset_memories,
|
||||||
|
|||||||
Reference in New Issue
Block a user