diff --git a/custom_components/stream_assist/core/__init__.py b/custom_components/stream_assist/core/__init__.py
index 4d7b5ab..a4f30cc 100644
--- a/custom_components/stream_assist/core/__init__.py
+++ b/custom_components/stream_assist/core/__init__.py
@@ -16,6 +16,7 @@
     PipelineRun,
     WakeWordSettings,
 )
+from homeassistant.helpers import chat_session
 from homeassistant.components.camera import Camera
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, Context
@@ -160,7 +161,9 @@ def internal_event_callback(event: PipelineEvent):
         stt_stream=stt_stream,
         intent_input=assist.get("intent_input"),
         tts_input=assist.get("tts_input"),
-        conversation_id=assist.get("conversation_id"),
+        session=chat_session.async_get_chat_session(
+            hass, assist.get("conversation_id")
+        ),
         device_id=assist.get("device_id"),
     )