File tree Expand file tree Collapse file tree
src/google/adk/tools/retrieval Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1313# limitations under the License.
1414
1515from .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
2521try :
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.' )
You can’t perform that action at this time.
0 commit comments