Module naptha_sdk.modules.memory

Classes

class Memory (deployment, *args, **kwargs)

Expand source code

class Memory:
    def __init__(self, 
        deployment,
        *args,
        **kwargs
    ):
        self.deployment = deployment
        self.node_client = NodeClient(self.deployment.node)

    async def run_module(self, module_run: Union[AgentRun, MemoryRunInput]):
        logger.info(f"Running memory module on worker node {self.deployment.node}")
        memory_run = await self.node_client.run_module(module_type="memory", run_input=module_run.model_dict())
        return memory_run

Methods

async def run_module(self,
module_run: AgentRun | MemoryRunInput)

Expand source code

async def run_module(self, module_run: Union[AgentRun, MemoryRunInput]):
    logger.info(f"Running memory module on worker node {self.deployment.node}")
    memory_run = await self.node_client.run_module(module_type="memory", run_input=module_run.model_dict())
    return memory_run