phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 05:02:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update anthropic, databricks, tgi, together after get_models rename

Browse source
This commit is contained in:
Matthew Farrellee 2025-10-06 09:08:09 -04:00
parent 733e72caf9
commit 8658949bbb
4 changed files with 4 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/remote/inference/databricks/databricks.py
View file

@ -33,7 +33,7 @@ class DatabricksInferenceAdapter(OpenAIMixin):
def get_base_url(self) -> str:
return f"{self.config.url}/serving-endpoints"
async def get_models(self) -> list[str] | None:
async def list_provider_model_ids(self) -> Iterable[str]:
return [
endpoint.name
for endpoint in WorkspaceClient(
@ -68,11 +68,3 @@ class DatabricksInferenceAdapter(OpenAIMixin):
suffix: str | None = None,
) -> OpenAICompletion:
raise NotImplementedError()
async def list_provider_model_ids(self) -> Iterable[str]:
return [
endpoint.name
for endpoint in WorkspaceClient(
host=self.config.url, token=self.get_api_key()
).serving_endpoints.list() # TODO: this is not async
]