From cb41848a2a8704d9a9ee767f7a84267b52e15ca1 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Thu, 16 Jan 2025 18:14:26 -0800 Subject: [PATCH] disable version check optionally --- llama_stack/distribution/server/server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 6a0047f69..bb9ef0361 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -342,7 +342,8 @@ def main(): app = FastAPI(lifespan=lifespan) app.add_middleware(TracingMiddleware) - app.add_middleware(ClientVersionMiddleware) + if not os.environ.get("LLAMA_STACK_DISABLE_VERSION_CHECK"): + app.add_middleware(ClientVersionMiddleware) try: impls = asyncio.run(construct_stack(config))