diff --git a/application/src/main/java/org/thingsboard/server/service/ai/Langchain4jChatModelConfigurerImpl.java b/application/src/main/java/org/thingsboard/server/service/ai/Langchain4jChatModelConfigurerImpl.java
index 156949fdc5..2a9c792b32 100644
--- a/application/src/main/java/org/thingsboard/server/service/ai/Langchain4jChatModelConfigurerImpl.java
+++ b/application/src/main/java/org/thingsboard/server/service/ai/Langchain4jChatModelConfigurerImpl.java
@@ -87,6 +87,7 @@ class Langchain4jChatModelConfigurerImpl implements Langchain4jChatModelConfigur
                 .modelName(modelConfig.modelId())
                 .temperature(modelConfig.temperature())
                 .topP(modelConfig.topP())
+                .topK(modelConfig.topK())
                 .timeout(toDuration(modelConfig.timeoutSeconds()))
                 .maxRetries(modelConfig.maxRetries())
                 .build();
@@ -149,6 +150,9 @@ class Langchain4jChatModelConfigurerImpl implements Langchain4jChatModelConfigur
         if (modelConfig.topP() != null) {
             generationConfigBuilder.setTopP(modelConfig.topP().floatValue());
         }
+        if (modelConfig.topK() != null) {
+            generationConfigBuilder.setTopK(modelConfig.topK());
+        }
         var generationConfig = generationConfigBuilder.build();
 
         // construct generative model instance
@@ -186,6 +190,7 @@ class Langchain4jChatModelConfigurerImpl implements Langchain4jChatModelConfigur
                 .modelName(modelConfig.modelId())
                 .temperature(modelConfig.temperature())
                 .topP(modelConfig.topP())
+                .topK(modelConfig.topK())
                 .timeout(toDuration(modelConfig.timeoutSeconds()))
                 .maxRetries(modelConfig.maxRetries())
                 .build();
diff --git a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/AnthropicChatModel.java b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/AnthropicChatModel.java
index a15a3f9078..ff218d99af 100644
--- a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/AnthropicChatModel.java
+++ b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/AnthropicChatModel.java
@@ -31,6 +31,7 @@ public record AnthropicChatModel(
             String modelId,
             Double temperature,
             Double topP,
+            Integer topK,
             Integer timeoutSeconds,
             Integer maxRetries
     ) implements AiChatModelConfig<AnthropicChatModel.Config> {}
diff --git a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleAiGeminiChatModel.java b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleAiGeminiChatModel.java
index 9b1d6b9e91..e5d282ee50 100644
--- a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleAiGeminiChatModel.java
+++ b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleAiGeminiChatModel.java
@@ -31,6 +31,7 @@ public record GoogleAiGeminiChatModel(
             String modelId,
             Double temperature,
             Double topP,
+            Integer topK,
             Integer timeoutSeconds,
             Integer maxRetries
     ) implements AiChatModelConfig<GoogleAiGeminiChatModel.Config> {}
diff --git a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleVertexAiGeminiChatModel.java b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleVertexAiGeminiChatModel.java
index 67a852e33c..2f5fac23c2 100644
--- a/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleVertexAiGeminiChatModel.java
+++ b/common/data/src/main/java/org/thingsboard/server/common/data/ai/model/chat/GoogleVertexAiGeminiChatModel.java
@@ -31,6 +31,7 @@ public record GoogleVertexAiGeminiChatModel(
             String modelId,
             Double temperature,
             Double topP,
+            Integer topK,
             Integer timeoutSeconds,
             Integer maxRetries
     ) implements AiChatModelConfig<GoogleVertexAiGeminiChatModel.Config> {}