From a5d7caf21bbbd04df58609ef3b4f9752382233a3 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin@meta.com>
Date: Wed, 9 Oct 2024 22:31:26 -0700
Subject: [PATCH] typedef fix

---
 llama_stack/distribution/datatypes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/llama_stack/distribution/datatypes.py b/llama_stack/distribution/datatypes.py
index 5212e6da1..0044de09e 100644
--- a/llama_stack/distribution/datatypes.py
+++ b/llama_stack/distribution/datatypes.py
@@ -34,8 +34,8 @@ RoutableObject = Union[
 
 RoutableObjectWithProvider = Union[
     ModelDefWithProvider,
-    ShieldDef,
-    MemoryBankDef,
+    ShieldDefWithProvider,
+    MemoryBankDefWithProvider,
 ]
 
 RoutedProtocol = Union[