diff --git a/src/base/deployBase.ts b/src/base/deployBase.ts
index da46608..66ff67f 100644
--- a/src/base/deployBase.ts
+++ b/src/base/deployBase.ts
@@ -30,6 +30,7 @@ export type BaseDeployFlags = {
   'target-org': Org;
   'cpu-size': string;
   network?: string;
+  'use-in-feature'?: string;
 };
 
 export type DeployResult = SharedResultProps & {
@@ -129,6 +130,7 @@ export abstract class DeployBase<TFlags extends BaseDeployFlags = BaseDeployFlag
     const targetOrg = flags['target-org'];
     const cpuSize = flags['cpu-size'] || 'CPU_2XL';
     const network = flags.network;
+    const useInFeature = flags['use-in-feature'];
 
     if (packageDir.length === 0) {
       throw new SfError(messages.getMessage('error.flagEmpty', ['package-dir']), 'InvalidFlagValue');
@@ -158,7 +160,8 @@ export abstract class DeployBase<TFlags extends BaseDeployFlags = BaseDeployFlag
         packageDir,
         orgUsername,
         cpuSize,
-        network
+        network,
+        useInFeature
       );
 
       this.log(cmdMessages.getMessage('info.deploymentComplete', [name, version]));
diff --git a/src/commands/data-code-extension/function/deploy.ts b/src/commands/data-code-extension/function/deploy.ts
index eac365a..c93261d 100644
--- a/src/commands/data-code-extension/function/deploy.ts
+++ b/src/commands/data-code-extension/function/deploy.ts
@@ -14,6 +14,7 @@
  * limitations under the License.
  */
 import { Messages } from '@salesforce/core';
+import { Flags } from '@salesforce/sf-plugins-core';
 import { DeployBase } from '../../../base/deployBase.js';
 
 Messages.importMessagesDirectoryFromMetaUrl(import.meta.url);
@@ -28,6 +29,10 @@ export default class Deploy extends DeployBase {
 
   public static readonly flags = {
     ...DeployBase.flags,
+    'use-in-feature': Flags.string({
+      summary: 'Feature where this function will be used.',
+      default: 'SearchIndexChunking',
+    }),
   };
 
   // eslint-disable-next-line class-methods-use-this
diff --git a/src/utils/datacodeBinaryExecutor.ts b/src/utils/datacodeBinaryExecutor.ts
index 154d949..01468a3 100644
--- a/src/utils/datacodeBinaryExecutor.ts
+++ b/src/utils/datacodeBinaryExecutor.ts
@@ -105,7 +105,8 @@ export class DatacodeBinaryExecutor {
     packageDir: string,
     targetOrg: string,
     cpuSize: string,
-    network?: string
+    network?: string,
+    useInFeature?: string
   ): Promise<DatacodeDeployExecutionResult> {
     // Build args array for spawn (avoids shell-escaping issues and enables streaming)
     const args = [
@@ -128,6 +129,10 @@ export class DatacodeBinaryExecutor {
       args.push('--network', network);
     }
 
+    if (useInFeature) {
+      args.push('--use-in-feature', useInFeature);
+    }
+
     return new Promise((resolve, reject) => {
       debug('deploy spawn: datacustomcode %o', args);
       const child = spawn('datacustomcode', args, {
diff --git a/test/commands/data-code-extension/deploy.test.ts b/test/commands/data-code-extension/deploy.test.ts
index 2c270a4..bb4b5e3 100644
--- a/test/commands/data-code-extension/deploy.test.ts
+++ b/test/commands/data-code-extension/deploy.test.ts
@@ -232,6 +232,59 @@ describe('data-code-extension deploy', () => {
       expect(binaryDeployStub.firstCall.args[0]).to.equal('test-function');
     });
 
+    it('should pass default use-in-feature to binary', async () => {
+      await FunctionDeploy.run([
+        '--name',
+        'test-function',
+        '--package-version',
+        '1.0.0',
+        '--description',
+        'Test function deployment',
+        '--package-dir',
+        testDir,
+        '--target-org',
+        'test@example.com',
+      ]);
+
+      expect(binaryDeployStub.firstCall.args[7]).to.equal('SearchIndexChunking');
+    });
+
+    it('should pass explicit use-in-feature to binary', async () => {
+      await FunctionDeploy.run([
+        '--name',
+        'test-function',
+        '--package-version',
+        '1.0.0',
+        '--description',
+        'Test function deployment',
+        '--package-dir',
+        testDir,
+        '--target-org',
+        'test@example.com',
+        '--use-in-feature',
+        'CustomFeature',
+      ]);
+
+      expect(binaryDeployStub.firstCall.args[7]).to.equal('CustomFeature');
+    });
+
+    it('should not pass use-in-feature for script deploy', async () => {
+      await ScriptDeploy.run([
+        '--name',
+        'test-script',
+        '--package-version',
+        '1.0.0',
+        '--description',
+        'Test script deployment',
+        '--package-dir',
+        testDir,
+        '--target-org',
+        'test@example.com',
+      ]);
+
+      expect(binaryDeployStub.firstCall.args[7]).to.be.undefined;
+    });
+
     it('should validate CPU size options', async () => {
       try {
         await FunctionDeploy.run([