core: Fix some private member accesses

Author: cbornet

libs/core/langchain_core/runnables/graph_ascii.py

@@ -163,16 +163,21 @@ def box(self, x0: int, y0: int, width: int, height: int) -> None:
         self.point(x0 + width, y0 + height, "+")
 
 
+class _EdgeViewer:
+    def __init__(self) -> None:
+        self.pts: list[tuple[float]] = []
+
+    def setpath(self, pts: list[tuple[float]]) -> None:
+        self.pts = pts
+
+
 def _build_sugiyama_layout(
     vertices: Mapping[str, str], edges: Sequence[LangEdge]
 ) -> Any:
     try:
         from grandalf.graphs import Edge, Graph, Vertex  # type: ignore[import-untyped]
         from grandalf.layouts import SugiyamaLayout  # type: ignore[import-untyped]
-        from grandalf.routing import (  # type: ignore[import-untyped]
-            EdgeViewer,
-            route_with_lines,
-        )
+        from grandalf.routing import route_with_lines  # type: ignore[import-untyped]
     except ImportError as exc:
         msg = "Install grandalf to draw graphs: `pip install grandalf`."
         raise ImportError(msg) from exc
@@ -199,7 +204,7 @@ def _build_sugiyama_layout(
     minw = min(v.view.w for v in vertices_list)
 
     for edge in edges_:
-        edge.view = EdgeViewer()
+        edge.view = _EdgeViewer()
 
     sug = SugiyamaLayout(graph.C[0])
     graph = graph.C[0]

libs/core/langchain_core/tracers/langchain.py

@@ -51,7 +51,7 @@ def log_error_once(method: str, exception: Exception) -> None:
 def wait_for_all_tracers() -> None:
     """Wait for all tracers to finish."""
     if rt._CLIENT is not None:
-        rt._CLIENT.flush()
+        get_client().flush()
 
 
 def get_client() -> Client: