Skip to content

Commit 1aa9bb1

Browse files
ankursharmascopybara-github
authored andcommitted
chore: Update vertex ai in example store and rag retrieval to use proxy
PiperOrigin-RevId: 825237803
1 parent 61adfcd commit 1aa9bb1

3 files changed

Lines changed: 5 additions & 2 deletions

File tree

src/google/adk/dependencies/vertexai.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,5 @@
1515
from __future__ import annotations
1616

1717
import vertexai
18+
from vertexai.preview import example_stores
1819
from vertexai.preview import rag

src/google/adk/examples/vertex_ai_example_store.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,12 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from __future__ import annotations
16+
1517
from google.genai import types
1618
from typing_extensions import override
17-
from vertexai.preview import example_stores
1819

20+
from ..dependencies.vertexai import example_stores
1921
from .base_example_provider import BaseExampleProvider
2022
from .example import Example
2123

src/google/adk/tools/retrieval/vertex_ai_rag_retrieval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222

2323
from google.genai import types
2424
from typing_extensions import override
25-
from vertexai.preview import rag
2625

26+
from ...dependencies.vertexai import rag
2727
from ...utils.model_name_utils import is_gemini_2_or_above
2828
from ..tool_context import ToolContext
2929
from .base_retrieval_tool import BaseRetrievalTool

0 commit comments

Comments
 (0)