diff --git a/llama_stack/apis/agents/agents.py b/llama_stack/apis/agents/agents.py
index a06bb1c26..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,
@@ -32,7 +33,6 @@ from llama_stack.apis.inference import (
     ToolResponse,
     ToolResponseMessage,
     UserMessage,
-    ResponseFormat
 )
 from llama_stack.apis.safety import SafetyViolation
 from llama_stack.apis.tools import ToolDef