Spaces:

Oldmangrizzz
/

Agent-Node

Paused

Agent-Node / agent_zero_framework /python /tools /memory_save.py

Upload folder using huggingface_hub

dff1e71 verified 10 months ago

509 Bytes

from python.helpers.memory import Memory
from python.helpers.tool import Tool, Response


class MemorySave(Tool):

    async def execute(self, text="", area="", **kwargs):

        if not area:
            area = Memory.Area.MAIN.value

        metadata = {"area": area, **kwargs}

        db = await Memory.get(self.agent)
        id = await db.insert_text(text, metadata)

        result = self.agent.read_prompt("fw.memory_saved.md", memory_id=id)
        return Response(message=result, break_loop=False)

	from python.helpers.memory import Memory
	from python.helpers.tool import Tool, Response


	class MemorySave(Tool):

	async def execute(self, text="", area="", **kwargs):

	if not area:
	area = Memory.Area.MAIN.value

	metadata = {"area": area, **kwargs}

	db = await Memory.get(self.agent)
	id = await db.insert_text(text, metadata)

	result = self.agent.read_prompt("fw.memory_saved.md", memory_id=id)
	return Response(message=result, break_loop=False)