fixes

.github/workflows/main.yml

@@ -166,7 +166,7 @@ jobs:
         run: uv sync --all-packages --group=docs --all-extras
 
       - name: Check if documentation can be built
-        run: uv run mkdocs --build
+        run: uv run mkdocs build
         id: build
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
@@ -182,6 +182,9 @@ jobs:
       contents: write   # to write to the `gh-pages` branch
 
     steps:
+      - name: Check out
+        uses: actions/checkout@v4
+
       - name: Set up the environment
         uses: ./.github/actions/setup-python-env
 

docs/examples/lazy-graph-rag.ipynb

@@ -774,9 +774,9 @@
    "source": [
     "from typing import Any\n",
     "\n",
+    "from graph_retriever.edges import EdgeSpec, MetadataEdgeFunction\n",
     "from langchain_core.language_models import BaseLanguageModel\n",
     "from langchain_core.runnables import chain\n",
-    "from graph_retriever.edges import EdgeSpec, MetadataEdgeFunction\n",
     "from langchain_graph_retriever import GraphRetriever\n",
     "from langchain_graph_retriever.document_graph import create_graph, group_by_community\n",
     "\n",
@@ -842,10 +842,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from graph_retriever.edges import Id\n",
     "from langchain_core.prompts import PromptTemplate\n",
     "from langchain_core.runnables import RunnablePassthrough\n",
     "from langchain_graph_retriever import GraphRetriever\n",
-    "from graph_retriever.edges import Id\n",
     "\n",
     "RETRIEVER = GraphRetriever(\n",
     "    store=store,\n",

packages/graph-retriever/src/graph_retriever/__init__.py

@@ -1,4 +1,5 @@
-"""Provides retrieval functions combining vector and graph traversal.
+"""
+Provides retrieval functions combining vector and graph traversal.
 
 The main methods are [`traverse`][graph_retriever.traverse] and
 [`atraverse`][graph_retriever.atraverse] which provide synchronous and

packages/graph-retriever/src/graph_retriever/edges/__init__.py

@@ -1,8 +1,10 @@
-"""Specification and implementation of edges functions.
+"""
+Specification and implementation of edges functions.
 
 These are responsible for extracting edges from nodes and expressing them in way
 that the adapters can implement.
 """
+
 from ._base import Edge, EdgeFunction, Edges, IdEdge, MetadataEdge
 from .metadata import EdgeSpec, Id, MetadataEdgeFunction