Skip to content

Commit f14230d

Browse files
Execute final tests before PR
1 parent 86a4487 commit f14230d

1 file changed

Lines changed: 15 additions & 4 deletions

File tree

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

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,9 @@
1313
# limitations under the License.
1414

1515
from .base_retrieval_tool import BaseRetrievalTool
16-
from .files_retrieval import FilesRetrieval
17-
from .llama_index_retrieval import LlamaIndexRetrieval
1816

1917
__all__ = [
2018
'BaseRetrievalTool',
21-
'FilesRetrieval',
22-
'LlamaIndexRetrieval',
2319
]
2420

2521
try:
@@ -34,3 +30,18 @@
3430
'The Vertex sdk is not installed. If you want to use the Vertex RAG with'
3531
' agents, please install it. If not, you can ignore this warning.'
3632
)
33+
34+
try:
35+
36+
from .files_retrieval import FilesRetrieval
37+
from .llama_index_retrieval import LlamaIndexRetrieval
38+
39+
__all__.extend([
40+
'FilesRetrieval',
41+
'LlamaIndexRetrieval',
42+
])
43+
except ImportError:
44+
import logging
45+
46+
logger = logging.getLogger('google_adk.' + __name__)
47+
logger.debug('Please install LLama-Index if you wish to use these imports.')

0 commit comments

Comments
 (0)