From fff18151e6cc690c8b95f2d6ca303c7effda3318 Mon Sep 17 00:00:00 2001 From: Swapna Lekkala Date: Wed, 27 Aug 2025 18:48:07 -0700 Subject: [PATCH] fix for pre-commit --- docs/source/providers/agents/index.md | 12 ++++++------ .../providers/remote/inference/vertexai/vertexai.py | 13 +------------ 2 files changed, 7 insertions(+), 18 deletions(-) diff --git a/docs/source/providers/agents/index.md b/docs/source/providers/agents/index.md index a2c48d4b9..046db6bff 100644 --- a/docs/source/providers/agents/index.md +++ b/docs/source/providers/agents/index.md @@ -4,12 +4,12 @@ Agents API for creating and interacting with agentic systems. - Main functionalities provided by this API: - - Create agents with specific instructions and ability to use tools. - - Interactions with agents are grouped into sessions ("threads"), and each interaction is called a "turn". - - Agents can be provided with various tools (see the ToolGroups and ToolRuntime APIs for more details). - - Agents can be provided with various shields (see the Safety API for more details). - - Agents can also use Memory to retrieve information from knowledge bases. See the RAG Tool and Vector IO APIs for more details. +Main functionalities provided by this API: +- Create agents with specific instructions and ability to use tools. +- Interactions with agents are grouped into sessions ("threads"), and each interaction is called a "turn". +- Agents can be provided with various tools (see the ToolGroups and ToolRuntime APIs for more details). +- Agents can be provided with various shields (see the Safety API for more details). +- Agents can also use Memory to retrieve information from knowledge bases. See the RAG Tool and Vector IO APIs for more details. This section contains documentation for all available providers for the **agents** API. diff --git a/llama_stack/providers/remote/inference/vertexai/vertexai.py b/llama_stack/providers/remote/inference/vertexai/vertexai.py index f195977b6..8807fd0e6 100644 --- a/llama_stack/providers/remote/inference/vertexai/vertexai.py +++ b/llama_stack/providers/remote/inference/vertexai/vertexai.py @@ -6,9 +6,7 @@ from typing import Any -from openai.types.chat import ChatCompletionContentPartImageParam, ChatCompletionContentPartTextParam - -from llama_stack.apis.inference import ChatCompletionRequest, RerankResponse +from llama_stack.apis.inference import ChatCompletionRequest from llama_stack.providers.utils.inference.litellm_openai_mixin import ( LiteLLMOpenAIMixin, ) @@ -52,12 +50,3 @@ class VertexAIInferenceAdapter(LiteLLMOpenAIMixin): params.pop("api_key", None) 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")