Add flake8-future-annotations ruff rule

griptape/artifacts/list_artifact.py

@@ -1,10 +1,14 @@
-from collections.abc import Sequence
-from typing import Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
 
 from attrs import define, field
 
 from griptape.artifacts import BaseArtifact
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
 
 @define
 class ListArtifact(BaseArtifact):

griptape/common/reference.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import uuid
 from typing import Optional
 

griptape/config/azure_openai_structure_config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Optional
 
 from attrs import Factory, define, field

griptape/config/structure_config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 from attrs import Factory, define, field

griptape/drivers/audio_transcription/dummy_audio_transcription_driver.py

@@ -1,11 +1,15 @@
-from typing import Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
 
 from attrs import define, field
 
-from griptape.artifacts import AudioArtifact, TextArtifact
 from griptape.drivers import BaseAudioTranscriptionDriver
 from griptape.exceptions import DummyException
 
+if TYPE_CHECKING:
+    from griptape.artifacts import AudioArtifact, TextArtifact
+
 
 @define
 class DummyAudioTranscriptionDriver(BaseAudioTranscriptionDriver):

griptape/drivers/embedding/dummy_embedding_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from attrs import define, field
 
 from griptape.drivers import BaseEmbeddingDriver

griptape/drivers/image_generation/dummy_image_generation_driver.py

@@ -1,11 +1,15 @@
-from typing import Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
 
 from attrs import define, field
 
-from griptape.artifacts import ImageArtifact
 from griptape.drivers import BaseImageGenerationDriver
 from griptape.exceptions import DummyException
 
+if TYPE_CHECKING:
+    from griptape.artifacts import ImageArtifact
+
 
 @define
 class DummyImageGenerationDriver(BaseImageGenerationDriver):

griptape/drivers/image_generation/leonardo_image_generation_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import time
 from typing import Literal, Optional

griptape/drivers/image_query/dummy_image_query_driver.py

@@ -1,9 +1,15 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from attrs import define, field
 
-from griptape.artifacts import ImageArtifact, TextArtifact
 from griptape.drivers import BaseImageQueryDriver
 from griptape.exceptions import DummyException
 
+if TYPE_CHECKING:
+    from griptape.artifacts import ImageArtifact, TextArtifact
+
 
 @define
 class DummyImageQueryDriver(BaseImageQueryDriver):

griptape/drivers/memory/conversation/local_conversation_memory_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from typing import Optional
 

griptape/drivers/prompt/azure_openai_chat_prompt_driver.py

@@ -1,11 +1,15 @@
-from typing import Callable, Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Callable, Optional
 
 import openai
 from attrs import Factory, define, field
 
-from griptape.common import PromptStack
 from griptape.drivers import OpenAiChatPromptDriver
 
+if TYPE_CHECKING:
+    from griptape.common import PromptStack
+
 
 @define
 class AzureOpenAiChatPromptDriver(OpenAiChatPromptDriver):

griptape/drivers/rerank/base_rerank_driver.py

@@ -1,8 +1,12 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
 
 from attrs import define
 
-from griptape.artifacts import TextArtifact
+if TYPE_CHECKING:
+    from griptape.artifacts import TextArtifact
 
 
 @define(kw_only=True)

griptape/drivers/sql/base_sql_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from typing import Any, Optional

griptape/drivers/structure_run/base_structure_run_driver.py

@@ -1,8 +1,12 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
 
 from attrs import Factory, define, field
 
-from griptape.artifacts import BaseArtifact
+if TYPE_CHECKING:
+    from griptape.artifacts import BaseArtifact
 
 
 @define

griptape/drivers/text_to_speech/dummy_text_to_speech_driver.py

@@ -1,9 +1,15 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from attrs import define, field
 
-from griptape.artifacts.audio_artifact import AudioArtifact
 from griptape.drivers import BaseTextToSpeechDriver
 from griptape.exceptions import DummyException
 
+if TYPE_CHECKING:
+    from griptape.artifacts.audio_artifact import AudioArtifact
+
 
 @define
 class DummyTextToSpeechDriver(BaseTextToSpeechDriver):

griptape/drivers/vector/dummy_vector_store_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 from attrs import Factory, define, field

griptape/drivers/vector/griptape_cloud_knowledge_base_vector_store_driver.py

@@ -1,12 +1,16 @@
-from typing import Any, Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Optional
 from urllib.parse import urljoin
 
 import requests
 from attrs import Factory, define, field
 
-from griptape.artifacts import ListArtifact, TextArtifact
 from griptape.drivers import BaseEmbeddingDriver, BaseVectorStoreDriver, DummyEmbeddingDriver
 
+if TYPE_CHECKING:
+    from griptape.artifacts import ListArtifact, TextArtifact
+
 
 @define
 class GriptapeCloudKnowledgeBaseVectorStoreDriver(BaseVectorStoreDriver):

