diff --git a/packages/kilo-gateway/package.json b/packages/kilo-gateway/package.json
index 4dd64c65690..493d3adce4d 100644
--- a/packages/kilo-gateway/package.json
+++ b/packages/kilo-gateway/package.json
@@ -17,6 +17,7 @@
   ],
   "exports": {
     ".": "./src/index.ts",
+    "./autocomplete": "./src/autocomplete.ts",
     "./tui": "./src/tui.ts"
   },
   "files": [
diff --git a/packages/kilo-vscode/src/shared/autocomplete-models.ts b/packages/kilo-vscode/src/shared/autocomplete-models.ts
index bd21484bacf..f1ea5f552e0 100644
--- a/packages/kilo-vscode/src/shared/autocomplete-models.ts
+++ b/packages/kilo-vscode/src/shared/autocomplete-models.ts
@@ -6,4 +6,4 @@ export {
   validAutocompleteProvider,
   type AutocompleteModelDef,
   type AutocompleteProviderID,
-} from "@kilocode/kilo-gateway"
+} from "@kilocode/kilo-gateway/autocomplete"