fix: resolve mypy type errors

- Add Transform type annotation for context_manager in proxy/server.py
- Import Transform from headroom.transforms
- Fix response.content[0].text access in judge.py with hasattr check
- Add _neo4j_driver type annotation in direct_mem0.py

headroom/evals/memory/judge.py

@@ -139,7 +139,9 @@ def create_anthropic_judge(
             messages=[{"role": "user", "content": prompt}],
         )
 
-        text = response.content[0].text if response.content else ""
+        text = ""
+        if response.content and hasattr(response.content[0], "text"):
+            text = response.content[0].text
         return _parse_judge_response(text)
 
     return judge

headroom/memory/backends/direct_mem0.py

@@ -136,6 +136,7 @@ class DirectMem0Adapter:
         self._mem0_client: Any = None
         self._embedder: Any = None
         self._neo4j_graph: Any = None
+        self._neo4j_driver: Any = None
         self._qdrant_client: Any = None
         self._initialized = False
 

headroom/proxy/server.py

@@ -91,6 +91,7 @@ from headroom.transforms import (
     IntelligentContextManager,
     RollingWindow,
     SmartCrusher,
+    Transform,
     TransformPipeline,
     is_tree_sitter_available,
 )
@@ -867,6 +868,7 @@ class HeadroomProxy:
 
         # Initialize transforms based on routing mode
         # Choose context manager: IntelligentContextManager (smart) or RollingWindow (legacy)
+        context_manager: Transform  # Can be either IntelligentContextManager or RollingWindow
         if config.intelligent_context:
             # Get TOIN instance for learned pattern integration
             toin = get_toin() if config.intelligent_context_scoring else None