fix tests

Author: rcholic

examples/lang-chain/README.md

@@ -11,4 +11,3 @@ pip install sentienceapi[langchain]
 Examples:
 - `langchain_tools_demo.py`: build a Sentience tool pack for LangChain
 - `langgraph_self_correcting_graph.py`: observe → act → verify → branch (retry) template
-

examples/lang-chain/langchain_tools_demo.py

@@ -16,7 +16,10 @@
 import asyncio
 
 from sentience import AsyncSentienceBrowser
-from sentience.integrations.langchain import SentienceLangChainContext, build_sentience_langchain_tools
+from sentience.integrations.langchain import (
+    SentienceLangChainContext,
+    build_sentience_langchain_tools,
+)
 
 
 async def main() -> None:
@@ -36,4 +39,3 @@ async def main() -> None:
 
 if __name__ == "__main__":
     asyncio.run(main())
-

examples/lang-chain/langgraph_self_correcting_graph.py

@@ -78,4 +78,3 @@ def branch(state: State) -> str:
 
 if __name__ == "__main__":
     asyncio.run(main())
-

examples/langgraph/sentience_self_correcting_graph.py

@@ -86,4 +86,3 @@ def should_continue(state: State) -> str:
 
 if __name__ == "__main__":
     asyncio.run(main())
-

sentience/cloud_tracing.py

@@ -203,7 +203,9 @@ def _finalize_trace_file_for_upload(self) -> bool:
         self._generate_index()
         return True
 
