@alloy-js/python 0.4.0-dev.3 → 0.4.0-dev.5

package/dist/src/components/SourceFile.js

@@ -1,9 +1,10 @@
 import { memo as _$memo, createIntrinsic as _$createIntrinsic, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { childrenArray, SourceFile as CoreSourceFile, createNamedContext, createScope, isComponentCreator, List, Scope, Show, SourceDirectoryContext, useContext } from "@alloy-js/core";
+import { childrenArray, computed, SourceFile as CoreSourceFile, createNamedContext, createScope, isComponentCreator, List, Scope, Show, SourceDirectoryContext, useContext } from "@alloy-js/core";
 import { join } from "pathe";
 import { PythonModuleScope } from "../symbols/index.js";
-import { ImportStatements } from "./ImportStatement.js";
+import { categorizeImportRecords, ImportStatements } from "./ImportStatement.js";
 import { SimpleCommentBlock } from "./PyDoc.js";
+import { PythonBlock } from "./PythonBlock.js";
 import { Reference } from "./Reference.js";
 
 // Non top-level definitions
@@ -100,6 +101,24 @@ export function SourceFile(props) {
 
   // Check if there's any preamble content (header, doc, imports, etc.)
   const hasPreamble = props.header !== undefined || props.headerComment !== undefined || props.doc !== undefined || props.futureImports !== undefined;
+  const imports = computed(() => {
+    // Quick scan for any type-only imports
+    const hasTypeImports = [...scope.importedModules.values()].some(props => props.symbols && [...props.symbols].some(s => s.local.isTypeOnly));
+
+    // Add TYPE_CHECKING before categorizing so it's naturally included
+    const typeImportSymbol = hasTypeImports ? scope.addTypeImport() : undefined;
+
+    // Single categorize - TYPE_CHECKING is already in scope.importedModules
+    const {
+      valueImports,
+      typeImports
+    } = categorizeImportRecords(scope.importedModules);
+    return {
+      valueImports,
+      typeImports,
+      typeImportSymbol
+    };
+  });
   return _$createComponent(CoreSourceFile, {
     get path() {
       return props.path;
@@ -170,20 +189,40 @@ export function SourceFile(props) {
         }
       }), _$createComponent(Show, {
         get when() {
-          return scope.importedModules.size > 0;
+          return imports.value.valueImports.size > 0;
         },
         get children() {
           return [_$createComponent(ImportStatements, {
             get records() {
-              return scope.importedModules;
+              return imports.value.valueImports;
             }
-          }), _$createComponent(Show, {
-            when: hasChildren,
+          }), _$createIntrinsic("hbr", {})];
+        }
+      }), _$createComponent(Show, {
+        get when() {
+          return imports.value.typeImports.size > 0;
+        },
+        get children() {
+          return [_$createIntrinsic("hbr", {}), _$createComponent(PythonBlock, {
+            get opener() {
+              return `if ${imports.value.typeImportSymbol.name}:`;
+            },
             get children() {
-              return [_$createIntrinsic("hbr", {}), _$createIntrinsic("hbr", {})];
+              return _$createComponent(ImportStatements, {
+                get records() {
+                  return imports.value.typeImports;
+                }
+              });
             }
           })];
         }
+      }), _$createComponent(Show, {
+        get when() {
+          return hasChildren && scope.importedModules.size > 0;
+        },
+        get children() {
+          return _$createIntrinsic("hbr", {});
+        }
       }), _$createComponent(Show, {
         get when() {
           return needsExtraSpacing && (hasPreamble || scope.importedModules.size > 0);

package/dist/src/components/SourceFile.js.map

@@ -1 +1 @@
-{"version":3,"names":["childrenArray","SourceFile","CoreSourceFile","createNamedContext","createScope","isComponentCreator","List","Scope","Show","SourceDirectoryContext","useContext","join","PythonModuleScope","ImportStatements","SimpleCommentBlock","Reference","NON_DEFINITION_NAMES","Set","WRAPPER_COMPONENT_NAMES","firstChildIsDefinition","children","arr","length","first","name","component","has","props","PythonSourceFileContext","useSourceFile","directoryContext","currentDir","path","replace","scope","undefined","sfContext","module","hasChildren","needsExtraSpacing","hasPreamble","header","headerComment","doc","futureImports","_$createComponent","filetype","reference","when","_$createIntrinsic","_$memo","importedModules","size","records","Provider","value","doubleHardline"],"sources":["../../../src/components/SourceFile.tsx"],"sourcesContent":[null],"mappings":";AAAA,SACEA,aAAa,EAEbC,UAAU,IAAIC,cAAc,EAC5BC,kBAAkB,EAClBC,WAAW,EACXC,kBAAkB,EAClBC,IAAI,EACJC,KAAK,EACLC,IAAI,EACJC,sBAAsB,EACtBC,UAAU,QAEL,gBAAgB;AACvB,SAASC,IAAI,QAAQ,OAAO;AAC5B,SAASC,iBAAiB,QAAQ,qBAAqB;AACvD,SAASC,gBAAgB,QAAQ,sBAAsB;AACvD,SAASC,kBAAkB,QAAQ,YAAY;AAC/C,SAASC,SAAS,QAAQ,gBAAgB;;AAE1C;AACA,MAAMC,oBAAoB,GAAG,IAAIC,GAAG,CAAC,CACnC,qBAAqB,EACrB,kBAAkB,EAClB,wBAAwB,EACxB,oBAAoB,EACpB,WAAW,CACZ,CAAC;;AAEF;AACA,MAAMC,uBAAuB,GAAG,IAAID,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC;;AAE1D;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAASE,sBAAsBA,CAACC,QAA8B,EAAW;EACvE,IAAI,CAACA,QAAQ,EAAE,OAAO,KAAK;EAC3B,MAAMC,GAAG,GAAGrB,aAAa,CAAC,MAAMoB,QAAQ,CAAC;EACzC,IAAIC,GAAG,CAACC,MAAM,KAAK,CAAC,EAAE,OAAO,KAAK;EAClC,MAAMC,KAAK,GAAGF,GAAG,CAAC,CAAC,CAAC;;EAEpB;EACA,IAAI,CAAChB,kBAAkB,CAACkB,KAAK,CAAC,EAAE;IAC9B,OAAO,KAAK;EACd;EAEA,MAAMC,IAAI,GAAGD,KAAK,CAACE,SAAS,CAACD,IAAI;EACjC,IAAIR,oBAAoB,CAACU,GAAG,CAACF,IAAI,CAAC,EAAE;IAClC,OAAO,KAAK;EACd;EACA;EACA,IAAIN,uBAAuB,CAACQ,GAAG,CAACF,IAAI,CAAC,IAAID,KAAK,CAACI,KAAK,EAAEP,QAAQ,EAAE;IAC9D,OAAOD,sBAAsB,CAACI,KAAK,CAACI,KAAK,CAACP,QAAoB,CAAC;EACjE;EACA;EACA,OAAO,IAAI;AACb;AAQA,OAAO,MAAMQ,uBAAkE,GAC7EzB,kBAAkB,CAAC,6BAA6B,CAAC;AAEnD,OAAO,SAAS0B,aAAaA,CAAA,EAAG;EAC9B,OAAOnB,UAAU,CAACkB,uBAAuB,CAAC;AAC5C;AA+BA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAAS3B,UAAUA,CAAC0B,KAAsB,EAAE;EACjD,MAAMG,gBAAgB,GAAGpB,UAAU,CAACD,sBAAsB,CAAE;EAC5D,MAAMsB,UAAU,GAAGD,gBAAgB,CAACE,IAAI;EACxC;EACA,MAAMA,IAAY,GAAGrB,IAAI,CAACoB,UAAU,EAAEJ,KAAK,CAACK,IAAI,CAAC,CAC9CC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CACpBA,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;EACtB,MAAMC,KAAK,GAAG9B,WAAW,CAACQ,iBAAiB,EAAEoB,IAAI,EAAEG,SAAS,CAAC;EAC7D,MAAMC,SAAkC,GAAG;IACzCF,KAAK,EAAEA,KAAK;IACZG,MAAM,EAAEL;EACV,CAAC;;EAED;EACA,MAAMM,WAAW,GACfX,KAAK,CAACP,QAAQ,KAAKe,SAAS,IAC5BnC,aAAa,CAAC,MAAM2B,KAAK,CAACP,QAAQ,CAAC,CAACE,MAAM,GAAG,CAAC;;EAEhD;EACA,MAAMiB,iBAAiB,GAAGpB,sBAAsB,CAACQ,KAAK,CAACP,QAAQ,CAAC;;EAEhE;EACA,MAAMoB,WAAW,GACfb,KAAK,CAACc,MAAM,KAAKN,SAAS,IAC1BR,KAAK,CAACe,aAAa,KAAKP,SAAS,IACjCR,KAAK,CAACgB,GAAG,KAAKR,SAAS,IACvBR,KAAK,CAACiB,aAAa,KAAKT,SAAS;EAEnC,OAAAU,iBAAA,CACG3C,cAAc;IAAA,IACb8B,IAAIA,CAAA;MAAA,OAAEL,KAAK,CAACK,IAAI;IAAA;IAChBc,QAAQ;IACRC,SAAS,EAAEhC,SAAS;IAAA,IACpB0B,MAAMA,CAAA;MAAA,OAAEd,KAAK,CAACc,MAAM;IAAA;IAAA,IAAArB,SAAA;MAAA,QAAAyB,iBAAA,CAGnBrC,IAAI;QAAA,IACHwC,IAAIA,CAAA;UAAA,OACFrB,KAAK,CAACc,MAAM,KAAKN,SAAS,IAC1BR,KAAK,CAACe,aAAa,KAAKP,SAAS,KAChCR,KAAK,CAACgB,GAAG,KAAKR,SAAS,IACtBR,KAAK,CAACiB,aAAa,KAAKT,SAAS,IACjCG,WAAW,CAAC;QAAA;QAAA,IAAAlB,SAAA;UAAA,OAAA6B,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAKjBrC,IAAI;QAAA,IAACwC,IAAIA,CAAA;UAAA,OAAErB,KAAK,CAACe,aAAa,KAAKP,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAAyB,iBAAA,CAC1C/B,kBAAkB;YAAA,IAAAM,SAAA;cAAA,OAAEO,KAAK,CAACe,aAAa;YAAA;UAAA,IAAAG,iBAAA,CAEvCrC,IAAI;YAAA,IAACwC,IAAIA,CAAA;cAAA,OAAErB,KAAK,CAACgB,GAAG,KAAKR,SAAS;YAAA;YAAA,IAAAf,SAAA;cAAA,OAAA6B,iBAAA;YAAA;UAAA,IAAAJ,iBAAA,CAIlCrC,IAAI;YAAA,IACHwC,IAAIA,CAAA;cAAA,OACFrB,KAAK,CAACgB,GAAG,KAAKR,SAAS,KACtBR,KAAK,CAACiB,aAAa,KAAKT,SAAS,IAAIG,WAAW,CAAC;YAAA;YAAA,IAAAlB,SAAA;cAAA,QAAA6B,iBAAA,aAAAA,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAOvDrC,IAAI;QAAA,IAACwC,IAAIA,CAAA;UAAA,OAAErB,KAAK,CAACgB,GAAG,KAAKR,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAA8B,MAAA,OAChCvB,KAAK,CAACgB,GAAG,GAAAE,iBAAA,CACTrC,IAAI;YAAA,IAACwC,IAAIA,CAAA;cAAA,OAAErB,KAAK,CAACiB,aAAa,KAAKT,SAAS,IAAIG,WAAW;YAAA;YAAA,IAAAlB,SAAA;cAAA,OAAA6B,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAI7DrC,IAAI;QAAA,IAACwC,IAAIA,CAAA;UAAA,OAAErB,KAAK,CAACiB,aAAa,KAAKT,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAA8B,MAAA,OAC1CvB,KAAK,CAACiB,aAAa,GAAAC,iBAAA,CACnBrC,IAAI;YAACwC,IAAI,EAAEV,WAAW;YAAA,IAAAlB,SAAA;cAAA,QAAA6B,iBAAA,aAAAA,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAKxBrC,IAAI;QAAA,IAACwC,IAAIA,CAAA;UAAA,OAAEd,KAAK,CAACiB,eAAe,CAACC,IAAI,GAAG,CAAC;QAAA;QAAA,IAAAhC,SAAA;UAAA,QAAAyB,iBAAA,CACvChC,gBAAgB;YAAA,IAACwC,OAAOA,CAAA;cAAA,OAAEnB,KAAK,CAACiB,eAAe;YAAA;UAAA,IAAAN,iBAAA,CAC/CrC,IAAI;YAACwC,IAAI,EAAEV,WAAW;YAAA,IAAAlB,SAAA;cAAA,QAAA6B,iBAAA,aAAAA,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAMxBrC,IAAI;QAAA,IACHwC,IAAIA,CAAA;UAAA,OACFT,iBAAiB,KAAKC,WAAW,IAAIN,KAAK,CAACiB,eAAe,CAACC,IAAI,GAAG,CAAC,CAAC;QAAA;QAAA,IAAAhC,SAAA;UAAA,OAAA6B,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAKvEjB,uBAAuB,CAAC0B,QAAQ;QAACC,KAAK,EAAEnB,SAAS;QAAA,IAAAhB,SAAA;UAAA,OAAAyB,iBAAA,CAC/CtC,KAAK;YAACgD,KAAK,EAAErB,KAAK;YAAA,IAAAd,SAAA;cAAA,OAAAyB,iBAAA,CAChBrC,IAAI;gBAACwC,IAAI,EAAEV,WAAW;gBAAA,IAAAlB,SAAA;kBAAA,OAAAyB,iBAAA,CACpBvC,IAAI;oBAACkD,cAAc;oBAAA,IAAApC,SAAA;sBAAA,OAAEO,KAAK,CAACP,QAAQ;oBAAA;kBAAA;gBAAA;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA;EAAA;AAMhD","ignoreList":[]}
+{"version":3,"names":["childrenArray","computed","SourceFile","CoreSourceFile","createNamedContext","createScope","isComponentCreator","List","Scope","Show","SourceDirectoryContext","useContext","join","PythonModuleScope","categorizeImportRecords","ImportStatements","SimpleCommentBlock","PythonBlock","Reference","NON_DEFINITION_NAMES","Set","WRAPPER_COMPONENT_NAMES","firstChildIsDefinition","children","arr","length","first","name","component","has","props","PythonSourceFileContext","useSourceFile","directoryContext","currentDir","path","replace","scope","undefined","sfContext","module","hasChildren","needsExtraSpacing","hasPreamble","header","headerComment","doc","futureImports","imports","hasTypeImports","importedModules","values","some","symbols","s","local","isTypeOnly","typeImportSymbol","addTypeImport","valueImports","typeImports","_$createComponent","filetype","reference","when","_$createIntrinsic","_$memo","value","size","records","opener","Provider","doubleHardline"],"sources":["../../../src/components/SourceFile.tsx"],"sourcesContent":[null],"mappings":";AAAA,SACEA,aAAa,EAEbC,QAAQ,EACRC,UAAU,IAAIC,cAAc,EAC5BC,kBAAkB,EAClBC,WAAW,EACXC,kBAAkB,EAClBC,IAAI,EACJC,KAAK,EACLC,IAAI,EACJC,sBAAsB,EACtBC,UAAU,QAEL,gBAAgB;AACvB,SAASC,IAAI,QAAQ,OAAO;AAC5B,SAASC,iBAAiB,QAAQ,qBAAqB;AACvD,SACEC,uBAAuB,EACvBC,gBAAgB,QACX,sBAAsB;AAC7B,SAASC,kBAAkB,QAAQ,YAAY;AAC/C,SAASC,WAAW,QAAQ,kBAAkB;AAC9C,SAASC,SAAS,QAAQ,gBAAgB;;AAE1C;AACA,MAAMC,oBAAoB,GAAG,IAAIC,GAAG,CAAC,CACnC,qBAAqB,EACrB,kBAAkB,EAClB,wBAAwB,EACxB,oBAAoB,EACpB,WAAW,CACZ,CAAC;;AAEF;AACA,MAAMC,uBAAuB,GAAG,IAAID,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC;;AAE1D;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAASE,sBAAsBA,CAACC,QAA8B,EAAW;EACvE,IAAI,CAACA,QAAQ,EAAE,OAAO,KAAK;EAC3B,MAAMC,GAAG,GAAGxB,aAAa,CAAC,MAAMuB,QAAQ,CAAC;EACzC,IAAIC,GAAG,CAACC,MAAM,KAAK,CAAC,EAAE,OAAO,KAAK;EAClC,MAAMC,KAAK,GAAGF,GAAG,CAAC,CAAC,CAAC;;EAEpB;EACA,IAAI,CAAClB,kBAAkB,CAACoB,KAAK,CAAC,EAAE;IAC9B,OAAO,KAAK;EACd;EAEA,MAAMC,IAAI,GAAGD,KAAK,CAACE,SAAS,CAACD,IAAI;EACjC,IAAIR,oBAAoB,CAACU,GAAG,CAACF,IAAI,CAAC,EAAE;IAClC,OAAO,KAAK;EACd;EACA;EACA,IAAIN,uBAAuB,CAACQ,GAAG,CAACF,IAAI,CAAC,IAAID,KAAK,CAACI,KAAK,EAAEP,QAAQ,EAAE;IAC9D,OAAOD,sBAAsB,CAACI,KAAK,CAACI,KAAK,CAACP,QAAoB,CAAC;EACjE;EACA;EACA,OAAO,IAAI;AACb;AAQA,OAAO,MAAMQ,uBAAkE,GAC7E3B,kBAAkB,CAAC,6BAA6B,CAAC;AAEnD,OAAO,SAAS4B,aAAaA,CAAA,EAAG;EAC9B,OAAOrB,UAAU,CAACoB,uBAAuB,CAAC;AAC5C;AA+BA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAAS7B,UAAUA,CAAC4B,KAAsB,EAAE;EACjD,MAAMG,gBAAgB,GAAGtB,UAAU,CAACD,sBAAsB,CAAE;EAC5D,MAAMwB,UAAU,GAAGD,gBAAgB,CAACE,IAAI;EACxC;EACA,MAAMA,IAAY,GAAGvB,IAAI,CAACsB,UAAU,EAAEJ,KAAK,CAACK,IAAI,CAAC,CAC9CC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CACpBA,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;EACtB,MAAMC,KAAK,GAAGhC,WAAW,CAACQ,iBAAiB,EAAEsB,IAAI,EAAEG,SAAS,CAAC;EAC7D,MAAMC,SAAkC,GAAG;IACzCF,KAAK,EAAEA,KAAK;IACZG,MAAM,EAAEL;EACV,CAAC;;EAED;EACA,MAAMM,WAAW,GACfX,KAAK,CAACP,QAAQ,KAAKe,SAAS,IAC5BtC,aAAa,CAAC,MAAM8B,KAAK,CAACP,QAAQ,CAAC,CAACE,MAAM,GAAG,CAAC;;EAEhD;EACA,MAAMiB,iBAAiB,GAAGpB,sBAAsB,CAACQ,KAAK,CAACP,QAAQ,CAAC;;EAEhE;EACA,MAAMoB,WAAW,GACfb,KAAK,CAACc,MAAM,KAAKN,SAAS,IAC1BR,KAAK,CAACe,aAAa,KAAKP,SAAS,IACjCR,KAAK,CAACgB,GAAG,KAAKR,SAAS,IACvBR,KAAK,CAACiB,aAAa,KAAKT,SAAS;EAEnC,MAAMU,OAAO,GAAG/C,QAAQ,CAAC,MAAM;IAC7B;IACA,MAAMgD,cAAc,GAAG,CAAC,GAAGZ,KAAK,CAACa,eAAe,CAACC,MAAM,CAAC,CAAC,CAAC,CAACC,IAAI,CAC5DtB,KAAK,IACJA,KAAK,CAACuB,OAAO,IAAI,CAAC,GAAGvB,KAAK,CAACuB,OAAO,CAAC,CAACD,IAAI,CAAEE,CAAC,IAAKA,CAAC,CAACC,KAAK,CAACC,UAAU,CACtE,CAAC;;IAED;IACA,MAAMC,gBAAgB,GAAGR,cAAc,GAAGZ,KAAK,CAACqB,aAAa,CAAC,CAAC,GAAGpB,SAAS;;IAE3E;IACA,MAAM;MAAEqB,YAAY;MAAEC;IAAY,CAAC,GAAG9C,uBAAuB,CAC3DuB,KAAK,CAACa,eACR,CAAC;IAED,OAAO;MACLS,YAAY;MACZC,WAAW;MACXH;IACF,CAAC;EACH,CAAC,CAAC;EAEF,OAAAI,iBAAA,CACG1D,cAAc;IAAA,IACbgC,IAAIA,CAAA;MAAA,OAAEL,KAAK,CAACK,IAAI;IAAA;IAChB2B,QAAQ;IACRC,SAAS,EAAE7C,SAAS;IAAA,IACpB0B,MAAMA,CAAA;MAAA,OAAEd,KAAK,CAACc,MAAM;IAAA;IAAA,IAAArB,SAAA;MAAA,QAAAsC,iBAAA,CAGnBpD,IAAI;QAAA,IACHuD,IAAIA,CAAA;UAAA,OACFlC,KAAK,CAACc,MAAM,KAAKN,SAAS,IAC1BR,KAAK,CAACe,aAAa,KAAKP,SAAS,KAChCR,KAAK,CAACgB,GAAG,KAAKR,SAAS,IACtBR,KAAK,CAACiB,aAAa,KAAKT,SAAS,IACjCG,WAAW,CAAC;QAAA;QAAA,IAAAlB,SAAA;UAAA,OAAA0C,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAKjBpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAElC,KAAK,CAACe,aAAa,KAAKP,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAAsC,iBAAA,CAC1C7C,kBAAkB;YAAA,IAAAO,SAAA;cAAA,OAAEO,KAAK,CAACe,aAAa;YAAA;UAAA,IAAAgB,iBAAA,CAEvCpD,IAAI;YAAA,IAACuD,IAAIA,CAAA;cAAA,OAAElC,KAAK,CAACgB,GAAG,KAAKR,SAAS;YAAA;YAAA,IAAAf,SAAA;cAAA,OAAA0C,iBAAA;YAAA;UAAA,IAAAJ,iBAAA,CAIlCpD,IAAI;YAAA,IACHuD,IAAIA,CAAA;cAAA,OACFlC,KAAK,CAACgB,GAAG,KAAKR,SAAS,KACtBR,KAAK,CAACiB,aAAa,KAAKT,SAAS,IAAIG,WAAW,CAAC;YAAA;YAAA,IAAAlB,SAAA;cAAA,QAAA0C,iBAAA,aAAAA,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAOvDpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAElC,KAAK,CAACgB,GAAG,KAAKR,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAA2C,MAAA,OAChCpC,KAAK,CAACgB,GAAG,GAAAe,iBAAA,CACTpD,IAAI;YAAA,IAACuD,IAAIA,CAAA;cAAA,OAAElC,KAAK,CAACiB,aAAa,KAAKT,SAAS,IAAIG,WAAW;YAAA;YAAA,IAAAlB,SAAA;cAAA,OAAA0C,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAI7DpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAElC,KAAK,CAACiB,aAAa,KAAKT,SAAS;QAAA;QAAA,IAAAf,SAAA;UAAA,QAAA2C,MAAA,OAC1CpC,KAAK,CAACiB,aAAa,GAAAc,iBAAA,CACnBpD,IAAI;YAACuD,IAAI,EAAEvB,WAAW;YAAA,IAAAlB,SAAA;cAAA,QAAA0C,iBAAA,aAAAA,iBAAA;YAAA;UAAA;QAAA;MAAA,IAAAJ,iBAAA,CAMxBpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAEhB,OAAO,CAACmB,KAAK,CAACR,YAAY,CAACS,IAAI,GAAG,CAAC;QAAA;QAAA,IAAA7C,SAAA;UAAA,QAAAsC,iBAAA,CAC5C9C,gBAAgB;YAAA,IAACsD,OAAOA,CAAA;cAAA,OAAErB,OAAO,CAACmB,KAAK,CAACR,YAAY;YAAA;UAAA,IAAAM,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAItDpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAEhB,OAAO,CAACmB,KAAK,CAACP,WAAW,CAACQ,IAAI,GAAG,CAAC;QAAA;QAAA,IAAA7C,SAAA;UAAA,QAAA0C,iBAAA,aAAAJ,iBAAA,CAE3C5C,WAAW;YAAA,IAACqD,MAAMA,CAAA;cAAA,OAAE,MAAMtB,OAAO,CAACmB,KAAK,CAACV,gBAAgB,CAAE9B,IAAI,GAAG;YAAA;YAAA,IAAAJ,SAAA;cAAA,OAAAsC,iBAAA,CAC/D9C,gBAAgB;gBAAA,IAACsD,OAAOA,CAAA;kBAAA,OAAErB,OAAO,CAACmB,KAAK,CAACP,WAAW;gBAAA;cAAA;YAAA;UAAA;QAAA;MAAA,IAAAC,iBAAA,CAIvDpD,IAAI;QAAA,IAACuD,IAAIA,CAAA;UAAA,OAAEvB,WAAW,IAAIJ,KAAK,CAACa,eAAe,CAACkB,IAAI,GAAG,CAAC;QAAA;QAAA,IAAA7C,SAAA;UAAA,OAAA0C,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAIxDpD,IAAI;QAAA,IACHuD,IAAIA,CAAA;UAAA,OACFtB,iBAAiB,KAAKC,WAAW,IAAIN,KAAK,CAACa,eAAe,CAACkB,IAAI,GAAG,CAAC,CAAC;QAAA;QAAA,IAAA7C,SAAA;UAAA,OAAA0C,iBAAA;QAAA;MAAA,IAAAJ,iBAAA,CAKvE9B,uBAAuB,CAACwC,QAAQ;QAACJ,KAAK,EAAE5B,SAAS;QAAA,IAAAhB,SAAA;UAAA,OAAAsC,iBAAA,CAC/CrD,KAAK;YAAC2D,KAAK,EAAE9B,KAAK;YAAA,IAAAd,SAAA;cAAA,OAAAsC,iBAAA,CAChBpD,IAAI;gBAACuD,IAAI,EAAEvB,WAAW;gBAAA,IAAAlB,SAAA;kBAAA,OAAAsC,iBAAA,CACpBtD,IAAI;oBAACiE,cAAc;oBAAA,IAAAjD,SAAA;sBAAA,OAAEO,KAAK,CAACP,QAAQ;oBAAA;kBAAA;gBAAA;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA;EAAA;AAMhD","ignoreList":[]}

package/dist/src/components/TypeRefContext.d.ts

@@ -0,0 +1,26 @@
+import type { Children } from "@alloy-js/core";
+import { isTypeRefContext } from "../context/type-ref-context.js";
+export { isTypeRefContext };
+export interface TypeRefContextProps {
+    /**
+     * Children
+     */
+    children: Children;
+}
+/**
+ * Set the current context of reference to be type reference.
+ *
+ * @remarks
+ * References used inside the children of this component will be treated as
+ * type-only references. When a symbol is only referenced in type contexts,
+ * it will be imported inside a `if TYPE_CHECKING:` block.
+ *
+ * @example
+ * ```tsx
+ * <TypeRefContext>
+ *   {someTypeRefkey}
+ * </TypeRefContext>
+ * ```
+ */
+export declare const TypeRefContext: ({ children }: TypeRefContextProps) => Children;
+//# sourceMappingURL=TypeRefContext.d.ts.map

package/dist/src/components/TypeRefContext.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TypeRefContext.d.ts","sourceRoot":"","sources":["../../../src/components/TypeRefContext.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EACL,gBAAgB,EAEjB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,EAAE,gBAAgB,EAAE,CAAC;AAE5B,MAAM,WAAW,mBAAmB;IAClC;;OAEG;IACH,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAED;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,cAAc,GAAI,cAAc,mBAAmB,aAI/D,CAAC"}

package/dist/src/components/TypeRefContext.js

@@ -0,0 +1,29 @@
+import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { isTypeRefContext, TypeRefContextDef } from "../context/type-ref-context.js";
+
+// Re-export for external use
+export { isTypeRefContext };
+/**
+ * Set the current context of reference to be type reference.
+ *
+ * @remarks
+ * References used inside the children of this component will be treated as
+ * type-only references. When a symbol is only referenced in type contexts,
+ * it will be imported inside a `if TYPE_CHECKING:` block.
+ *
+ * @example
+ * ```tsx
+ * <TypeRefContext>
+ *   {someTypeRefkey}
+ * </TypeRefContext>
+ * ```
+ */
+export const TypeRefContext = ({
+  children
+}) => {
+  return _$createComponent(TypeRefContextDef.Provider, {
+    value: true,
+    children: children
+  });
+};
+//# sourceMappingURL=TypeRefContext.js.map

package/dist/src/components/TypeRefContext.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["isTypeRefContext","TypeRefContextDef","TypeRefContext","children","_$createComponent","Provider","value"],"sources":["../../../src/components/TypeRefContext.tsx"],"sourcesContent":[null],"mappings":";AACA,SACEA,gBAAgB,EAChBC,iBAAiB,QACZ,gCAAgC;;AAEvC;AACA,SAASD,gBAAgB;AASzB;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,MAAME,cAAc,GAAGA,CAAC;EAAEC;AAA8B,CAAC,KAAK;EACnE,OAAAC,iBAAA,CACGH,iBAAiB,CAACI,QAAQ;IAACC,KAAK;IAAAH,QAAA,EAAEA;EAAQ;AAE/C,CAAC","ignoreList":[]}

package/dist/src/components/TypeReference.d.ts

@@ -9,6 +9,11 @@ export interface TypeReferenceProps {
 }
 /**
  * A type reference like Foo[T, P] or int.
+ *
+ * @remarks
+ * This component automatically wraps its content in a type reference context,
+ * so any symbols referenced via refkey will be imported as type-only
+ * (inside a `if TYPE_CHECKING:` block) unless also used as values elsewhere.
  */
 export declare function TypeReference(props: TypeReferenceProps): Children;
 //# sourceMappingURL=TypeReference.d.ts.map

package/dist/src/components/TypeReference.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TypeReference.d.ts","sourceRoot":"","sources":["../../../src/components/TypeReference.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAQ,MAAM,gBAAgB,CAAC;AAGxD,MAAM,WAAW,kBAAkB;IACjC,qCAAqC;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,0DAA0D;IAC1D,IAAI,CAAC,EAAE,QAAQ,CAAC;IAChB,gDAAgD;IAChD,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,kBAAkB,YAiBtD"}
+{"version":3,"file":"TypeReference.d.ts","sourceRoot":"","sources":["../../../src/components/TypeReference.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAQ,MAAM,gBAAgB,CAAC;AAIxD,MAAM,WAAW,kBAAkB;IACjC,qCAAqC;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,0DAA0D;IAC1D,IAAI,CAAC,EAAE,QAAQ,CAAC;IAChB,gDAAgD;IAChD,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED;;;;;;;GAOG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,kBAAkB,YAmBtD"}

package/dist/src/components/TypeReference.js

@@ -1,8 +1,14 @@
 import { createComponent as _$createComponent, createIntrinsic as _$createIntrinsic } from "@alloy-js/core/jsx-runtime";
 import { Show } from "@alloy-js/core";
 import { TypeArguments } from "./TypeArguments.js";
+import { TypeRefContext } from "./TypeRefContext.js";
 /**
  * A type reference like Foo[T, P] or int.
+ *
+ * @remarks
+ * This component automatically wraps its content in a type reference context,
+ * so any symbols referenced via refkey will be imported as type-only
+ * (inside a `if TYPE_CHECKING:` block) unless also used as values elsewhere.
  */
 export function TypeReference(props) {
   const type = props.refkey ? props.refkey : props.name;
@@ -11,18 +17,22 @@ export function TypeReference(props) {
       return props.typeArgs;
     }
   }) : undefined;
-  return _$createIntrinsic("group", {
+  return _$createComponent(TypeRefContext, {
     get children() {
-      return [_$createIntrinsic("indent", {
+      return _$createIntrinsic("group", {
         get children() {
-          return [_$createIntrinsic("sbr", {}), type, _$createComponent(Show, {
-            get when() {
-              return Boolean(typeArgs);
-            },
-            children: typeArgs
-          })];
+          return [_$createIntrinsic("indent", {
+            get children() {
+              return [_$createIntrinsic("sbr", {}), type, _$createComponent(Show, {
+                get when() {
+                  return Boolean(typeArgs);
+                },
+                children: typeArgs
+              })];
+            }
+          }), _$createIntrinsic("sbr", {})];
         }
-      }), _$createIntrinsic("sbr", {})];
+      });
     }
   });
 }

package/dist/src/components/TypeReference.js.map

@@ -1 +1 @@
-{"version":3,"names":["Show","TypeArguments","TypeReference","props","type","refkey","name","typeArgs","length","_$createComponent","args","undefined","_$createIntrinsic","children","when","Boolean"],"sources":["../../../src/components/TypeReference.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAA2BA,IAAI,QAAQ,gBAAgB;AACvD,SAASC,aAAa,QAAQ,oBAAoB;AAWlD;AACA;AACA;AACA,OAAO,SAASC,aAAaA,CAACC,KAAyB,EAAE;EACvD,MAAMC,IAAI,GAAGD,KAAK,CAACE,MAAM,GAAGF,KAAK,CAACE,MAAM,GAAGF,KAAK,CAACG,IAAI;EACrD,MAAMC,QAAQ,GACZJ,KAAK,CAACI,QAAQ,IAAIJ,KAAK,CAACI,QAAQ,CAACC,MAAM,GAAAC,iBAAA,CACpCR,aAAa;IAAA,IAACS,IAAIA,CAAA;MAAA,OAAEP,KAAK,CAACI,QAAQ;IAAA;EAAA,KACnCI,SAAS;EAEb,OAAAC,iBAAA;IAAA,IAAAC,SAAA;MAAA,QAAAD,iBAAA;QAAA,IAAAC,SAAA;UAAA,QAAAD,iBAAA,aAIOR,IAAI,EAAAK,iBAAA,CACJT,IAAI;YAAA,IAACc,IAAIA,CAAA;cAAA,OAAEC,OAAO,CAACR,QAAQ,CAAC;YAAA;YAAAM,QAAA,EAAGN;UAAQ;QAAA;MAAA,IAAAK,iBAAA;IAAA;EAAA;AAKhD","ignoreList":[]}
+{"version":3,"names":["Show","TypeArguments","TypeRefContext","TypeReference","props","type","refkey","name","typeArgs","length","_$createComponent","args","undefined","children","_$createIntrinsic","when","Boolean"],"sources":["../../../src/components/TypeReference.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAA2BA,IAAI,QAAQ,gBAAgB;AACvD,SAASC,aAAa,QAAQ,oBAAoB;AAClD,SAASC,cAAc,QAAQ,qBAAqB;AAWpD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,aAAaA,CAACC,KAAyB,EAAE;EACvD,MAAMC,IAAI,GAAGD,KAAK,CAACE,MAAM,GAAGF,KAAK,CAACE,MAAM,GAAGF,KAAK,CAACG,IAAI;EACrD,MAAMC,QAAQ,GACZJ,KAAK,CAACI,QAAQ,IAAIJ,KAAK,CAACI,QAAQ,CAACC,MAAM,GAAAC,iBAAA,CACpCT,aAAa;IAAA,IAACU,IAAIA,CAAA;MAAA,OAAEP,KAAK,CAACI,QAAQ;IAAA;EAAA,KACnCI,SAAS;EAEb,OAAAF,iBAAA,CACGR,cAAc;IAAA,IAAAW,SAAA;MAAA,OAAAC,iBAAA;QAAA,IAAAD,SAAA;UAAA,QAAAC,iBAAA;YAAA,IAAAD,SAAA;cAAA,QAAAC,iBAAA,aAIRT,IAAI,EAAAK,iBAAA,CACJV,IAAI;gBAAA,IAACe,IAAIA,CAAA;kBAAA,OAAEC,OAAO,CAACR,QAAQ,CAAC;gBAAA;gBAAAK,QAAA,EAAGL;cAAQ;YAAA;UAAA,IAAAM,iBAAA;QAAA;MAAA;IAAA;EAAA;AAMlD","ignoreList":[]}

package/dist/src/components/VariableDeclaration.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"VariableDeclaration.d.ts","sourceRoot":"","sources":["../../../src/components/VariableDeclaration.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EAMT,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAGzD,MAAM,WAAW,wBAAyB,SAAQ,oBAAoB;IACpE;;OAEG;IACH,WAAW,CAAC,EAAE,QAAQ,CAAC;IACvB;;OAEG;IACH,IAAI,CAAC,EAAE,QAAQ,CAAC;IAChB;;OAEG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,wBAAwB,YA6ElE"}
+{"version":3,"file":"VariableDeclaration.d.ts","sourceRoot":"","sources":["../../../src/components/VariableDeclaration.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EAMT,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAIzD,MAAM,WAAW,wBAAyB,SAAQ,oBAAoB;IACpE;;OAEG;IACH,WAAW,CAAC,EAAE,QAAQ,CAAC;IACvB;;OAEG;IACH,IAAI,CAAC,EAAE,QAAQ,CAAC;IAChB;;OAEG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,wBAAwB,YAgFlE"}

package/dist/src/components/VariableDeclaration.js

@@ -3,6 +3,7 @@ import { Declaration as CoreDeclaration, Name, Show, createSymbolSlot, memo } fr
 import { createPythonSymbol } from "../symbol-creation.js";
 import { Atom } from "./Atom.js";
 import { SimpleCommentBlock } from "./PyDoc.js";
+import { TypeRefContext } from "./TypeRefContext.js";
 /**
  * A variable declaration component for Python.
  *
@@ -52,9 +53,13 @@ export function VariableDeclaration(props) {
   // Handle optional type annotation
   const type = memo(() => {
     if (!props.type || props.callStatementVar) return undefined;
-    return [": ", _$createComponent(TypeSymbolSlot, {
+    return [":", " ", _$createComponent(TypeRefContext, {
       get children() {
-        return props.type;
+        return _$createComponent(TypeSymbolSlot, {
+          get children() {
+            return props.type;
+          }
+        });
       }
     })];
   });

package/dist/src/components/VariableDeclaration.js.map

@@ -1 +1 @@
-{"version":3,"names":["Declaration","CoreDeclaration","Name","Show","createSymbolSlot","memo","createPythonSymbol","Atom","SimpleCommentBlock","VariableDeclaration","props","TypeSymbolSlot","ValueTypeSymbolSlot","sym","name","instance","instanceVariable","refkeys","refkey","type","firstSymbol","undefined","moveMembersTo","callStatementVar","_$createComponent","children","value","initializer","assignmentOperator","getRightSide","omitNone","renderRightSideOperator","jsValue","rightSide","when","Boolean","doc","_$createIntrinsic","symbol","_$memo"],"sources":["../../../src/components/VariableDeclaration.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAEEA,WAAW,IAAIC,eAAe,EAC9BC,IAAI,EACJC,IAAI,EACJC,gBAAgB,EAChBC,IAAI,QACC,gBAAgB;AACvB,SAASC,kBAAkB,QAAQ,uBAAuB;AAC1D,SAASC,IAAI;AAEb,SAASC,kBAAkB;AA2B3B;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,mBAAmBA,CAACC,KAA+B,EAAE;EACnE,MAAMC,cAAc,GAAGP,gBAAgB,CAAC,CAAC;EACzC,MAAMQ,mBAAmB,GAAGR,gBAAgB,CAAC,CAAC;EAE9C,MAAMS,GAAG,GAAGP,kBAAkB,CAC5BI,KAAK,CAACI,IAAI,EACV;IACEC,QAAQ,EAAEL,KAAK,CAACM,gBAAgB;IAChCC,OAAO,EAAEP,KAAK,CAACQ,MAAM;IACrBC,IAAI,EAAET,KAAK,CAACS,IAAI,GAAGR,cAAc,CAACS,WAAW,GAAGC;EAClD,CAAC,EACD,UACF,CAAC;EAED,IAAI,CAACX,KAAK,CAACS,IAAI,EAAE;IACfP,mBAAmB,CAACU,aAAa,CAACT,GAAG,CAAC;EACxC;;EAEA;EACA,MAAMM,IAAI,GAAGd,IAAI,CAAC,MAAM;IACtB,IAAI,CAACK,KAAK,CAACS,IAAI,IAAIT,KAAK,CAACa,gBAAgB,EAAE,OAAOF,SAAS;IAC3D,cAAAG,iBAAA,CAEOb,cAAc;MAAA,IAAAc,SAAA;QAAA,OAAEf,KAAK,CAACS,IAAI;MAAA;IAAA;EAGnC,CAAC,CAAC;;EAEF;EACA,MAAMO,KAAK,GACT,OAAOhB,KAAK,CAACiB,WAAW,KAAK,QAAQ,GACnCtB,IAAI,CAAC,MAAMK,KAAK,CAACiB,WAAW,CAAC,GAC7BjB,KAAK,CAACiB,WAAW;EACrB,MAAMC,kBAAkB,GAAGlB,KAAK,CAACa,gBAAgB,GAAG,GAAG,GAAG,KAAK;EAC/D,MAAMM,YAAY,GAAGA,CAAA,KAAM;IACzB;IACA,IAAInB,KAAK,CAACoB,QAAQ,IAAIpB,KAAK,CAACiB,WAAW,KAAKN,SAAS,EAAE;MACrD,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;IACpB;;IAEA;IACA,IAAIK,KAAK,KAAK,IAAI,IAAIA,KAAK,KAAKL,SAAS,EAAE;MACzC,OAAO,CAAC,IAAI,WAAY;IAC1B;IAEA,IAAIU,uBAAuB,GAAG,IAAI;IAClC;IACA,IACErB,KAAK,CAACa,gBAAgB,KACrBb,KAAK,CAACI,IAAI,KAAKO,SAAS,IAAIX,KAAK,CAACI,IAAI,KAAK,EAAE,CAAC,EAC/C;MACAiB,uBAAuB,GAAG,KAAK;IACjC;;IAEA;IACA,OAAO,CACLA,uBAAuB,EAAAP,iBAAA,CACtBZ,mBAAmB;MAAA,IAAAa,SAAA;QAAA,OAAAD,iBAAA,CACjBjB,IAAI;UAACyB,OAAO,EAAEN;QAAK;MAAA;IAAA,GAEvB;EACH,CAAC;EACD,MAAM,CAACK,uBAAuB,EAAEE,SAAS,CAAC,GAAGJ,YAAY,CAAC,CAAC;EAC3D,QAAAL,iBAAA,CAEKrB,IAAI;IAAA,IAAC+B,IAAIA,CAAA;MAAA,OAAEC,OAAO,CAACzB,KAAK,CAAC0B,GAAG,CAAC;IAAA;IAAA,IAAAX,SAAA;MAAA,QAAAD,iBAAA,CAC3BhB,kBAAkB;QAAA,IAACiB,QAAQA,CAAA;UAAA,OAAEf,KAAK,CAAC0B,GAAG;QAAA;MAAA,IAAAC,iBAAA;IAAA;EAAA,IAAAb,iBAAA,CAGxCvB,eAAe;IAACqC,MAAM,EAAEzB,GAAG;IAAA,IAAAY,SAAA;MAAA,QAAAc,MAAA,OAAAf,iBAAA,CACxBtB,IAAI,QACLiB,IAAI,EACJY,uBAAuB,IAAIH,kBAAkB,EAC7CK,SAAS;IAAA;EAAA;AAIlB","ignoreList":[]}
+{"version":3,"names":["Declaration","CoreDeclaration","Name","Show","createSymbolSlot","memo","createPythonSymbol","Atom","SimpleCommentBlock","TypeRefContext","VariableDeclaration","props","TypeSymbolSlot","ValueTypeSymbolSlot","sym","name","instance","instanceVariable","refkeys","refkey","type","firstSymbol","undefined","moveMembersTo","callStatementVar","_$createComponent","children","value","initializer","assignmentOperator","getRightSide","omitNone","renderRightSideOperator","jsValue","rightSide","when","Boolean","doc","_$createIntrinsic","symbol","_$memo"],"sources":["../../../src/components/VariableDeclaration.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAEEA,WAAW,IAAIC,eAAe,EAC9BC,IAAI,EACJC,IAAI,EACJC,gBAAgB,EAChBC,IAAI,QACC,gBAAgB;AACvB,SAASC,kBAAkB,QAAQ,uBAAuB;AAC1D,SAASC,IAAI;AAEb,SAASC,kBAAkB;AAC3B,SAASC,cAAc;AA2BvB;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,mBAAmBA,CAACC,KAA+B,EAAE;EACnE,MAAMC,cAAc,GAAGR,gBAAgB,CAAC,CAAC;EACzC,MAAMS,mBAAmB,GAAGT,gBAAgB,CAAC,CAAC;EAE9C,MAAMU,GAAG,GAAGR,kBAAkB,CAC5BK,KAAK,CAACI,IAAI,EACV;IACEC,QAAQ,EAAEL,KAAK,CAACM,gBAAgB;IAChCC,OAAO,EAAEP,KAAK,CAACQ,MAAM;IACrBC,IAAI,EAAET,KAAK,CAACS,IAAI,GAAGR,cAAc,CAACS,WAAW,GAAGC;EAClD,CAAC,EACD,UACF,CAAC;EAED,IAAI,CAACX,KAAK,CAACS,IAAI,EAAE;IACfP,mBAAmB,CAACU,aAAa,CAACT,GAAG,CAAC;EACxC;;EAEA;EACA,MAAMM,IAAI,GAAGf,IAAI,CAAC,MAAM;IACtB,IAAI,CAACM,KAAK,CAACS,IAAI,IAAIT,KAAK,CAACa,gBAAgB,EAAE,OAAOF,SAAS;IAC3D,aAEM,GAAG,EAAAG,iBAAA,CACJhB,cAAc;MAAA,IAAAiB,SAAA;QAAA,OAAAD,iBAAA,CACZb,cAAc;UAAA,IAAAc,SAAA;YAAA,OAAEf,KAAK,CAACS,IAAI;UAAA;QAAA;MAAA;IAAA;EAInC,CAAC,CAAC;;EAEF;EACA,MAAMO,KAAK,GACT,OAAOhB,KAAK,CAACiB,WAAW,KAAK,QAAQ,GACnCvB,IAAI,CAAC,MAAMM,KAAK,CAACiB,WAAW,CAAC,GAC7BjB,KAAK,CAACiB,WAAW;EACrB,MAAMC,kBAAkB,GAAGlB,KAAK,CAACa,gBAAgB,GAAG,GAAG,GAAG,KAAK;EAC/D,MAAMM,YAAY,GAAGA,CAAA,KAAM;IACzB;IACA,IAAInB,KAAK,CAACoB,QAAQ,IAAIpB,KAAK,CAACiB,WAAW,KAAKN,SAAS,EAAE;MACrD,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;IACpB;;IAEA;IACA,IAAIK,KAAK,KAAK,IAAI,IAAIA,KAAK,KAAKL,SAAS,EAAE;MACzC,OAAO,CAAC,IAAI,WAAY;IAC1B;IAEA,IAAIU,uBAAuB,GAAG,IAAI;IAClC;IACA,IACErB,KAAK,CAACa,gBAAgB,KACrBb,KAAK,CAACI,IAAI,KAAKO,SAAS,IAAIX,KAAK,CAACI,IAAI,KAAK,EAAE,CAAC,EAC/C;MACAiB,uBAAuB,GAAG,KAAK;IACjC;;IAEA;IACA,OAAO,CACLA,uBAAuB,EAAAP,iBAAA,CACtBZ,mBAAmB;MAAA,IAAAa,SAAA;QAAA,OAAAD,iBAAA,CACjBlB,IAAI;UAAC0B,OAAO,EAAEN;QAAK;MAAA;IAAA,GAEvB;EACH,CAAC;EACD,MAAM,CAACK,uBAAuB,EAAEE,SAAS,CAAC,GAAGJ,YAAY,CAAC,CAAC;EAC3D,QAAAL,iBAAA,CAEKtB,IAAI;IAAA,IAACgC,IAAIA,CAAA;MAAA,OAAEC,OAAO,CAACzB,KAAK,CAAC0B,GAAG,CAAC;IAAA;IAAA,IAAAX,SAAA;MAAA,QAAAD,iBAAA,CAC3BjB,kBAAkB;QAAA,IAACkB,QAAQA,CAAA;UAAA,OAAEf,KAAK,CAAC0B,GAAG;QAAA;MAAA,IAAAC,iBAAA;IAAA;EAAA,IAAAb,iBAAA,CAGxCxB,eAAe;IAACsC,MAAM,EAAEzB,GAAG;IAAA,IAAAY,SAAA;MAAA,QAAAc,MAAA,OAAAf,iBAAA,CACxBvB,IAAI,QACLkB,IAAI,EACJY,uBAAuB,IAAIH,kBAAkB,EAC7CK,SAAS;IAAA;EAAA;AAIlB","ignoreList":[]}

package/dist/src/components/index.d.ts

@@ -13,7 +13,6 @@ export type { CommonFunctionProps } from "./FunctionBase.js";
 export * from "./FunctionCallExpression.js";
 export * from "./FunctionDeclaration.js";
 export * from "./FutureStatement.js";
-export * from "./ImportStatement.js";
 export * from "./LexicalScope.js";
 export * from "./MemberExpression.js";
 export * from "./MemberScope.jsx";
@@ -27,6 +26,7 @@ export * from "./SourceFile.js";
 export * from "./StatementList.js";
 export * from "./StaticMethodDeclaration.js";
 export * from "./TypeArguments.js";
+export * from "./TypeRefContext.js";
 export * from "./TypeReference.js";
 export * from "./UnionTypeExpression.js";
 export * from "./VariableDeclaration.js";

package/dist/src/components/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/components/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,oBAAoB,CAAC;AACnC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,kBAAkB,CAAC;AACjC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,YAAY,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,cAAc,6BAA6B,CAAC;AAC5C,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,YAAY,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAClE,cAAc,wBAAwB,CAAC;AACvC,cAAc,0BAA0B,CAAC;AACzC,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,oBAAoB,CAAC;AACnC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oBAAoB,CAAC;AACnC,cAAc,oBAAoB,CAAC;AACnC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/components/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,oBAAoB,CAAC;AACnC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,kBAAkB,CAAC;AACjC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,YAAY,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,cAAc,6BAA6B,CAAC;AAC5C,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,YAAY,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAClE,cAAc,wBAAwB,CAAC;AACvC,cAAc,0BAA0B,CAAC;AACzC,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,oBAAoB,CAAC;AACnC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC"}

package/dist/src/components/index.js

@@ -12,7 +12,6 @@ export * from "./EnumMember.js";
 export * from "./FunctionCallExpression.js";
 export * from "./FunctionDeclaration.js";
 export * from "./FutureStatement.js";
-export * from "./ImportStatement.js";
 export * from "./LexicalScope.js";
 export * from "./MemberExpression.js";
 export * from "./MemberScope.js";
@@ -25,6 +24,7 @@ export * from "./SourceFile.js";
 export * from "./StatementList.js";
 export * from "./StaticMethodDeclaration.js";
 export * from "./TypeArguments.js";
+export * from "./TypeRefContext.js";
 export * from "./TypeReference.js";
 export * from "./UnionTypeExpression.js";
 export * from "./VariableDeclaration.js";

package/dist/src/components/index.js.map

@@ -1 +1 @@
-{"version":3,"names":[],"sources":["../../../src/components/index.ts"],"sourcesContent":[null],"mappings":"AAAA,cAAc,WAAW;AACzB,cAAc,oBAAoB;AAClC,cAAc,uBAAuB;AACrC,cAAc,yBAAyB;AACvC,cAAc,6BAA6B;AAC3C,cAAc,6BAA6B;AAC3C,cAAc,2BAA2B;AACzC,cAAc,kBAAkB;AAChC,cAAc,8BAA8B;AAC5C,cAAc,sBAAsB;AACpC,cAAc,iBAAiB;AAE/B,cAAc,6BAA6B;AAC3C,cAAc,0BAA0B;AACxC,cAAc,sBAAsB;AACpC,cAAc,sBAAsB;AACpC,cAAc,mBAAmB;AACjC,cAAc,uBAAuB;AACrC;AAEA,cAAc,wBAAwB;AACtC,cAAc,0BAA0B;AACxC,cAAc,YAAY;AAC1B,cAAc,kBAAkB;AAChC,cAAc,gBAAgB;AAC9B,cAAc,iBAAiB;AAC/B,cAAc,oBAAoB;AAClC,cAAc,8BAA8B;AAC5C,cAAc,oBAAoB;AAClC,cAAc,oBAAoB;AAClC,cAAc,0BAA0B;AACxC,cAAc,0BAA0B;AAAC","ignoreList":[]}
+{"version":3,"names":[],"sources":["../../../src/components/index.ts"],"sourcesContent":[null],"mappings":"AAAA,cAAc,WAAW;AACzB,cAAc,oBAAoB;AAClC,cAAc,uBAAuB;AACrC,cAAc,yBAAyB;AACvC,cAAc,6BAA6B;AAC3C,cAAc,6BAA6B;AAC3C,cAAc,2BAA2B;AACzC,cAAc,kBAAkB;AAChC,cAAc,8BAA8B;AAC5C,cAAc,sBAAsB;AACpC,cAAc,iBAAiB;AAE/B,cAAc,6BAA6B;AAC3C,cAAc,0BAA0B;AACxC,cAAc,sBAAsB;AACpC,cAAc,mBAAmB;AACjC,cAAc,uBAAuB;AACrC;AAEA,cAAc,wBAAwB;AACtC,cAAc,0BAA0B;AACxC,cAAc,YAAY;AAC1B,cAAc,kBAAkB;AAChC,cAAc,gBAAgB;AAC9B,cAAc,iBAAiB;AAC/B,cAAc,oBAAoB;AAClC,cAAc,8BAA8B;AAC5C,cAAc,oBAAoB;AAClC,cAAc,qBAAqB;AACnC,cAAc,oBAAoB;AAClC,cAAc,0BAA0B;AACxC,cAAc,0BAA0B;AAAC","ignoreList":[]}

package/dist/src/context/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./type-ref-context.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/src/context/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/context/index.ts"],"names":[],"mappings":"AAAA,cAAc,uBAAuB,CAAC"}

package/dist/src/context/index.js

@@ -0,0 +1,2 @@
+export * from "./type-ref-context.js";
+//# sourceMappingURL=index.js.map

package/dist/src/context/index.js.map

@@ -0,0 +1 @@
+{"version":3,"names":[],"sources":["../../../src/context/index.ts"],"sourcesContent":[null],"mappings":"AAAA,cAAc,uBAAuB","ignoreList":[]}

package/dist/src/context/type-ref-context.d.ts

@@ -0,0 +1,13 @@
+import { ComponentContext } from "@alloy-js/core";
+/**
+ * Context for tracking whether we are in a type annotation position.
+ * Used to determine if imports should be guarded with TYPE_CHECKING.
+ *
+ * @internal Use the TypeRefContext component instead.
+ */
+export declare const TypeRefContextDef: ComponentContext<true>;
+/**
+ * @returns 'true' if in a type context, 'false' if in a value context.
+ */
+export declare function isTypeRefContext(): boolean;
+//# sourceMappingURL=type-ref-context.d.ts.map

package/dist/src/context/type-ref-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"type-ref-context.d.ts","sourceRoot":"","sources":["../../../src/context/type-ref-context.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAA6B,MAAM,gBAAgB,CAAC;AAE7E;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB,EAAE,gBAAgB,CAAC,IAAI,CAAyB,CAAC;AAE/E;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,OAAO,CAE1C"}

package/dist/src/context/type-ref-context.js

@@ -0,0 +1,17 @@
+import { createContext, useContext } from "@alloy-js/core";
+
+/**
+ * Context for tracking whether we are in a type annotation position.
+ * Used to determine if imports should be guarded with TYPE_CHECKING.
+ *
+ * @internal Use the TypeRefContext component instead.
+ */
+export const TypeRefContextDef = createContext();
+
+/**
+ * @returns 'true' if in a type context, 'false' if in a value context.
+ */
+export function isTypeRefContext() {
+  return useContext(TypeRefContextDef) === true;
+}
+//# sourceMappingURL=type-ref-context.js.map

package/dist/src/context/type-ref-context.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["createContext","useContext","TypeRefContextDef","isTypeRefContext"],"sources":["../../../src/context/type-ref-context.tsx"],"sourcesContent":[null],"mappings":"AAAA,SAA2BA,aAAa,EAAEC,UAAU,QAAQ,gBAAgB;;AAE5E;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,MAAMC,iBAAyC,GAAGF,aAAa,CAAO,CAAC;;AAE9E;AACA;AACA;AACA,OAAO,SAASG,gBAAgBA,CAAA,EAAY;EAC1C,OAAOF,UAAU,CAACC,iBAAiB,CAAC,KAAK,IAAI;AAC/C","ignoreList":[]}

package/dist/src/symbols/python-module-scope.d.ts

@@ -11,11 +11,23 @@ export interface ImportRecordProps {
 }
 export declare class ImportRecords extends Map<PythonModuleScope, ImportRecordProps> {
 }
+export interface AddImportOptions {
+    /**
+     * If true, this import is only used in type annotation contexts.
+     * Such imports will be guarded with `if TYPE_CHECKING:`.
+     */
+    type?: boolean;
+}
 export declare class PythonModuleScope extends PythonLexicalScope {
     #private;
     get importedSymbols(): Map<PythonOutputSymbol, PythonOutputSymbol>;
     get importedModules(): ImportRecords;
-    addImport(targetSymbol: PythonOutputSymbol, targetModule: PythonModuleScope): PythonOutputSymbol;
+    addImport(targetSymbol: PythonOutputSymbol, targetModule: PythonModuleScope, options?: AddImportOptions): PythonOutputSymbol;
+    /**
+     * Add TYPE_CHECKING import from the typing module.
+     * Returns the local symbol for use in the if block opener.
+     */
+    addTypeImport(): PythonOutputSymbol;
     get debugInfo(): Record<string, unknown>;
 }
 //# sourceMappingURL=python-module-scope.d.ts.map

package/dist/src/symbols/python-module-scope.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"python-module-scope.d.ts","sourceRoot":"","sources":["../../../src/symbols/python-module-scope.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAE/D,qBAAa,cAAc;IACzB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,kBAAkB,CAAC;gBAEf,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,kBAAkB;IAKjE,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,kBAAkB;CAGlE;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;CAC9B;AAED,qBAAa,aAAc,SAAQ,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;CAAG;AAE/E,qBAAa,iBAAkB,SAAQ,kBAAkB;;IAGvD,IAAI,eAAe,gDAElB;IAGD,IAAI,eAAe,kBAElB;IAED,SAAS,CAAC,YAAY,EAAE,kBAAkB,EAAE,YAAY,EAAE,iBAAiB;IA4B3E,IAAa,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAMhD;CACF"}
+{"version":3,"file":"python-module-scope.d.ts","sourceRoot":"","sources":["../../../src/symbols/python-module-scope.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AA0B/D,qBAAa,cAAc;IACzB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,kBAAkB,CAAC;gBAEf,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,kBAAkB;IAKjE,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,kBAAkB;CAGlE;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;CAC9B;AAED,qBAAa,aAAc,SAAQ,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;CAAG;AAE/E,MAAM,WAAW,gBAAgB;IAC/B;;;OAGG;IACH,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED,qBAAa,iBAAkB,SAAQ,kBAAkB;;IAGvD,IAAI,eAAe,gDAElB;IAGD,IAAI,eAAe,kBAElB;IAED,SAAS,CACP,YAAY,EAAE,kBAAkB,EAChC,YAAY,EAAE,iBAAiB,EAC/B,OAAO,CAAC,EAAE,gBAAgB;IAqC5B;;;OAGG;IACH,aAAa,IAAI,kBAAkB;IAKnC,IAAa,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAMhD;CACF"}

package/dist/src/symbols/python-module-scope.js

@@ -1,6 +1,26 @@
 import { createSymbol, reactive, shallowReactive } from "@alloy-js/core";
 import { PythonLexicalScope } from "./python-lexical-scope.js";
 import { PythonOutputSymbol } from "./python-output-symbol.js";
+
+// Internal typing module for TYPE_CHECKING imports
+let _typingModuleScope;
+let _typeCheckingSymbol;
+
+/**
+ * Get the internal typing module scope and TYPE_CHECKING symbol.
+ * Used by addTypeImport() to add TYPE_CHECKING imports without
+ * going through the binder's refkey resolution.
+ */
+function getTypingModuleInternal() {
+  if (!_typingModuleScope) {
+    _typingModuleScope = new PythonModuleScope("typing", undefined);
+    _typeCheckingSymbol = new PythonOutputSymbol("TYPE_CHECKING", _typingModuleScope.symbols, {});
+  }
+  return {
+    scope: _typingModuleScope,
+    TYPE_CHECKING: _typeCheckingSymbol
+  };
+}
 export class ImportedSymbol {
   constructor(target, local) {
     this.target = target;
@@ -20,9 +40,13 @@ export class PythonModuleScope extends PythonLexicalScope {
   get importedModules() {
     return this.#importedModules;
   }
-  addImport(targetSymbol, targetModule) {
+  addImport(targetSymbol, targetModule, options) {
     const existing = this.importedSymbols.get(targetSymbol);
     if (existing) {
+      // If existing is type-only but now used as value, upgrade it
+      if (!options?.type && existing.isTypeOnly) {
+        existing.markAsValue();
+      }
       return existing;
     }
     if (!this.importedModules.has(targetModule)) {
@@ -32,7 +56,8 @@ export class PythonModuleScope extends PythonLexicalScope {
     }
     const localSymbol = createSymbol(PythonOutputSymbol, targetSymbol.name, this.symbols, {
       binder: this.binder,
-      aliasTarget: targetSymbol
+      aliasTarget: targetSymbol,
+      typeOnly: options?.type
     });
     this.importedSymbols.set(targetSymbol, localSymbol);
     this.importedModules.get(targetModule).symbols?.add({
@@ -41,6 +66,15 @@ export class PythonModuleScope extends PythonLexicalScope {
     });
     return localSymbol;
   }
+
+  /**
+   * Add TYPE_CHECKING import from the typing module.
+   * Returns the local symbol for use in the if block opener.
+   */
+  addTypeImport() {
+    const typing = getTypingModuleInternal();
+    return this.addImport(typing.TYPE_CHECKING, typing.scope);
+  }
   get debugInfo() {
     return {
       ...super.debugInfo,

package/dist/src/symbols/python-module-scope.js.map

@@ -1 +1 @@
-{"version":3,"names":["createSymbol","reactive","shallowReactive","PythonLexicalScope","PythonOutputSymbol","ImportedSymbol","constructor","target","local","from","ImportRecords","Map","PythonModuleScope","importedSymbols","importedModules","addImport","targetSymbol","targetModule","existing","get","has","set","symbols","Set","localSymbol","name","binder","aliasTarget","add","debugInfo","importedSymbolCount","size","importedModuleCount"],"sources":["../../../src/symbols/python-module-scope.ts"],"sourcesContent":[null],"mappings":"AAAA,SAASA,YAAY,EAAEC,QAAQ,EAAEC,eAAe,QAAQ,gBAAgB;AACxE,SAASC,kBAAkB,QAAQ,2BAA2B;AAC9D,SAASC,kBAAkB,QAAQ,2BAA2B;AAE9D,OAAO,MAAMC,cAAc,CAAC;EAI1BC,WAAWA,CAACC,MAA0B,EAAEC,KAAyB,EAAE;IACjE,IAAI,CAACD,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,KAAK,GAAGA,KAAK;EACpB;EAEA,OAAOC,IAAIA,CAACF,MAA0B,EAAEC,KAAyB,EAAE;IACjE,OAAO,IAAIH,cAAc,CAACE,MAAM,EAAEC,KAAK,CAAC;EAC1C;AACF;AAMA,OAAO,MAAME,aAAa,SAASC,GAAG,CAAuC;AAE7E,OAAO,MAAMC,iBAAiB,SAAST,kBAAkB,CAAC;EACxD,CAACU,eAAe,GACdX,eAAe,CAAC,IAAIS,GAAG,CAAC,CAAC,CAAC;EAC5B,IAAIE,eAAeA,CAAA,EAAG;IACpB,OAAO,IAAI,CAAC,CAACA,eAAe;EAC9B;EAEA,CAACC,eAAe,GAAkBb,QAAQ,CAAC,IAAIU,GAAG,CAAC,CAAC,CAAC;EACrD,IAAIG,eAAeA,CAAA,EAAG;IACpB,OAAO,IAAI,CAAC,CAACA,eAAe;EAC9B;EAEAC,SAASA,CAACC,YAAgC,EAAEC,YAA+B,EAAE;IAC3E,MAAMC,QAAQ,GAAG,IAAI,CAACL,eAAe,CAACM,GAAG,CAACH,YAAY,CAAC;IACvD,IAAIE,QAAQ,EAAE;MACZ,OAAOA,QAAQ;IACjB;IAEA,IAAI,CAAC,IAAI,CAACJ,eAAe,CAACM,GAAG,CAACH,YAAY,CAAC,EAAE;MAC3C,IAAI,CAACH,eAAe,CAACO,GAAG,CAACJ,YAAY,EAAE;QACrCK,OAAO,EAAE,IAAIC,GAAG,CAAiB;MACnC,CAAC,CAAC;IACJ;IAEA,MAAMC,WAAW,GAAGxB,YAAY,CAC9BI,kBAAkB,EAClBY,YAAY,CAACS,IAAI,EACjB,IAAI,CAACH,OAAO,EACZ;MAAEI,MAAM,EAAE,IAAI,CAACA,MAAM;MAAEC,WAAW,EAAEX;IAAa,CACnD,CAAC;IAED,IAAI,CAACH,eAAe,CAACQ,GAAG,CAACL,YAAY,EAAEQ,WAAW,CAAC;IACnD,IAAI,CAACV,eAAe,CAACK,GAAG,CAACF,YAAY,CAAC,CAAEK,OAAO,EAAEM,GAAG,CAAC;MACnDpB,KAAK,EAAEgB,WAAW;MAClBjB,MAAM,EAAES;IACV,CAAC,CAAC;IAEF,OAAOQ,WAAW;EACpB;EAEA,IAAaK,SAASA,CAAA,EAA4B;IAChD,OAAO;MACL,GAAG,KAAK,CAACA,SAAS;MAClBC,mBAAmB,EAAE,IAAI,CAAC,CAACjB,eAAe,CAACkB,IAAI;MAC/CC,mBAAmB,EAAE,IAAI,CAAC,CAAClB,eAAe,CAACiB;IAC7C,CAAC;EACH;AACF","ignoreList":[]}
+{"version":3,"names":["createSymbol","reactive","shallowReactive","PythonLexicalScope","PythonOutputSymbol","_typingModuleScope","_typeCheckingSymbol","getTypingModuleInternal","PythonModuleScope","undefined","symbols","scope","TYPE_CHECKING","ImportedSymbol","constructor","target","local","from","ImportRecords","Map","importedSymbols","importedModules","addImport","targetSymbol","targetModule","options","existing","get","type","isTypeOnly","markAsValue","has","set","Set","localSymbol","name","binder","aliasTarget","typeOnly","add","addTypeImport","typing","debugInfo","importedSymbolCount","size","importedModuleCount"],"sources":["../../../src/symbols/python-module-scope.ts"],"sourcesContent":[null],"mappings":"AAAA,SAASA,YAAY,EAAEC,QAAQ,EAAEC,eAAe,QAAQ,gBAAgB;AACxE,SAASC,kBAAkB,QAAQ,2BAA2B;AAC9D,SAASC,kBAAkB,QAAQ,2BAA2B;;AAE9D;AACA,IAAIC,kBAAiD;AACrD,IAAIC,mBAAmD;;AAEvD;AACA;AACA;AACA;AACA;AACA,SAASC,uBAAuBA,CAAA,EAG9B;EACA,IAAI,CAACF,kBAAkB,EAAE;IACvBA,kBAAkB,GAAG,IAAIG,iBAAiB,CAAC,QAAQ,EAAEC,SAAS,CAAC;IAC/DH,mBAAmB,GAAG,IAAIF,kBAAkB,CAC1C,eAAe,EACfC,kBAAkB,CAACK,OAAO,EAC1B,CAAC,CACH,CAAC;EACH;EACA,OAAO;IAAEC,KAAK,EAAEN,kBAAkB;IAAEO,aAAa,EAAEN;EAAqB,CAAC;AAC3E;AAEA,OAAO,MAAMO,cAAc,CAAC;EAI1BC,WAAWA,CAACC,MAA0B,EAAEC,KAAyB,EAAE;IACjE,IAAI,CAACD,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,KAAK,GAAGA,KAAK;EACpB;EAEA,OAAOC,IAAIA,CAACF,MAA0B,EAAEC,KAAyB,EAAE;IACjE,OAAO,IAAIH,cAAc,CAACE,MAAM,EAAEC,KAAK,CAAC;EAC1C;AACF;AAMA,OAAO,MAAME,aAAa,SAASC,GAAG,CAAuC;AAU7E,OAAO,MAAMX,iBAAiB,SAASL,kBAAkB,CAAC;EACxD,CAACiB,eAAe,GACdlB,eAAe,CAAC,IAAIiB,GAAG,CAAC,CAAC,CAAC;EAC5B,IAAIC,eAAeA,CAAA,EAAG;IACpB,OAAO,IAAI,CAAC,CAACA,eAAe;EAC9B;EAEA,CAACC,eAAe,GAAkBpB,QAAQ,CAAC,IAAIkB,GAAG,CAAC,CAAC,CAAC;EACrD,IAAIE,eAAeA,CAAA,EAAG;IACpB,OAAO,IAAI,CAAC,CAACA,eAAe;EAC9B;EAEAC,SAASA,CACPC,YAAgC,EAChCC,YAA+B,EAC/BC,OAA0B,EAC1B;IACA,MAAMC,QAAQ,GAAG,IAAI,CAACN,eAAe,CAACO,GAAG,CAACJ,YAAY,CAAC;IACvD,IAAIG,QAAQ,EAAE;MACZ;MACA,IAAI,CAACD,OAAO,EAAEG,IAAI,IAAIF,QAAQ,CAACG,UAAU,EAAE;QACzCH,QAAQ,CAACI,WAAW,CAAC,CAAC;MACxB;MACA,OAAOJ,QAAQ;IACjB;IAEA,IAAI,CAAC,IAAI,CAACL,eAAe,CAACU,GAAG,CAACP,YAAY,CAAC,EAAE;MAC3C,IAAI,CAACH,eAAe,CAACW,GAAG,CAACR,YAAY,EAAE;QACrCd,OAAO,EAAE,IAAIuB,GAAG,CAAiB;MACnC,CAAC,CAAC;IACJ;IAEA,MAAMC,WAAW,GAAGlC,YAAY,CAC9BI,kBAAkB,EAClBmB,YAAY,CAACY,IAAI,EACjB,IAAI,CAACzB,OAAO,EACZ;MACE0B,MAAM,EAAE,IAAI,CAACA,MAAM;MACnBC,WAAW,EAAEd,YAAY;MACzBe,QAAQ,EAAEb,OAAO,EAAEG;IACrB,CACF,CAAC;IAED,IAAI,CAACR,eAAe,CAACY,GAAG,CAACT,YAAY,EAAEW,WAAW,CAAC;IACnD,IAAI,CAACb,eAAe,CAACM,GAAG,CAACH,YAAY,CAAC,CAAEd,OAAO,EAAE6B,GAAG,CAAC;MACnDvB,KAAK,EAAEkB,WAAW;MAClBnB,MAAM,EAAEQ;IACV,CAAC,CAAC;IAEF,OAAOW,WAAW;EACpB;;EAEA;AACF;AACA;AACA;EACEM,aAAaA,CAAA,EAAuB;IAClC,MAAMC,MAAM,GAAGlC,uBAAuB,CAAC,CAAC;IACxC,OAAO,IAAI,CAACe,SAAS,CAACmB,MAAM,CAAC7B,aAAa,EAAE6B,MAAM,CAAC9B,KAAK,CAAC;EAC3D;EAEA,IAAa+B,SAASA,CAAA,EAA4B;IAChD,OAAO;MACL,GAAG,KAAK,CAACA,SAAS;MAClBC,mBAAmB,EAAE,IAAI,CAAC,CAACvB,eAAe,CAACwB,IAAI;MAC/CC,mBAAmB,EAAE,IAAI,CAAC,CAACxB,eAAe,CAACuB;IAC7C,CAAC;EACH;AACF","ignoreList":[]}

package/dist/src/symbols/python-output-symbol.d.ts

@@ -1,6 +1,8 @@
 import { Namekey, OutputSpace, OutputSymbol, OutputSymbolOptions } from "@alloy-js/core";
 export interface PythonOutputSymbolOptions extends OutputSymbolOptions {
     module?: string;
+    /** Whether this symbol is only used in type annotation contexts */
+    typeOnly?: boolean;
 }
 export interface CreatePythonSymbolFunctionOptions extends PythonOutputSymbolOptions {
     name: string;
@@ -13,6 +15,15 @@ export declare class PythonOutputSymbol extends OutputSymbol {
     static readonly memberSpaces: readonly ["static", "instance"];
     constructor(name: string | Namekey, spaces: OutputSpace[] | OutputSpace | undefined, options: PythonOutputSymbolOptions);
     get module(): string | undefined;
+    /**
+     * Returns true if this symbol is only used in type annotation contexts.
+     * Such symbols can be imported inside a TYPE_CHECKING block.
+     */
+    get isTypeOnly(): boolean;
+    /**
+     * Mark this symbol as also being used as a value (not just a type).
+     */
+    markAsValue(): void;
     get staticMembers(): import("@alloy-js/core").OutputMemberSpace;
     get instanceMembers(): import("@alloy-js/core").OutputMemberSpace;
     get isStaticMemberSymbol(): boolean;

package/dist/src/symbols/python-output-symbol.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"python-output-symbol.d.ts","sourceRoot":"","sources":["../../../src/symbols/python-output-symbol.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,OAAO,EACP,WAAW,EACX,YAAY,EACZ,mBAAmB,EAEpB,MAAM,gBAAgB,CAAC;AAExB,MAAM,WAAW,yBAA0B,SAAQ,mBAAmB;IACpE,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iCACf,SAAQ,yBAAyB;IACjC,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,YAAY;;IAClD,MAAM,CAAC,QAAQ,CAAC,YAAY,kCAAmC;gBAG7D,IAAI,EAAE,MAAM,GAAG,OAAO,EACtB,MAAM,EAAE,WAAW,EAAE,GAAG,WAAW,GAAG,SAAS,EAC/C,OAAO,EAAE,yBAAyB;IASpC,IAAI,MAAM,uBAET;IAED,IAAI,aAAa,+CAEhB;IAED,IAAI,eAAe,+CAElB;IAED,IAAI,oBAAoB,YAEvB;IAED,IAAI,sBAAsB,YAEzB;IAED,IAAa,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAOhD;IAED,IAAI;CAaL"}
+{"version":3,"file":"python-output-symbol.d.ts","sourceRoot":"","sources":["../../../src/symbols/python-output-symbol.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,OAAO,EACP,WAAW,EACX,YAAY,EACZ,mBAAmB,EAKpB,MAAM,gBAAgB,CAAC;AAExB,MAAM,WAAW,yBAA0B,SAAQ,mBAAmB;IACpE,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,mEAAmE;IACnE,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,iCACf,SAAQ,yBAAyB;IACjC,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,YAAY;;IAClD,MAAM,CAAC,QAAQ,CAAC,YAAY,kCAAmC;gBAG7D,IAAI,EAAE,MAAM,GAAG,OAAO,EACtB,MAAM,EAAE,WAAW,EAAE,GAAG,WAAW,GAAG,SAAS,EAC/C,OAAO,EAAE,yBAAyB;IAUpC,IAAI,MAAM,uBAET;IAID;;;OAGG;IACH,IAAI,UAAU,YAGb;IAED;;OAEG;IACH,WAAW;IASX,IAAI,aAAa,+CAEhB;IAED,IAAI,eAAe,+CAElB;IAED,IAAI,oBAAoB,YAEvB;IAED,IAAI,sBAAsB,YAEzB;IAED,IAAa,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAOhD;IAED,IAAI;CAcL"}

package/dist/src/symbols/python-output-symbol.js

@@ -1,4 +1,4 @@
-import { OutputSymbol, createSymbol } from "@alloy-js/core";
+import { createSymbol, OutputSymbol, track, TrackOpTypes, trigger, TriggerOpTypes } from "@alloy-js/core";
 /**
  * Represents an 'exported' symbol from a .py file. Class, enum, interface etc.
  */
@@ -7,6 +7,7 @@ export class PythonOutputSymbol extends OutputSymbol {
   constructor(name, spaces, options) {
     super(name, spaces, options);
     this.#module = options.module ?? undefined;
+    this.#typeOnly = options.typeOnly ?? false;
   }
 
   // The module in which the symbol is defined
@@ -14,6 +15,28 @@ export class PythonOutputSymbol extends OutputSymbol {
   get module() {
     return this.#module;
   }
+  #typeOnly;
+
+  /**
+   * Returns true if this symbol is only used in type annotation contexts.
+   * Such symbols can be imported inside a TYPE_CHECKING block.
+   */
+  get isTypeOnly() {
+    track(this, TrackOpTypes.GET, "typeOnly");
+    return this.#typeOnly;
+  }
+
+  /**
+   * Mark this symbol as also being used as a value (not just a type).
+   */
+  markAsValue() {
+    if (!this.#typeOnly) {
+      return;
+    }
+    const oldValue = this.#typeOnly;
+    this.#typeOnly = false;
+    trigger(this, TriggerOpTypes.SET, "typeOnly", false, oldValue);
+  }
   get staticMembers() {
     return this.memberSpaceFor("static");
   }
@@ -40,7 +63,8 @@ export class PythonOutputSymbol extends OutputSymbol {
       binder,
       aliasTarget: this.aliasTarget,
       module: this.module,
-      metadata: this.metadata
+      metadata: this.metadata,
+      typeOnly: this.isTypeOnly
     });
     this.initializeCopy(copy);
     return copy;