diff --git a/llama_stack/apis/agents/agents.py b/llama_stack/apis/agents/agents.py
index b6d6e6578..d41abc846 100644
--- a/llama_stack/apis/agents/agents.py
+++ b/llama_stack/apis/agents/agents.py
@@ -25,6 +25,7 @@ from pydantic import BaseModel, ConfigDict, Field
 from llama_stack.apis.common.content_types import ContentDelta, InterleavedContent, URL
 from llama_stack.apis.inference import (
     CompletionMessage,
+    ResponseFormat,
     SamplingParams,
     ToolCall,
     ToolChoice,