Fix doc examples (#819)

griptape-ai · Jun 3, 2024 · 385cb62 · 385cb62
1 parent 81ec3b6
commit 385cb62
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docs/griptape-framework/misc/events.md b/docs/griptape-framework/misc/events.md
@@ -163,7 +163,7 @@ You can also use the [Stream](../../reference/griptape/utils/stream.md) utility
 from griptape.utils import Stream
 from griptape.tasks import ToolkitTask
 from griptape.structures import Pipeline
-from griptape.tools import WebScraper
+from griptape.tools import WebScraper, TaskMemoryClient
 
 
 pipeline = Pipeline()

diff --git a/docs/griptape-framework/structures/task-memory.md b/docs/griptape-framework/structures/task-memory.md
@@ -135,7 +135,7 @@ Let's say we want to query the contents of a very large webpage.
 from griptape.structures import Agent
 from griptape.tools import WebScraper
 
-# Create an agent with the Calculator tool
+# Create an agent with the WebScraper tool
 agent = Agent(tools=[WebScraper()])
 
 agent.run(
@@ -315,7 +315,7 @@ This can create a loop where the same data is stored and queried over and over a
 
 ```python
 from griptape.structures import Agent
-from griptape.tools import WebScraper
+from griptape.tools import WebScraper, TaskMemoryClient
 
 agent = Agent(
     tools=[
@@ -331,7 +331,7 @@ Task Memory may not be necessary for all use cases. If the data returned by a To
 
 ```python
 from griptape.structures import Agent
-from griptape.tools import WebScraper
+from griptape.tools import Calculator
 
 agent = Agent(
     tools=[