mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
30ef8ed70b
* CLI command added * Added reset agent knowledge function * Reduced verbose * Added test cases * Added docs * Llama test case failing * Changed _reset_agent_knowledge function * Fixed new line error * Added docs * fixed the new line error * Refractored * Uncommmented some test cases * ruff check fixed * fixed run type checks * fixed run type checks * fixed run type checks * Made reset_fn callable by casting to silence run type checks * Changed the reset_knowledge as it expects only list of knowledge * Fixed typo in docs * Refractored the memory_system * Minor Changes * fixed test case * Fixed linting issues * Network test cases failing --------- Co-authored-by: Lucas Gomide <lucaslg200@gmail.com>
84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
import subprocess
|
|
|
|
import click
|
|
|
|
from crewai.cli.utils import get_crews
|
|
|
|
|
|
def reset_memories_command(
|
|
long,
|
|
short,
|
|
entity,
|
|
knowledge,
|
|
agent_knowledge,
|
|
kickoff_outputs,
|
|
all,
|
|
) -> None:
|
|
"""
|
|
Reset the crew memories.
|
|
|
|
Args:
|
|
long (bool): Whether to reset the long-term memory.
|
|
short (bool): Whether to reset the short-term memory.
|
|
entity (bool): Whether to reset the entity memory.
|
|
kickoff_outputs (bool): Whether to reset the latest kickoff task outputs.
|
|
all (bool): Whether to reset all memories.
|
|
knowledge (bool): Whether to reset the knowledge.
|
|
agent_knowledge (bool): Whether to reset the agents knowledge.
|
|
"""
|
|
|
|
try:
|
|
if not any([long, short, entity, kickoff_outputs, knowledge, agent_knowledge, all]):
|
|
click.echo(
|
|
"No memory type specified. Please specify at least one type to reset."
|
|
)
|
|
return
|
|
|
|
crews = get_crews()
|
|
if not crews:
|
|
raise ValueError("No crew found.")
|
|
for crew in crews:
|
|
if all:
|
|
crew.reset_memories(command_type="all")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Reset memories command has been completed."
|
|
)
|
|
continue
|
|
if long:
|
|
crew.reset_memories(command_type="long")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Long term memory has been reset."
|
|
)
|
|
if short:
|
|
crew.reset_memories(command_type="short")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Short term memory has been reset."
|
|
)
|
|
if entity:
|
|
crew.reset_memories(command_type="entity")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Entity memory has been reset."
|
|
)
|
|
if kickoff_outputs:
|
|
crew.reset_memories(command_type="kickoff_outputs")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Latest Kickoff outputs stored has been reset."
|
|
)
|
|
if knowledge:
|
|
crew.reset_memories(command_type="knowledge")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Knowledge has been reset."
|
|
)
|
|
if agent_knowledge:
|
|
crew.reset_memories(command_type="agent_knowledge")
|
|
click.echo(
|
|
f"[Crew ({crew.name if crew.name else crew.id})] Agents knowledge has been reset."
|
|
)
|
|
|
|
except subprocess.CalledProcessError as e:
|
|
click.echo(f"An error occurred while resetting the memories: {e}", err=True)
|
|
click.echo(e.output, err=True)
|
|
|
|
except Exception as e:
|
|
click.echo(f"An unexpected error occurred: {e}", err=True)
|