diff --git a/packages/ai/src/themes/PromptInput.css b/packages/ai/src/themes/PromptInput.css
index 1d59e70ff61f..23cdbd35e4f0 100644
--- a/packages/ai/src/themes/PromptInput.css
+++ b/packages/ai/src/themes/PromptInput.css
@@ -1,3 +1,7 @@
+:host(:not([hidden])) {
+    display: inline-block;
+}
+
 .ai-prompt-input-button {
 	margin-left: .5rem;
 	margin-top: 3px;
@@ -6,7 +10,7 @@
 .ai-prompt-input-wrapper {
 	display: flex;
 	flex-direction: column;
-	width: 270px;
+	min-width: 270px;
 }
 
 .ai-prompt-input-form-wrapper {
diff --git a/packages/ai/test/pages/PromptInput.html b/packages/ai/test/pages/PromptInput.html
index 8d0623d62f6e..d3d2e8ab76a6 100644
--- a/packages/ai/test/pages/PromptInput.html
+++ b/packages/ai/test/pages/PromptInput.html
@@ -11,7 +11,6 @@
 
 	<style>
 		ui5-ai-prompt-input {
-			display: inline-block;
 			margin-top: 1rem;
 		}