Skip to content

Commit 7b8b073

Browse files
committed
fix: handle aliased fastapi.FastAPI and fastapi.APIRouter imports
1 parent c4e1fb3 commit 7b8b073

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/core/extractors.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,9 @@ export function collectRecognizedNames(nodesByType: Map<string, Node[]>): {
459459
if (named.name === "fastapi") {
460460
fastAPINames.add(`${named.alias}.FastAPI`)
461461
apiRouterNames.add(`${named.alias}.APIRouter`)
462-
}
462+
} else if (named.name === "fastapi.FastAPI") fastAPINames.add(named.alias)
463+
else if (named.name === "fastapi.APIRouter")
464+
apiRouterNames.add(named.alias)
463465
}
464466
}
465467

0 commit comments

Comments
 (0)