Skip to content

Commit 97ae8bc

Browse files
committed
Provide default return hints for typing
1 parent a04e346 commit 97ae8bc

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

src/completionProvider.ts

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import { TypeHintProvider } from "./typeHintProvider";
1414
import { paramHintTrigger, returnHintTrigger, PythonType, getDataTypeContainer } from "./python";
1515
import { TypeHintSettings } from "./settings";
1616
import { WorkspaceSearcher } from "./workspaceSearcher";
17+
import { TypingHintProvider } from "./typingHintProvider";
1718

1819

1920
export abstract class CompletionProvider {
@@ -182,7 +183,12 @@ export class ReturnHintCompletionProvider extends CompletionProvider implements
182183
const items: CompletionItem[] = [];
183184
const line = doc.lineAt(pos);
184185

185-
if (this.shouldProvideItems(line, pos)) {
186+
if (this.shouldProvideItems(line, pos)) {
187+
const provider = new TypingHintProvider(getDataTypeContainer());
188+
await provider.detectTypingImport(doc.getText());
189+
this.pushHintsToItems(provider.getRemainingHints(), items);
190+
this.bottomOfListSortPrefix--;
191+
186192
this.pushHintsToItems(Object.values(PythonType), items);
187193
}
188194
return Promise.resolve(new CompletionList(items, false));

0 commit comments

Comments
 (0)