diff --git a/src/llama_stack/models/llama/datatypes.py b/src/llama_stack/models/llama/datatypes.py
index 7cb7aa7bd..b5f0bbf0e 100644
--- a/src/llama_stack/models/llama/datatypes.py
+++ b/src/llama_stack/models/llama/datatypes.py
@@ -31,7 +31,7 @@ class BuiltinTool(Enum):
 
 
 Primitive = str | int | float | bool | None
-RecursiveType = Primitive | list[Primitive] | dict[str, Primitive]
+type RecursiveType = Primitive | list[RecursiveType] | dict[str, RecursiveType]
 
 
 class ToolCall(BaseModel):