mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-07 07:08:31 +00:00
0b0f2d30ab
* sort imports * update --------- Co-authored-by: Brandon Hancock (bhancock_ai) <109994880+bhancockio@users.noreply.github.com> Co-authored-by: Eduardo Chiarotti <dudumelgaco@hotmail.com>
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import subprocess
|
|
|
|
import click
|
|
|
|
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
|
from crewai.memory.entity.entity_memory import EntityMemory
|
|
from crewai.memory.long_term.long_term_memory import LongTermMemory
|
|
from crewai.memory.short_term.short_term_memory import ShortTermMemory
|
|
from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
|
|
|
|
|
|
def reset_memories_command(
|
|
long,
|
|
short,
|
|
entity,
|
|
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.
|
|
"""
|
|
|
|
try:
|
|
if all:
|
|
ShortTermMemory().reset()
|
|
EntityMemory().reset()
|
|
LongTermMemory().reset()
|
|
TaskOutputStorageHandler().reset()
|
|
KnowledgeStorage().reset()
|
|
click.echo("All memories have been reset.")
|
|
else:
|
|
if long:
|
|
LongTermMemory().reset()
|
|
click.echo("Long term memory has been reset.")
|
|
|
|
if short:
|
|
ShortTermMemory().reset()
|
|
click.echo("Short term memory has been reset.")
|
|
if entity:
|
|
EntityMemory().reset()
|
|
click.echo("Entity memory has been reset.")
|
|
if kickoff_outputs:
|
|
TaskOutputStorageHandler().reset()
|
|
click.echo("Latest Kickoff outputs stored has been reset.")
|
|
if knowledge:
|
|
KnowledgeStorage().reset()
|
|
click.echo("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)
|