Skip to content

Commit 97a224f

Browse files
DeanChensjcopybara-github
authored andcommitted
chore: Clean up rag memory dependencies
PiperOrigin-RevId: 823556599
1 parent 64d9d69 commit 97a224f

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

src/google/adk/dependencies/vertexai.py

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

1717
import vertexai
18+
from vertexai.preview import rag

src/google/adk/memory/vertex_ai_rag_memory_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@
2424

2525
from google.genai import types
2626
from typing_extensions import override
27-
from vertexai.preview import rag
2827

2928
from . import _utils
29+
from ..dependencies.vertexai import rag
3030
from .base_memory_service import BaseMemoryService
3131
from .base_memory_service import SearchMemoryResponse
3232
from .memory_entry import MemoryEntry

0 commit comments

Comments
 (0)