griptape/drivers/vector/local_vector_store_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import os
 import threading

griptape/drivers/vector/pgvector_vector_store_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import uuid
 from collections import OrderedDict
 from dataclasses import dataclass

griptape/drivers/web_scraper/markdownify_web_scraper_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import re
 from typing import Optional
 

griptape/drivers/web_search/google_web_search_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 
 import requests

griptape/engines/image_query/image_query_engine.py

@@ -1,7 +1,12 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from attrs import define, field
 
-from griptape.artifacts import ImageArtifact, TextArtifact
-from griptape.drivers import BaseImageQueryDriver
+if TYPE_CHECKING:
+    from griptape.artifacts import ImageArtifact, TextArtifact
+    from griptape.drivers import BaseImageQueryDriver
 
 
 @define

griptape/engines/rag/modules/base_rag_module.py

@@ -1,11 +1,15 @@
+from __future__ import annotations
+
 from abc import ABC
 from concurrent import futures
-from typing import Any, Callable, Optional
+from typing import TYPE_CHECKING, Any, Callable, Optional
 
 from attrs import Factory, define, field
 
 from griptape.common import Message, PromptStack
-from griptape.engines.rag import RagContext
+
+if TYPE_CHECKING:
+    from griptape.engines.rag import RagContext
 
 
 @define(kw_only=True)

griptape/engines/rag/modules/response/footnote_prompt_response_rag_module.py

@@ -1,11 +1,17 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from attrs import define
 
 from griptape import utils
-from griptape.artifacts import TextArtifact
-from griptape.engines.rag import RagContext
 from griptape.engines.rag.modules import PromptResponseRagModule
 from griptape.utils import J2
 
+if TYPE_CHECKING:
+    from griptape.artifacts import TextArtifact
+    from griptape.engines.rag import RagContext
+
 
 @define(kw_only=True)
 class FootnotePromptResponseRagModule(PromptResponseRagModule):

griptape/engines/rag/modules/response/metadata_before_response_rag_module.py

@@ -1,11 +1,15 @@
-from typing import Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
 
 from attrs import define, field
 
-from griptape.engines.rag import RagContext
 from griptape.engines.rag.modules import BaseBeforeResponseRagModule
 from griptape.utils import J2
 
+if TYPE_CHECKING:
+    from griptape.engines.rag import RagContext
+
 
 @define(kw_only=True)
 class MetadataBeforeResponseRagModule(BaseBeforeResponseRagModule):

griptape/engines/rag/modules/response/prompt_response_rag_module.py

@@ -1,13 +1,17 @@
-from typing import Callable
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Callable
 
 from attrs import Factory, define, field
 
 from griptape.artifacts.text_artifact import TextArtifact
-from griptape.drivers import BasePromptDriver
-from griptape.engines.rag import RagContext
 from griptape.engines.rag.modules import BaseResponseRagModule
 from griptape.utils import J2
 
+if TYPE_CHECKING:
+    from griptape.drivers import BasePromptDriver
+    from griptape.engines.rag import RagContext
+
 
 @define(kw_only=True)
 class PromptResponseRagModule(BaseResponseRagModule):

griptape/engines/rag/modules/response/rulesets_before_response_rag_module.py

@@ -1,10 +1,16 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from attrs import define, field
 
-from griptape.engines.rag import RagContext
 from griptape.engines.rag.modules import BaseBeforeResponseRagModule
-from griptape.rules import Ruleset
 from griptape.utils import J2
 
+if TYPE_CHECKING:
+    from griptape.engines.rag import RagContext
+    from griptape.rules import Ruleset
+
 
 @define
 class RulesetsBeforeResponseRagModule(BaseBeforeResponseRagModule):

griptape/engines/rag/rag_engine.py

@@ -1,9 +1,13 @@
-from typing import Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
 
 from attrs import define, field
 
 from griptape.engines.rag import RagContext
-from griptape.engines.rag.stages import QueryRagStage, ResponseRagStage, RetrievalRagStage
+
+if TYPE_CHECKING:
+    from griptape.engines.rag.stages import QueryRagStage, ResponseRagStage, RetrievalRagStage
 
 
 @define(kw_only=True)

griptape/engines/rag/stages/query_rag_stage.py

@@ -1,13 +1,19 @@
+from __future__ import annotations
+
 import logging
-from collections.abc import Sequence
+from typing import TYPE_CHECKING
 
 from attrs import define, field
 
 from griptape import utils
-from griptape.engines.rag import RagContext
-from griptape.engines.rag.modules import BaseQueryRagModule, BaseRagModule
 from griptape.engines.rag.stages import BaseRagStage
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+    from griptape.engines.rag import RagContext
+    from griptape.engines.rag.modules import BaseQueryRagModule, BaseRagModule
+
 
 @define(kw_only=True)
 class QueryRagStage(BaseRagStage):