From 28059365c6fd51ea9e141df9bd6592a77f2c1876 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mateusz=20S=C5=82uszniak?= <mateusz.sluszniak@swmansion.com>
Date: Fri, 16 Jan 2026 18:27:16 +0100
Subject: [PATCH] Change configuration in LLMModule so it can be set before
 loading model

---
 .../natural_language_processing/LLMModule.ts    | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/packages/react-native-executorch/src/modules/natural_language_processing/LLMModule.ts b/packages/react-native-executorch/src/modules/natural_language_processing/LLMModule.ts
index d4b46f4cd..d1bd59a5b 100644
--- a/packages/react-native-executorch/src/modules/natural_language_processing/LLMModule.ts
+++ b/packages/react-native-executorch/src/modules/natural_language_processing/LLMModule.ts
@@ -10,6 +10,11 @@ import {
 
 export class LLMModule {
   private controller: LLMController;
+  private pendingConfig?: {
+    chatConfig?: Partial<ChatConfig>;
+    toolsConfig?: ToolsConfig;
+    generationConfig?: GenerationConfig;
+  };
 
   constructor({
     tokenCallback,
@@ -39,6 +44,11 @@ export class LLMModule {
       ...model,
       onDownloadProgressCallback,
     });
+
+    if (this.pendingConfig) {
+      this.controller.configure(this.pendingConfig);
+      this.pendingConfig = undefined;
+    }
   }
 
   setTokenCallback({
@@ -58,7 +68,12 @@ export class LLMModule {
     toolsConfig?: ToolsConfig;
     generationConfig?: GenerationConfig;
   }) {
-    this.controller.configure({ chatConfig, toolsConfig, generationConfig });
+    const config = { chatConfig, toolsConfig, generationConfig };
+    if (this.controller.isReady) {
+      this.controller.configure(config);
+    } else {
+      this.pendingConfig = config;
+    }
   }
 
   async forward(input: string): Promise<string> {