Fixing builds

Author: Aayush Kataria

agent_memory_toolkit/_counters.py

@@ -73,9 +73,7 @@ def crosses_threshold(old_count: int, new_count: int, n: int) -> bool:
     return old_count // n != new_count // n
 
 
-def _maybe_warn_owner_mismatch(
-    counter_id: str, existing_owner: Optional[str], observer_owner: Optional[str]
-) -> None:
+def _maybe_warn_owner_mismatch(counter_id: str, existing_owner: Optional[str], observer_owner: Optional[str]) -> None:
     """One-shot WARN when the previous writer disagrees with the current one.
 
     Operators run with either ``MEMORY_PROCESSOR_OWNER=inprocess`` or

agent_memory_toolkit/aio/cosmos_memory_client.py

@@ -696,7 +696,6 @@ async def push_to_cosmos(self, batch_size: int = 25) -> None:
             except RuntimeError:  # pragma: no cover - no running loop
                 logger.warning("No running event loop for auto-trigger; skipping")
 
-
     async def get_memories(
         self,
         memory_id: Optional[str] = None,
@@ -1272,9 +1271,7 @@ def _get_counter_container(self) -> Any:
             return None
         try:
             db = self._cosmos_client.get_database_client(self._cosmos_database)
-            self._counter_container_client = db.get_container_client(
-                self._cosmos_counter_container
-            )
+            self._counter_container_client = db.get_container_client(self._cosmos_counter_container)
             return self._counter_container_client
         except Exception as exc:  # pragma: no cover - defensive
             logger.warning(
@@ -1285,9 +1282,7 @@ def _get_counter_container(self) -> Any:
             )
             return None
 
-    async def _maybe_auto_trigger(
-        self, turn_counts: dict[tuple[str, str], int]
-    ) -> None:
+    async def _maybe_auto_trigger(self, turn_counts: dict[tuple[str, str], int]) -> None:
         """Async mirror of ``CosmosMemoryClient._maybe_auto_trigger``.
 
         Scheduled by ``push_to_cosmos`` as a background ``asyncio.Task``
@@ -1329,9 +1324,7 @@ async def _maybe_auto_trigger(
                     "Further skips will be logged at DEBUG level."
                 )
             else:
-                logger.debug(
-                    "Skipping SDK auto-trigger: MEMORY_PROCESSOR_OWNER=durable"
-                )
+                logger.debug("Skipping SDK auto-trigger: MEMORY_PROCESSOR_OWNER=durable")
             return
 
         n_facts = get_fact_extraction_every_n()
@@ -1471,7 +1464,6 @@ async def _run_auto_trigger_steps(
                             f"process_user_summary: {exc!r}",
                         )
 
-
     async def extract_memories(
         self,
         user_id: str,

agent_memory_toolkit/cosmos_memory_client.py

@@ -653,9 +653,7 @@ def _maybe_auto_trigger(self, turn_counts: dict[tuple[str, str], int]) -> None:
                     "Further skips will be logged at DEBUG level."
                 )
             else:
-                logger.debug(
-                    "Skipping SDK auto-trigger: MEMORY_PROCESSOR_OWNER=durable"
-                )
+                logger.debug("Skipping SDK auto-trigger: MEMORY_PROCESSOR_OWNER=durable")
             return
         # When unset (None) or PROCESSOR_OWNER_INPROCESS, proceed.
 

agent_memory_toolkit/pipeline.py

@@ -1012,9 +1012,7 @@ def deduplicate_facts(
                     )
 
                     source_confidences = [
-                        c for f in cluster_facts
-                        if f["id"] in source_ids
-                        and (c := f.get("confidence")) is not None
+                        c for f in cluster_facts if f["id"] in source_ids and (c := f.get("confidence")) is not None
                     ]
                     merged_confidence = max(source_confidences) if source_confidences else None
 

tests/unit/aio/test_auto_trigger.py

@@ -54,9 +54,7 @@ async def fake_upsert(body):
             elapsed = loop.time() - t0
 
             # If push had awaited the slow trigger we'd see >= 0.5s here.
