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): { 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) } }