diff --git a/reme/memory/vector_tools/__init__.py b/reme/memory/vector_tools/__init__.py index b6601c1c..9fb04923 100644 --- a/reme/memory/vector_tools/__init__.py +++ b/reme/memory/vector_tools/__init__.py @@ -1,5 +1,7 @@ """memory tools""" +# pylint: disable=no-name-in-module + from .base_memory_tool import BaseMemoryTool # chunk tools diff --git a/reme/memory/vector_tools/profiles/__init__.py b/reme/memory/vector_tools/profiles/__init__.py index e69de29b..150455db 100644 --- a/reme/memory/vector_tools/profiles/__init__.py +++ b/reme/memory/vector_tools/profiles/__init__.py @@ -0,0 +1 @@ +"""Profile memory tools.""" diff --git a/reme/memory/vector_tools/profiles/profile_handler.py b/reme/memory/vector_tools/profiles/profile_handler.py index de5cddf4..0df3707a 100644 --- a/reme/memory/vector_tools/profiles/profile_handler.py +++ b/reme/memory/vector_tools/profiles/profile_handler.py @@ -1,5 +1,7 @@ """Profile handler facade for filesystem and vector backends.""" +# pylint: disable=missing-function-docstring + import asyncio from pathlib import Path @@ -66,7 +68,9 @@ class ProfileHandler: asyncio.get_running_loop() except RuntimeError: return asyncio.run(coro) - raise RuntimeError("Synchronous profile access is not available in an active event loop. Use async methods instead.") + raise RuntimeError( + "Synchronous profile access is not available in an active event loop. Use async methods instead.", + ) async def adelete(self, profile_id: str | list[str]) -> bool | int: return await self.backend.delete(profile_id) @@ -74,7 +78,13 @@ class ProfileHandler: async def adelete_all(self) -> int: return await self.backend.delete_all() - async def aadd(self, message_time: str, profile_key: str, profile_value: str, ref_memory_id: str = "") -> MemoryNode: + async def aadd( + self, + message_time: str, + profile_key: str, + profile_value: str, + ref_memory_id: str = "", + ) -> MemoryNode: return await self.backend.add(message_time, profile_key, profile_value, ref_memory_id) async def aadd_batch(self, profiles: list[dict], ref_memory_id: str = "") -> list[MemoryNode]: diff --git a/reme/memory/vector_tools/profiles/update_profile.py b/reme/memory/vector_tools/profiles/update_profile.py index 54e59372..04c932bf 100644 --- a/reme/memory/vector_tools/profiles/update_profile.py +++ b/reme/memory/vector_tools/profiles/update_profile.py @@ -98,7 +98,10 @@ class UpdateProfile(BaseMemoryTool): # Add profiles for each target for target, target_profiles in profiles_by_target.items(): profile_handler = self.get_profile_handler(target) - new_nodes = await profile_handler.aadd_batch(profiles=target_profiles, ref_memory_id=self.history_id) + new_nodes = await profile_handler.aadd_batch( + profiles=target_profiles, + ref_memory_id=self.history_id, + ) self.memory_nodes.extend(new_nodes) added_count += len(new_nodes) else: diff --git a/reme/reme.py b/reme/reme.py index cc4ed4cd..c7c04d2e 100644 --- a/reme/reme.py +++ b/reme/reme.py @@ -238,7 +238,9 @@ class ReMe(Application): if self.profile_store_name in vector_store_configs: if self.profile_collection_name: - vector_store_configs[self.profile_store_name] = vector_store_configs[self.profile_store_name].model_copy( + vector_store_configs[self.profile_store_name] = vector_store_configs[ + self.profile_store_name + ].model_copy( update={"collection_name": profile_collection_name}, ) return