From e605d57fb78285828530b2603d21aaa8593df75d Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Tue, 19 Nov 2024 15:59:47 -0800 Subject: [PATCH] use API version in "remote" stack client --- llama_stack/distribution/client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_stack/distribution/client.py b/llama_stack/distribution/client.py index b36ef94e4..e1243cb7a 100644 --- a/llama_stack/distribution/client.py +++ b/llama_stack/distribution/client.py @@ -15,6 +15,8 @@ import httpx from pydantic import BaseModel, parse_obj_as from termcolor import cprint +from llama_stack.apis.version import LLAMA_STACK_API_VERSION + from llama_stack.providers.datatypes import RemoteProviderConfig _CLIENT_CLASSES = {} @@ -117,7 +119,7 @@ def create_api_client_class(protocol) -> Type: break kwargs[param.name] = args[i] - url = f"{self.base_url}{webmethod.route}" + url = f"{self.base_url}/{LLAMA_STACK_API_VERSION}/{webmethod.route.lstrip('/')}" def convert(value): if isinstance(value, list):