diff --git a/llama_stack/apis/telemetry/telemetry.py b/llama_stack/apis/telemetry/telemetry.py
index 8119c3c4d..b982bb9cc 100644
--- a/llama_stack/apis/telemetry/telemetry.py
+++ b/llama_stack/apis/telemetry/telemetry.py
@@ -190,5 +190,5 @@ class Telemetry(Protocol):
     async def export_agent_trace(
         self,
         session_ids: List[str],
-        dataset_id: str = None,
+        dataset_id: str,
     ) -> None: ...
diff --git a/llama_stack/providers/inline/meta_reference/telemetry/console.py b/llama_stack/providers/inline/meta_reference/telemetry/console.py
index 61f75e911..35dcf9561 100644
--- a/llama_stack/providers/inline/meta_reference/telemetry/console.py
+++ b/llama_stack/providers/inline/meta_reference/telemetry/console.py
@@ -63,7 +63,7 @@ class ConsoleTelemetryImpl(Telemetry):
     async def export_agent_trace(
         self,
         session_ids: List[str],
-        dataset_id: str = None,
+        dataset_id: str,
     ) -> None:
         raise NotImplementedError(
             "Console telemetry does not support agent trace export"
diff --git a/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py b/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
index 9a5d1b8c2..3bea3b921 100644
--- a/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
+++ b/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
@@ -223,9 +223,7 @@ class OpenTelemetryAdapter(Telemetry):
 
         return traces
 
-    async def export_agent_trace(
-        self, session_ids: List[str], dataset_id: str = None
-    ) -> None:
+    async def export_agent_trace(self, session_ids: List[str], dataset_id: str) -> None:
         traces = await self.get_agent_trace(session_ids)
         traces_dict = [
             {