From 7b8b0736ad4da57890a8dd4307502ed86d33aae6 Mon Sep 17 00:00:00 2001
From: Ashif4354 <ashif.abhas@gmail.com>
Date: Fri, 6 Mar 2026 11:00:53 +0530
Subject: [PATCH] fix: handle aliased fastapi.FastAPI and fastapi.APIRouter
 imports

---
 src/core/extractors.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/core/extractors.ts b/src/core/extractors.ts
index c306b6d..864ef6b 100644
--- a/src/core/extractors.ts
+++ b/src/core/extractors.ts
@@ -459,7 +459,9 @@ export function collectRecognizedNames(nodesByType: Map<string, Node[]>): {
       if (named.name === "fastapi") {
         fastAPINames.add(`${named.alias}.FastAPI`)
         apiRouterNames.add(`${named.alias}.APIRouter`)
-      }
+      } else if (named.name === "fastapi.FastAPI") fastAPINames.add(named.alias)
+      else if (named.name === "fastapi.APIRouter")
+        apiRouterNames.add(named.alias)
     }
   }