fix tool import links (#6892)

* fix import links * update load and search tool import
run-llama · Jul 13, 2023 · 54ff240 · 54ff240
1 parent fc3eed1
commit 54ff240
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 8 deletions.
diff --git a/docs/core_modules/agent_modules/tools/llamahub_tools_guide.md b/docs/core_modules/agent_modules/tools/llamahub_tools_guide.md
@@ -29,7 +29,7 @@ A usage example is given below:
 
 ```python
 from llama_hub.wikipedia.base import WikipediaReader
-from llama_hub.tools.on_demand_loader_tool import OnDemandLoaderTool
+from llama_index.tools.on_demand_loader_tool import OnDemandLoaderTool
 
 tool = OnDemandLoaderTool.from_defaults(
 	reader,
@@ -49,8 +49,8 @@ This is helpful for any API endpoint that will by default return large volumes o
 Example usage is shown below:
 
 ```python
-from llama_hub.tools.tool_spec.wikipedia.base import WikipediaToolSpec
-from llama_hub.tools.tool_spec.load_and_search.base import LoadAndSearchToolSpec
+from llama_hub.tools.wikipedia.base import WikipediaToolSpec
+from llama_index.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
 
 wiki_spec = WikipediaToolSpec()
 # Get the search wikipedia tool

diff --git a/docs/core_modules/agent_modules/tools/root.md b/docs/core_modules/agent_modules/tools/root.md
@@ -25,7 +25,7 @@ Our Tool Specs and Tools can be imported from the `llama-hub` package.
 To use with our agent,
 ```python
 from llama_index.agent import OpenAIAgent
-from llama_hub.tools.tool_spec.gmail.base import GmailToolSpec
+from llama_hub.tools.gmail.base import GmailToolSpec
 
 tool_spec = GmailToolSpec()
 agent = OpenAIAgent.from_tools(tool_spec.to_tool_list(), verbose=True)

diff --git a/docs/core_modules/agent_modules/tools/usage_pattern.md b/docs/core_modules/agent_modules/tools/usage_pattern.md
@@ -7,7 +7,7 @@ LlamaHub Tool Specs and Tools can be imported from the `llama-hub` package. They
 To use with our OpenAIAgent,
 ```python
 from llama_index.agent import OpenAIAgent
-from llama_hub.tools.tool_spec.gmail.base import GmailToolSpec
+from llama_hub.tools.gmail.base import GmailToolSpec
 
 tool_spec = GmailToolSpec()
 agent = OpenAIAgent.from_tools(tool_spec.to_tool_list(), verbose=True)

diff --git a/llama_index/tools/tool_spec/load_and_search/README.md b/llama_index/tools/tool_spec/load_and_search/README.md
@@ -8,9 +8,9 @@ This Tool Spec is intended to wrap other tools, allowing the Agent to perform se
 Here's an example usage of the LoadAndSearchToolSpec.
 
 ```python
-from llama_hub.tools.tool_spec.load_and_search.base import LoadAndSearchToolSpec
+from llama_index.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
 from llama_index.agent import OpenAIAgent
-from llama_hub.tools.tool_spec.wikipedia.base import WikipediaToolSpec
+from llama_hub.tools.wikipedia.base import WikipediaToolSpec
 
 wiki_spec = WikipediaToolSpec()
 

diff --git a/llama_index/tools/tool_spec/load_and_search/__init__.py b/llama_index/tools/tool_spec/load_and_search/__init__.py
@@ -1 +1,3 @@
-# init
+from llama_index.tools.tool_spec.load_and_search.base import LoadAndSearchToolSpec
+
+__all__ = ["LoadAndSearchToolSpec"]