fixing ruff issue

Author: jcodella

agent_memory_toolkit/_utils.py

@@ -78,8 +78,7 @@ def _resolve_cosmos_throughput_mode(val: Optional[str]) -> str:
     if raw not in {"serverless", "autoscale"}:
         raise ConfigurationError(
             message=(
-                "Invalid configuration for cosmos_throughput_mode: "
-                f"expected 'serverless' or 'autoscale', got '{raw}'"
+                f"Invalid configuration for cosmos_throughput_mode: expected 'serverless' or 'autoscale', got '{raw}'"
             ),
             parameter="cosmos_throughput_mode",
         )
@@ -95,18 +94,12 @@ def _resolve_cosmos_autoscale_max_ru(val: Optional[int]) -> int:
         parsed = int(raw)
     except ValueError as exc:
         raise ConfigurationError(
-            message=(
-                "Invalid configuration for cosmos_autoscale_max_ru: "
-                f"expected an integer, got '{raw}'"
-            ),
+            message=(f"Invalid configuration for cosmos_autoscale_max_ru: expected an integer, got '{raw}'"),
             parameter="cosmos_autoscale_max_ru",
         ) from exc
     if parsed <= 0:
         raise ConfigurationError(
-            message=(
-                "Invalid configuration for cosmos_autoscale_max_ru: "
-                f"expected a positive integer, got '{raw}'"
-            ),
+            message=(f"Invalid configuration for cosmos_autoscale_max_ru: expected a positive integer, got '{raw}'"),
             parameter="cosmos_autoscale_max_ru",
         )
     return parsed

agent_memory_toolkit/aio/cosmos_memory_client.py

@@ -20,8 +20,8 @@
     VALID_ROLES,
     VALID_TYPES,
     _build_memory_query_builder,
-    _cosmos_container_offer_throughput,
     _container_policies,
+    _cosmos_container_offer_throughput,
     _make_memory,
     _resolve_cosmos_autoscale_max_ru,
     _resolve_cosmos_throughput_mode,
@@ -371,9 +371,7 @@ async def create_memory_store(
         self._cosmos_container = container or self._cosmos_container
         self._cosmos_counter_container = counter_container or self._cosmos_counter_container
         self._cosmos_lease_container = lease_container or self._cosmos_lease_container
-        self._cosmos_throughput_mode = _resolve_cosmos_throughput_mode(
-            throughput_mode or self._cosmos_throughput_mode
-        )
+        self._cosmos_throughput_mode = _resolve_cosmos_throughput_mode(throughput_mode or self._cosmos_throughput_mode)
         self._cosmos_autoscale_max_ru = _resolve_cosmos_autoscale_max_ru(
             autoscale_max_ru if autoscale_max_ru is not None else self._cosmos_autoscale_max_ru
         )

agent_memory_toolkit/cosmos_memory_client.py

@@ -16,8 +16,8 @@
     VALID_ROLES,
     VALID_TYPES,
     _build_memory_query_builder,
-    _cosmos_container_offer_throughput,
     _container_policies,
+    _cosmos_container_offer_throughput,
     _make_memory,
     _resolve_cosmos_autoscale_max_ru,
     _resolve_cosmos_throughput_mode,
@@ -100,9 +100,7 @@ def __init__(
         self._cosmos_credential = cosmos_credential
         self._cosmos_database = cosmos_database or "ai_memory"
         self._cosmos_container = cosmos_container or "memories"
-        self._cosmos_counter_container = (
-            cosmos_counter_container or "counter"
-        )
+        self._cosmos_counter_container = cosmos_counter_container or "counter"
         self._cosmos_lease_container = cosmos_lease_container or "leases"
         self._cosmos_throughput_mode = _resolve_cosmos_throughput_mode(cosmos_throughput_mode)
         self._cosmos_autoscale_max_ru = _resolve_cosmos_autoscale_max_ru(cosmos_autoscale_max_ru)
@@ -383,9 +381,7 @@ def create_memory_store(
         self._cosmos_container = container or self._cosmos_container
         self._cosmos_counter_container = counter_container or self._cosmos_counter_container
         self._cosmos_lease_container = lease_container or self._cosmos_lease_container
-        self._cosmos_throughput_mode = _resolve_cosmos_throughput_mode(
-            throughput_mode or self._cosmos_throughput_mode
-        )
+        self._cosmos_throughput_mode = _resolve_cosmos_throughput_mode(throughput_mode or self._cosmos_throughput_mode)
         self._cosmos_autoscale_max_ru = _resolve_cosmos_autoscale_max_ru(
             autoscale_max_ru if autoscale_max_ru is not None else self._cosmos_autoscale_max_ru
         )