From 804eab302015720717975e6d02643be6eef58993 Mon Sep 17 00:00:00 2001
From: phnix <null.phnix@gmail.com>
Date: Sat, 9 May 2026 22:13:57 -0400
Subject: [PATCH] fix: add missing CrawlGraph import in mapper.py

---
 huginn/mapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/huginn/mapper.py b/huginn/mapper.py
index f0882b3..f2ee7d8 100644
--- a/huginn/mapper.py
+++ b/huginn/mapper.py
@@ -228,7 +228,7 @@ async def map_site_graph(
         Returns a CrawlGraph with nodes and edges. Faster than full crawling
         because only links are extracted, not content.
         """
-        from .models import PageNode, PageEdge
+        from .models import CrawlGraph, PageNode, PageEdge
 
         parsed = urlparse(start_url)
         base_domain = parsed.netloc