-    def _close_and_upload_background(self, on_progress: Callable[[int, int], None] | None = None) -> None:
+    def _close_and_upload_background(
+        self, on_progress: Callable[[int, int], None] | None = None
+    ) -> None:
         """
         Background worker for non-blocking close.
 

sentience/integrations/langchain/__init__.py

@@ -10,4 +10,3 @@
 from .tools import build_sentience_langchain_tools
 
 __all__ = ["SentienceLangChainContext", "SentienceLangChainCore", "build_sentience_langchain_tools"]
-

sentience/integrations/langchain/context.py

@@ -16,4 +16,3 @@ class SentienceLangChainContext:
 
     browser: AsyncSentienceBrowser
     tracer: Tracer | None = None
-

sentience/integrations/langchain/core.py

@@ -100,7 +100,10 @@ async def _trace(self, tool_name: str, exec_coro, exec_meta: dict[str, Any]):
                 if success is not None:
                     exec_data["success"] = success
 
-                verify_data = {"passed": bool(success) if success is not None else True, "signals": {}}
+                verify_data = {
+                    "passed": bool(success) if success is not None else True,
+                    "signals": {},
+                }
 
                 step_end_data = TraceEventBuilder.build_step_end_event(
                     step_id=step_id,
@@ -123,7 +126,9 @@ async def _trace(self, tool_name: str, exec_coro, exec_meta: dict[str, Any]):
             raise
 
     # ===== Observe =====
-    async def snapshot_state(self, limit: int = 50, include_screenshot: bool = False) -> BrowserState:
+    async def snapshot_state(
+        self, limit: int = 50, include_screenshot: bool = False
+    ) -> BrowserState:
         async def _run():
             opts = SnapshotOptions(limit=limit, screenshot=include_screenshot)
             snap = await snapshot_async(self.ctx.browser, opts)
@@ -153,7 +158,9 @@ async def read_page(
         enhance_markdown: bool = True,
     ) -> ReadResult:
         async def _run():
-            return await read_async(self.ctx.browser, output_format=format, enhance_markdown=enhance_markdown)
+            return await read_async(
+                self.ctx.browser, output_format=format, enhance_markdown=enhance_markdown
+            )
 
         return await self._trace(
             "read_page",
@@ -178,7 +185,9 @@ async def type_text(self, element_id: int, text: str):
         )
 
     async def press_key(self, key: str):
-        return await self._trace("press_key", lambda: press_async(self.ctx.browser, key), {"key": key})
+        return await self._trace(
+            "press_key", lambda: press_async(self.ctx.browser, key), {"key": key}
+        )
 
     async def scroll_to(
         self,
@@ -221,7 +230,14 @@ async def _run():
         return await self._trace(
             "click_rect",
             _run,
-            {"x": x, "y": y, "width": width, "height": height, "button": button, "click_count": click_count},
+            {
+                "x": x,
+                "y": y,
+                "width": width,
+                "height": height,
+                "button": button,
+                "click_count": click_count,
+            },
         )
 
     async def find_text_rect(
@@ -243,7 +259,12 @@ async def _run():
         return await self._trace(
             "find_text_rect",
             _run,
-            {"query": text, "case_sensitive": case_sensitive, "whole_word": whole_word, "max_results": max_results},
+            {
+                "query": text,
+                "case_sensitive": case_sensitive,
+                "whole_word": whole_word,
+                "max_results": max_results,
+            },
         )
 
     # ===== Verify / guard =====
@@ -272,7 +293,9 @@ async def verify_text_present(
         async def _run():
             result = await read_async(self.ctx.browser, output_format=format, enhance_markdown=True)
             if result.status != "success":
-                return AssertionResult(passed=False, reason=f"read failed: {result.error}", details={})
+                return AssertionResult(
+                    passed=False, reason=f"read failed: {result.error}", details={}
+                )
 
             haystack = result.content if case_sensitive else result.content.lower()
             needle = text if case_sensitive else text.lower()
@@ -301,4 +324,3 @@ async def assert_eventually_url_matches(
                 return last
             await asyncio.sleep(poll_s)
         return last or AssertionResult(passed=False, reason="No attempts executed", details={})
-

sentience/integrations/langchain/tools.py

@@ -26,11 +26,15 @@ def build_sentience_langchain_tools(ctx: SentienceLangChainContext) -> list[Any]
     # ---- Schemas ----
     class SnapshotStateArgs(BaseModel):
         limit: int = Field(50, ge=1, le=500, description="Max elements to return (default 50)")
-        include_screenshot: bool = Field(False, description="Include screenshot in snapshot (default false)")
+        include_screenshot: bool = Field(
+            False, description="Include screenshot in snapshot (default false)"
+        )
 
     class ReadPageArgs(BaseModel):
         format: Literal["raw", "text", "markdown"] = Field("text", description="Output format")
-        enhance_markdown: bool = Field(True, description="Enhance markdown conversion (default true)")
+        enhance_markdown: bool = Field(
+            True, description="Enhance markdown conversion (default true)"
+        )
 
     class ClickArgs(BaseModel):
         element_id: int = Field(..., description="Sentience element id from snapshot_state()")
@@ -44,8 +48,12 @@ class PressKeyArgs(BaseModel):
 
     class ScrollToArgs(BaseModel):
         element_id: int = Field(..., description="Sentience element id from snapshot_state()")
-        behavior: Literal["smooth", "instant", "auto"] = Field("smooth", description="Scroll behavior")
-        block: Literal["start", "center", "end", "nearest"] = Field("center", description="Vertical alignment")
+        behavior: Literal["smooth", "instant", "auto"] = Field(
+            "smooth", description="Scroll behavior"
+        )
+        block: Literal["start", "center", "end", "nearest"] = Field(
+            "center", description="Vertical alignment"
+        )
 
     class NavigateArgs(BaseModel):
         url: str = Field(..., description="URL to navigate to")
@@ -79,7 +87,9 @@ class AssertEventuallyUrlMatchesArgs(BaseModel):
 
     # ---- Sync wrappers (explicitly unsupported) ----
     def _sync_unsupported(*args, **kwargs):
-        raise RuntimeError("Sentience LangChain tools are async-only. Use an async LangChain agent/runner.")
+        raise RuntimeError(
+            "Sentience LangChain tools are async-only. Use an async LangChain agent/runner."
+        )
 
     # ---- Tools ----
     return [
@@ -168,4 +178,3 @@ def _sync_unsupported(*args, **kwargs):
             coroutine=lambda **kw: core.assert_eventually_url_matches(**kw),
         ),
     ]
-

tests/unit/test_langchain_integration_core.py

@@ -86,7 +86,9 @@ async def _fake_snapshot_async(browser, options):
             ],
         )
 
-    monkeypatch.setattr("sentience.integrations.langchain.core.snapshot_async", _fake_snapshot_async)
+    monkeypatch.setattr(
+        "sentience.integrations.langchain.core.snapshot_async", _fake_snapshot_async
+    )
 
     ctx = SentienceLangChainContext(browser=_FakeAsyncBrowser())  # type: ignore[arg-type]
     core = SentienceLangChainCore(ctx)
@@ -95,4 +97,3 @@ async def _fake_snapshot_async(browser, options):
     assert state.url == "https://example.com/"
     assert len(state.elements) == 1
     assert state.elements[0].id == 1
-