Rename TaskMemory directory and update imports (#452)

Author: andrewfrench

griptape/memory/__init__.py

@@ -1,4 +1,4 @@
-from .tool.task_memory import TaskMemory
+from .task.task_memory import TaskMemory
 
 
 __all__ = ["TaskMemory"]

griptape/memory/tool/__init__.py griptape/memory/task/__init__.py

@@ -1,6 +1,6 @@
 from attr import define, field
 from griptape.artifacts import BaseArtifact, ListArtifact, BlobArtifact, InfoArtifact
-from griptape.memory.tool.storage import BaseArtifactStorage
+from griptape.memory.task.storage import BaseArtifactStorage
 
 
 @define

griptape/memory/tool/storage/__init__.py griptape/memory/task/storage/__init__.py

@@ -2,7 +2,7 @@
 from typing import TYPE_CHECKING
 from attr import define, field
 from griptape.artifacts import TextArtifact, BaseArtifact, ListArtifact
-from griptape.memory.tool.storage import BaseArtifactStorage
+from griptape.memory.task.storage import BaseArtifactStorage
 
 if TYPE_CHECKING:
     from griptape.engines import BaseSummaryEngine, CsvExtractionEngine, JsonExtractionEngine, VectorQueryEngine

griptape/memory/tool/storage/base_artifact_storage.py griptape/memory/task/storage/base_artifact_storage.py

@@ -6,7 +6,7 @@
 from griptape.mixins import ActivityMixin
 
 if TYPE_CHECKING:
-    from griptape.memory.tool.storage import BaseArtifactStorage
+    from griptape.memory.task.storage import BaseArtifactStorage
     from griptape.tasks import ActionSubtask
 
 

griptape/memory/tool/storage/blob_artifact_storage.py griptape/memory/task/storage/blob_artifact_storage.py

@@ -14,7 +14,7 @@
 from griptape.memory.meta import MetaMemory
 from griptape.memory.structure import ConversationMemory
 from griptape.memory import TaskMemory
-from griptape.memory.tool.storage import BlobArtifactStorage, TextArtifactStorage
+from griptape.memory.task.storage import BlobArtifactStorage, TextArtifactStorage
 from griptape.rules import Ruleset, Rule
 from griptape.events import BaseEvent
 from griptape.tokenizers import OpenAiTokenizer

griptape/memory/tool/storage/text_artifact_storage.py griptape/memory/task/storage/text_artifact_storage.py

@@ -1,6 +1,6 @@
 import pytest
 from griptape.artifacts import BlobArtifact, TextArtifact
-from griptape.memory.tool.storage import BlobArtifactStorage
+from griptape.memory.task.storage import BlobArtifactStorage
 
 
 class TestBlobArtifactStorage:

griptape/memory/tool/task_memory.py griptape/memory/task/task_memory.py

@@ -2,7 +2,7 @@
 from griptape.artifacts import CsvRowArtifact, BlobArtifact, ErrorArtifact, InfoArtifact
 from griptape.artifacts import TextArtifact, ListArtifact
 from griptape.memory import TaskMemory
-from griptape.memory.tool.storage import BlobArtifactStorage, TextArtifactStorage
+from griptape.memory.task.storage import BlobArtifactStorage, TextArtifactStorage
 from griptape.structures import Agent
 from griptape.tasks import ActionSubtask
 from tests.mocks.mock_tool.tool import MockTool

griptape/structures/structure.py

@@ -1,7 +1,7 @@
 import pytest
 from griptape.memory.structure import ConversationMemory
 from griptape.memory import TaskMemory
-from griptape.memory.tool.storage import TextArtifactStorage
+from griptape.memory.task.storage import TextArtifactStorage
 from griptape.rules import Rule, Ruleset
 from griptape.structures import Agent
 from griptape.tasks import PromptTask, BaseTask, ToolkitTask

tests/unit/memory/tool/storage/test_blob_artifact_storage.py

@@ -1,7 +1,7 @@
 import pytest
 
 from griptape.artifacts import TextArtifact
-from griptape.memory.tool.storage import TextArtifactStorage
+from griptape.memory.task.storage import TextArtifactStorage
 from griptape.rules import Rule, Ruleset
 from griptape.tokenizers import OpenAiTokenizer
 from griptape.tasks import PromptTask, BaseTask, ToolkitTask

tests/unit/memory/tool/test_task_memory.py

@@ -1,6 +1,6 @@
 import pytest
 
-from griptape.memory.tool.storage import TextArtifactStorage
+from griptape.memory.task.storage import TextArtifactStorage
 from tests.mocks.mock_prompt_driver import MockPromptDriver
 from griptape.rules import Rule, Ruleset
 from griptape.tasks import PromptTask, BaseTask, ToolkitTask

tests/unit/structures/test_agent.py

@@ -2,7 +2,7 @@
 from griptape.drivers import LocalVectorStoreDriver
 from griptape.engines import VectorQueryEngine, PromptSummaryEngine, CsvExtractionEngine, JsonExtractionEngine
 from griptape.memory import TaskMemory
-from griptape.memory.tool.storage import TextArtifactStorage, BlobArtifactStorage
+from griptape.memory.task.storage import TextArtifactStorage, BlobArtifactStorage
 from tests.mocks.mock_embedding_driver import MockEmbeddingDriver
 from tests.mocks.mock_prompt_driver import MockPromptDriver