diff --git a/mempalace/convo_miner.py b/mempalace/convo_miner.py index 0034d70..aae3f71 100644 --- a/mempalace/convo_miner.py +++ b/mempalace/convo_miner.py @@ -352,7 +352,7 @@ def mine_convos( chunk_room = chunk.get("memory_type", room) if extract_mode == "general" else room if extract_mode == "general": room_counts[chunk_room] += 1 - drawer_id = f"drawer_{wing}_{chunk_room}_{hashlib.md5((source_file + str(chunk['chunk_index'])).encode()).hexdigest()[:16]}" + drawer_id = f"drawer_{wing}_{chunk_room}_{hashlib.md5((source_file + str(chunk['chunk_index'])).encode(), usedforsecurity=False).hexdigest()[:16]}" try: collection.add( documents=[chunk["content"]], diff --git a/mempalace/miner.py b/mempalace/miner.py index ecd313d..70f715f 100644 --- a/mempalace/miner.py +++ b/mempalace/miner.py @@ -202,7 +202,7 @@ def add_drawer( collection, wing: str, room: str, content: str, source_file: str, chunk_index: int, agent: str ): """Add one drawer to the palace.""" - drawer_id = f"drawer_{wing}_{room}_{hashlib.md5((source_file + str(chunk_index)).encode()).hexdigest()[:16]}" + drawer_id = f"drawer_{wing}_{room}_{hashlib.md5((source_file + str(chunk_index)).encode(), usedforsecurity=False).hexdigest()[:16]}" try: collection.add( documents=[content],