lint: fix ruff E501/F841 violations and apply ruff format

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: 2 people

.gitignore

@@ -35,3 +35,10 @@ venv/
 
 # macOS
 .DS_Store
+
+# JetBrains IDEs
+.idea/
+
+# Local scratch notes
+Agent Memory Toolkit.txt
+Cosmos AI Memory Service.txt

agent_memory_toolkit/_base/base_client.py

@@ -208,8 +208,7 @@ def _require_cosmos(self) -> None:
             raise CosmosNotConnectedError()
 
     def _warn_on_embedding_dim_mismatch(self, container: Any = None) -> None:
-        """Log a warning when the configured embedding dim differs from the container policy.
-        """
+        """Log a warning when the configured embedding dim differs from the container policy."""
         container = container if container is not None else self._container_client
         if container is None or self._embedding_dimensions is None:
             return

agent_memory_toolkit/_counters.py

@@ -28,13 +28,14 @@
 
 from __future__ import annotations
 
-from agent_memory_toolkit.logging import get_logger
 from datetime import datetime, timezone
 from typing import Any, Optional
 
 from azure.core import MatchConditions
 from azure.cosmos.exceptions import CosmosHttpResponseError, CosmosResourceNotFoundError
 
+from agent_memory_toolkit.logging import get_logger
+
 logger = get_logger(__name__)
 
 USER_COUNTER_THREAD_ID = "__counters__"

agent_memory_toolkit/_utils.py

@@ -15,7 +15,8 @@
 
 from ._query_builder import _QueryBuilder
 from .exceptions import ConfigurationError, ValidationError
-from .thresholds import DEFAULT_TTL_BY_TYPE as DEFAULT_TTL_BY_TYPE, default_ttl_for
+from .thresholds import DEFAULT_TTL_BY_TYPE as DEFAULT_TTL_BY_TYPE
+from .thresholds import default_ttl_for
 
 VALID_ROLES = {"agent", "user", "tool", "system"}
 VALID_TYPES = {"turn", "summary", "fact", "user_summary", "procedural", "episodic"}
@@ -58,6 +59,7 @@ def new_user_summary_id() -> str:
     """Return a fresh ``user_summary_*`` id."""
     return new_id("user_summary")
 
+
 _WHITESPACE_RE = re.compile(r"\s+")
 
 

agent_memory_toolkit/aio/auto_trigger.py

@@ -7,14 +7,14 @@
 
 import asyncio
 import inspect
-from agent_memory_toolkit.logging import get_logger
 from collections.abc import Callable
 from typing import Any
 
 from agent_memory_toolkit import _counters
 from agent_memory_toolkit import thresholds as default_thresholds
 from agent_memory_toolkit.aio.processors import AsyncInProcessProcessor
 from agent_memory_toolkit.auto_trigger import _threshold_int, _threshold_value
+from agent_memory_toolkit.logging import get_logger
 
 logger = get_logger(__name__)
 
@@ -135,10 +135,25 @@ async def _fire_thread_steps(
         )
     )
     calls = (
-        (fire_extract, "process_extract_memories", processor.process_extract_memories, {"user_id": user_id, "thread_id": thread_id}),
+        (
+            fire_extract,
+            "process_extract_memories",
+            processor.process_extract_memories,
+            {"user_id": user_id, "thread_id": thread_id},
+        ),
         (fire_dedup, "process_reconcile", processor.process_reconcile, {"user_id": user_id}),
-        (fire_procedural, "synthesize_procedural", processor.synthesize_procedural, {"user_id": user_id}),
-        (fire_summary, "process_thread_summary", processor.process_thread_summary, {"user_id": user_id, "thread_id": thread_id}),
+        (
+            fire_procedural,
+            "synthesize_procedural",
+            processor.synthesize_procedural,
+            {"user_id": user_id},
+        ),
+        (
+            fire_summary,
+            "process_thread_summary",
+            processor.process_thread_summary,
+            {"user_id": user_id, "thread_id": thread_id},
+        ),
     )
     for enabled, label, method, kwargs in calls:
         if not enabled:

