Try add "python.analysis.includeAliasesFromUserFiles": true on your settings.json, as per https://code.visualstudio.com/docs/python/settings-reference#:~:text=includeAliasesFromUserFiles it will:

"include alias symbols from user files in auto-import suggestions and in the add import Quick Fix"