-            assert elapsed < 0.4, (
-                f"push_to_cosmos awaited trigger inline (elapsed={elapsed:.3f}s)"
-            )
+            assert elapsed < 0.4, f"push_to_cosmos awaited trigger inline (elapsed={elapsed:.3f}s)"
 
             # Drain the background task so pytest doesn't warn about a
             # destroyed-but-pending task at teardown.

tests/unit/function_app/test_change_feed.py

@@ -456,10 +456,7 @@ def test_runs_normally_when_owner_durable():
     # Counter was written.
     assert container._state["thread:u1:t1"]["count"] == 2
     # Threshold (2) crossed — orchestrator started.
-    summary_starts = [
-        c for c in starter.start_new.await_args_list
-        if c.args[0] == "ThreadSummaryOrchestrator"
-    ]
+    summary_starts = [c for c in starter.start_new.await_args_list if c.args[0] == "ThreadSummaryOrchestrator"]
     assert len(summary_starts) == 1
 
 
@@ -483,8 +480,5 @@ def test_runs_normally_when_owner_unset(monkeypatch):
     asyncio.run(process_changefeed_batch(docs, starter, counter_container=container))
 
     assert container._state["thread:u1:t1"]["count"] == 2
-    summary_starts = [
-        c for c in starter.start_new.await_args_list
-        if c.args[0] == "ThreadSummaryOrchestrator"
-    ]
+    summary_starts = [c for c in starter.start_new.await_args_list if c.args[0] == "ThreadSummaryOrchestrator"]
     assert len(summary_starts) == 1

tests/unit/test_pipeline_confidence.py

@@ -208,17 +208,17 @@ def test_merge_carries_max_confidence_and_count(self):
         ]
         pipeline._container.query_items.return_value = iter(facts)
 
-        with patch.object(
-            pipeline, "_cluster_by_similarity", return_value=[[0, 1]]
-        ), patch.object(
-            pipeline,
-            "_run_prompty",
-            return_value=(
-                '{"actions":[{"action":"MERGE","source_ids":["f1","f2"],'
-                '"merged_text":"User likes coffee in the morning.","salience":0.7}]}'
+        with (
+            patch.object(pipeline, "_cluster_by_similarity", return_value=[[0, 1]]),
+            patch.object(
+                pipeline,
+                "_run_prompty",
+                return_value=(
+                    '{"actions":[{"action":"MERGE","source_ids":["f1","f2"],'
+                    '"merged_text":"User likes coffee in the morning.","salience":0.7}]}'
+                ),
             ),
-        ), patch.object(
-            pipeline, "_parse_llm_json", side_effect=lambda s: json.loads(s)
+            patch.object(pipeline, "_parse_llm_json", side_effect=lambda s: json.loads(s)),
         ):
             pipeline.deduplicate_facts(user_id="u1")
 
@@ -237,17 +237,16 @@ def test_merge_omits_confidence_when_no_sources_have_it(self):
         ]
         pipeline._container.query_items.return_value = iter(facts)
 
-        with patch.object(
-            pipeline, "_cluster_by_similarity", return_value=[[0, 1]]
-        ), patch.object(
-            pipeline,
-            "_run_prompty",
-            return_value=(
-                '{"actions":[{"action":"MERGE","source_ids":["f1","f2"],'
-                '"merged_text":"merged","salience":0.5}]}'
+        with (
+            patch.object(pipeline, "_cluster_by_similarity", return_value=[[0, 1]]),
+            patch.object(
+                pipeline,
+                "_run_prompty",
+                return_value=(
+                    '{"actions":[{"action":"MERGE","source_ids":["f1","f2"],"merged_text":"merged","salience":0.5}]}'
+                ),
             ),
-        ), patch.object(
-            pipeline, "_parse_llm_json", side_effect=lambda s: json.loads(s)
+            patch.object(pipeline, "_parse_llm_json", side_effect=lambda s: json.loads(s)),
         ):
             pipeline.deduplicate_facts(user_id="u1")