agent_memory_toolkit/aio/chat.py

@@ -9,7 +9,6 @@
 from __future__ import annotations
 
 import asyncio
-from agent_memory_toolkit.logging import get_logger
 from typing import Any
 
 from agent_memory_toolkit.chat import (
@@ -19,7 +18,8 @@
     resolve_api_version,
     unsupported_param,
 )
-from agent_memory_toolkit.exceptions import ConfigurationError, LLMError
+from agent_memory_toolkit.exceptions import ConfigurationError
+from agent_memory_toolkit.logging import get_logger
 
 logger = get_logger(__name__)
 

agent_memory_toolkit/aio/cosmos_memory_client.py

@@ -6,8 +6,6 @@
 from datetime import datetime
 from typing import TYPE_CHECKING, Any, Optional
 
-from agent_memory_toolkit.logging import get_logger
-
 from agent_memory_toolkit._base import _BaseMemoryClient
 from agent_memory_toolkit._utils import (
     _build_container_kwargs,
@@ -20,13 +18,14 @@
     _resolve_full_text_language,
     _validate_connection,
 )
+from agent_memory_toolkit.aio.auto_trigger import maybe_trigger_steps
 from agent_memory_toolkit.aio.chat import AsyncChatClient
 from agent_memory_toolkit.aio.embeddings import AsyncEmbeddingsClient
 from agent_memory_toolkit.aio.processors import AsyncInProcessProcessor, AsyncMemoryProcessor
-from agent_memory_toolkit.aio.auto_trigger import maybe_trigger_steps
 from agent_memory_toolkit.aio.services.pipeline import AsyncPipelineService
 from agent_memory_toolkit.aio.store import AsyncMemoryStore
 from agent_memory_toolkit.exceptions import CosmosNotConnectedError, CosmosOperationError
+from agent_memory_toolkit.logging import get_logger
 from agent_memory_toolkit.thresholds import DEFAULT_TTL_BY_TYPE
 
 if TYPE_CHECKING:  # pragma: no cover - typing-only import
@@ -174,7 +173,12 @@ async def connect_cosmos(
         self._cosmos_container = container or self._cosmos_container
         if turns_container is not None:
             self._cosmos_turns_container = turns_container
-        _validate_connection(self._cosmos_endpoint, self._cosmos_credential, self._cosmos_database, self._cosmos_container)
+        _validate_connection(
+            self._cosmos_endpoint,
+            self._cosmos_credential,
+            self._cosmos_database,
+            self._cosmos_container,
+        )
         try:
             from azure.cosmos.aio import CosmosClient
 
@@ -234,7 +238,12 @@ async def create_memory_store(
             throughput_mode=self._cosmos_throughput_mode,
             autoscale_max_ru=autoscale_max_ru if autoscale_max_ru is not None else self._cosmos_autoscale_max_ru,
         )
-        _validate_connection(self._cosmos_endpoint, self._cosmos_credential, self._cosmos_database, self._cosmos_container)
+        _validate_connection(
+            self._cosmos_endpoint,
+            self._cosmos_credential,
+            self._cosmos_database,
+            self._cosmos_container,
+        )
         try:
             from azure.cosmos import PartitionKey, ThroughputProperties
             from azure.cosmos.aio import CosmosClient
@@ -266,10 +275,18 @@ async def create_memory_store(
                 )
             )
             await db.create_container_if_not_exists(
-                **_build_container_kwargs(container_id=self._cosmos_counter_container, partition_key=partition_key, offer_throughput=offer)
+                **_build_container_kwargs(
+                    container_id=self._cosmos_counter_container,
+                    partition_key=partition_key,
+                    offer_throughput=offer,
+                )
             )
             await db.create_container_if_not_exists(
-                **_build_container_kwargs(container_id=self._cosmos_lease_container, partition_key=PartitionKey(path="/id"), offer_throughput=offer)
+                **_build_container_kwargs(
+                    container_id=self._cosmos_lease_container,
+                    partition_key=PartitionKey(path="/id"),
+                    offer_throughput=offer,
+                )
             )
             self._cosmos_client = client
             if self._cosmos_turns_container:
