From 227b69e6e675824ef966aa804622680006148a07 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Wed, 2 Oct 2024 10:13:09 -0700
Subject: [PATCH] Fix sample memory impl

---
 llama_stack/providers/adapters/memory/sample/sample.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/llama_stack/providers/adapters/memory/sample/sample.py b/llama_stack/providers/adapters/memory/sample/sample.py
index d083bc28e..7ef4a625d 100644
--- a/llama_stack/providers/adapters/memory/sample/sample.py
+++ b/llama_stack/providers/adapters/memory/sample/sample.py
@@ -9,10 +9,17 @@ from .config import SampleConfig
 
 from llama_stack.apis.memory import *  # noqa: F403
 
+from llama_stack.distribution.datatypes import RoutableProvider
 
-class SampleMemoryImpl(Memory):
+
+class SampleMemoryImpl(Memory, RoutableProvider):
     def __init__(self, config: SampleConfig):
         self.config = config
 
+    async def validate_routing_keys(self, routing_keys: list[str]) -> None:
+        # these are the memory banks the Llama Stack will use to route requests to this provider
+        # perform validation here if necessary
+        pass
+
     async def initialize(self):
         pass