diff --git a/crates/ty_server/src/server/api/requests/completion.rs b/crates/ty_server/src/server/api/requests/completion.rs index 540546cf49..8ba6d14281 100644 --- a/crates/ty_server/src/server/api/requests/completion.rs +++ b/crates/ty_server/src/server/api/requests/completion.rs @@ -81,8 +81,10 @@ impl BackgroundDocumentRequestHandler for CompletionRequestHandler { }) }); - let name = comp.name.to_string(); - let import_suffix = comp.module_name.map(|name| format!(" (import {name})")); + let name = comp.insert.as_deref().unwrap_or(&comp.name).to_string(); + let import_suffix = comp + .module_name + .and_then(|name| import_edit.is_some().then(|| format!(" (import {name})"))); let (label, label_details) = if snapshot .resolved_client_capabilities() .supports_completion_item_label_details() diff --git a/crates/ty_server/tests/e2e/completions.rs b/crates/ty_server/tests/e2e/completions.rs index 24efe82ea9..23e074a715 100644 --- a/crates/ty_server/tests/e2e/completions.rs +++ b/crates/ty_server/tests/e2e/completions.rs @@ -110,13 +110,13 @@ TypedDi insta::assert_json_snapshot!(completions, @r#" [ { - "label": "TypedDict (import typing)", + "label": "typing.TypedDict", "kind": 6, "sortText": "0", "insertText": "typing.TypedDict" }, { - "label": "is_typeddict (import typing)", + "label": "typing.is_typeddict", "kind": 3, "sortText": "1", "insertText": "typing.is_typeddict"