phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove unecessary method argument

Browse source
This commit is contained in:
Celina Hanouti 2024-09-09 19:04:21 +02:00
parent fff1b6d6bf
commit 2ac8e7b901
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_toolchain/inference/adapters/tgi/tgi.py
View file

@ -235,9 +235,10 @@ class TGIAdapter(Inference):
class InferenceEndpointAdapter(TGIAdapter): class InferenceEndpointAdapter(TGIAdapter):
def __init__(self, config: TGIImplConfig) -> None: def __init__(self, config: TGIImplConfig) -> None:
super().__init__(config) super().__init__(config)
self.config.url = self._construct_endpoint_url(config.hf_endpoint_name) self.config.url = self._construct_endpoint_url()
def _construct_endpoint_url(self, hf_endpoint_name: str) -> str: def _construct_endpoint_url(self) -> str:
hf_endpoint_name = self.config.hf_endpoint_name
assert hf_endpoint_name.count("/") <= 1, ( assert hf_endpoint_name.count("/") <= 1, (
"Endpoint name must be in the format of 'namespace/endpoint_name' " "Endpoint name must be in the format of 'namespace/endpoint_name' "
"or 'endpoint_name'" "or 'endpoint_name'"