@@ -335,7 +352,8 @@ def _require_pipeline(self) -> None:
         if self._pipeline is None:
             if self._pipeline_init_error is not None:
                 raise CosmosNotConnectedError(
-                    f"Processing pipeline failed to initialize ({type(self._pipeline_init_error).__name__}: {self._pipeline_init_error})."
+                    f"Processing pipeline failed to initialize "
+                    f"({type(self._pipeline_init_error).__name__}: {self._pipeline_init_error})."
                 ) from self._pipeline_init_error
             raise CosmosNotConnectedError("Processing pipeline requires Cosmos DB connection.")
 
@@ -381,7 +399,12 @@ def _get_counter_container(self) -> Any:
         except Exception as exc:  # pragma: no cover - defensive
             if not self._warned_counter_unreachable:
                 self._warned_counter_unreachable = True
-                logger.warning("Counter container %s/%s unreachable: %s", self._cosmos_database, self._cosmos_counter_container, exc)
+                logger.warning(
+                    "Counter container %s/%s unreachable: %s",
+                    self._cosmos_database,
+                    self._cosmos_counter_container,
+                    exc,
+                )
             return None
 
     async def _maybe_auto_trigger(self, turn_counts: dict[tuple[str, str], int]) -> None:
@@ -462,7 +485,6 @@ def _containers_for_query(self, memory_types: Optional[list[str]] = None) -> lis
             return [self._turns_container_client]
         return [self._container_client]
 
-
     async def add_cosmos(
         self,
         user_id: str,
@@ -477,7 +499,19 @@ async def add_cosmos(
         embedding: Optional[list[float]] = None,
         embed: Optional[bool] = None,
     ) -> str:
-        return await self._get_store().add(user_id, role, content, memory_type, metadata, thread_id, tags, ttl, salience, embedding, embed)
+        return await self._get_store().add(
+            user_id,
+            role,
+            content,
+            memory_type,
+            metadata,
+            thread_id,
+            tags,
+            ttl,
+            salience,
+            embedding,
+            embed,
+        )
 
     async def push_to_cosmos(self, batch_size: int = 25) -> None:
         """Insert all local memories into Cosmos DB and schedule processing."""
@@ -639,7 +673,13 @@ async def get_procedural_memories(
         min_salience: Optional[float] = None,
         include_superseded: bool = False,
     ) -> list[dict[str, Any]]:
-        return await self._get_store().get_procedural_memories(user_id, priority, category, min_salience, include_superseded)
+        return await self._get_store().get_procedural_memories(
+            user_id,
+            priority,
+            category,
+            min_salience,
+            include_superseded,
+        )
 
     async def search_episodic_memories(
         self,
@@ -725,4 +765,3 @@ async def _summary_exists(self, *, user_id: str, thread_id: str) -> bool:
         except Exception:
             return False
         return bool(results)
-

agent_memory_toolkit/aio/embeddings.py

@@ -8,11 +8,11 @@
 from __future__ import annotations
 
 import asyncio
-from agent_memory_toolkit.chat import resolve_api_version
-from agent_memory_toolkit.logging import get_logger
 from typing import Any
 
+from agent_memory_toolkit.chat import resolve_api_version
 from agent_memory_toolkit.exceptions import ConfigurationError
+from agent_memory_toolkit.logging import get_logger
 
 logger = get_logger(__name__)
 

agent_memory_toolkit/aio/processors/durable.py

@@ -2,9 +2,9 @@
 
 from __future__ import annotations
 
-from agent_memory_toolkit.logging import get_logger
 from typing import Any, Optional
 
+from agent_memory_toolkit.logging import get_logger
 from agent_memory_toolkit.processors.base import (
     ProcessThreadResult,
     UserSummaryResult,

agent_memory_toolkit/aio/processors/inprocess.py

@@ -147,4 +147,3 @@ async def close(self) -> None:
 
 
 __all__ = ["AsyncInProcessProcessor"]
-