diff --git a/dist/index.js b/dist/index.js
index c0b351f..8218954 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -59,6 +59,12 @@ function docCFlags(options, useSPMPlugin) {
         args.push('--transform-for-static-hosting');
     if (options.enableInheritedDocs)
         args.push('--enable-inherited-docs');
+    if (options.sourceRepository?.checkoutPath)
+        args.push('--checkout-path', options.sourceRepository.checkoutPath);
+    if (options.sourceRepository?.service) {
+        args.push('--source-service', options.sourceRepository.service.type);
+        args.push('--source-service-base-url', options.sourceRepository.service.baseUrl);
+    }
     if (options.bundleVersion)
         args.push('--bundle-version', options.bundleVersion);
     if (options.hostingBasePath)
@@ -100,6 +106,9 @@ async function main() {
     const packageVersion = core.getInput('package-version');
     const enableIndexBuilding = core.getBooleanInput('enable-index-building', { required: true });
     const enableInheritedDocs = core.getBooleanInput('enable-inherited-docs', { required: true });
+    const checkoutPath = core.getInput('checkout-path');
+    const repoService = core.getInput('repository-service');
+    const repoBaseUrl = core.getInput('repository-base-url');
     const transformForStaticHosting = core.getBooleanInput('transform-for-static-hosting', { required: true });
     const hostingBasePath = core.getInput('hosting-base-path');
     const outputDir = core.getInput('output');
@@ -127,6 +136,13 @@ async function main() {
             enableIndexBuilding: enableIndexBuilding,
             transformForStaticHosting: transformForStaticHosting,
             enableInheritedDocs: enableInheritedDocs,
+            sourceRepository: checkoutPath || repoService || repoBaseUrl ? {
+                checkoutPath: nonEmpty(checkoutPath),
+                service: repoService && repoBaseUrl ? {
+                    type: repoService,
+                    baseUrl: repoBaseUrl,
+                } : null,
+            } : null,
             bundleVersion: nonEmpty(packageVersion),
             hostingBasePath: nonEmpty(hostingBasePath),
             outputPath: mapNonNull(nonEmpty(outputDir), path_1.default.resolve),