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

fix for pre-commit

Browse source
This commit is contained in:
Swapna Lekkala 2025-08-27 18:48:07 -07:00
parent 21d8c70954
commit fff18151e6
2 changed files with 7 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_stack/providers/remote/inference/vertexai/vertexai.py
View file

@ -6,9 +6,7 @@
from typing import Any from typing import Any
from openai.types.chat import ChatCompletionContentPartImageParam, ChatCompletionContentPartTextParam from llama_stack.apis.inference import ChatCompletionRequest
from llama_stack.apis.inference import ChatCompletionRequest, RerankResponse
from llama_stack.providers.utils.inference.litellm_openai_mixin import ( from llama_stack.providers.utils.inference.litellm_openai_mixin import (
LiteLLMOpenAIMixin, LiteLLMOpenAIMixin,
) )
@ -52,12 +50,3 @@ class VertexAIInferenceAdapter(LiteLLMOpenAIMixin):
params.pop("api_key", None) params.pop("api_key", None)
return params return params
async def rerank(
self,
model: str,
query: str | ChatCompletionContentPartTextParam | ChatCompletionContentPartImageParam,
items: list[str | ChatCompletionContentPartTextParam | ChatCompletionContentPartImageParam],
max_num_results: int | None = None,
) -> RerankResponse:
raise NotImplementedError("Reranking is not supported for Vertex AI")