Namespace imports

griptape/drivers/__init__.py

@@ -1,138 +1,125 @@
-from .prompt.base_prompt_driver import BasePromptDriver
-from .prompt.openai_chat_prompt_driver import OpenAiChatPromptDriver
-from .prompt.azure_openai_chat_prompt_driver import AzureOpenAiChatPromptDriver
-from .prompt.cohere_prompt_driver import CoherePromptDriver
-from .prompt.huggingface_pipeline_prompt_driver import HuggingFacePipelinePromptDriver
-from .prompt.huggingface_hub_prompt_driver import HuggingFaceHubPromptDriver
-from .prompt.anthropic_prompt_driver import AnthropicPromptDriver
-from .prompt.amazon_sagemaker_jumpstart_prompt_driver import AmazonSageMakerJumpstartPromptDriver
-from .prompt.amazon_bedrock_prompt_driver import AmazonBedrockPromptDriver
-from .prompt.google_prompt_driver import GooglePromptDriver
-from .prompt.dummy_prompt_driver import DummyPromptDriver
-from .prompt.ollama_prompt_driver import OllamaPromptDriver
-
-from .memory.conversation.base_conversation_memory_driver import BaseConversationMemoryDriver
-from .memory.conversation.local_conversation_memory_driver import LocalConversationMemoryDriver
-from .memory.conversation.amazon_dynamodb_conversation_memory_driver import AmazonDynamoDbConversationMemoryDriver
-from .memory.conversation.redis_conversation_memory_driver import RedisConversationMemoryDriver
-from .memory.conversation.griptape_cloud_conversation_memory_driver import GriptapeCloudConversationMemoryDriver
-
-from .embedding.base_embedding_driver import BaseEmbeddingDriver
-from .embedding.openai_embedding_driver import OpenAiEmbeddingDriver
-from .embedding.azure_openai_embedding_driver import AzureOpenAiEmbeddingDriver
-from .embedding.amazon_sagemaker_jumpstart_embedding_driver import AmazonSageMakerJumpstartEmbeddingDriver
-from .embedding.amazon_bedrock_titan_embedding_driver import AmazonBedrockTitanEmbeddingDriver
-from .embedding.amazon_bedrock_cohere_embedding_driver import AmazonBedrockCohereEmbeddingDriver
-from .embedding.voyageai_embedding_driver import VoyageAiEmbeddingDriver
-from .embedding.huggingface_hub_embedding_driver import HuggingFaceHubEmbeddingDriver
-from .embedding.google_embedding_driver import GoogleEmbeddingDriver
-from .embedding.dummy_embedding_driver import DummyEmbeddingDriver
-from .embedding.cohere_embedding_driver import CohereEmbeddingDriver
-from .embedding.ollama_embedding_driver import OllamaEmbeddingDriver
-
-from .vector.base_vector_store_driver import BaseVectorStoreDriver
-from .vector.local_vector_store_driver import LocalVectorStoreDriver
-from .vector.pinecone_vector_store_driver import PineconeVectorStoreDriver
-from .vector.marqo_vector_store_driver import MarqoVectorStoreDriver
-from .vector.mongodb_atlas_vector_store_driver import MongoDbAtlasVectorStoreDriver
-from .vector.redis_vector_store_driver import RedisVectorStoreDriver
-from .vector.opensearch_vector_store_driver import OpenSearchVectorStoreDriver
-from .vector.amazon_opensearch_vector_store_driver import AmazonOpenSearchVectorStoreDriver
-from .vector.pgvector_vector_store_driver import PgVectorVectorStoreDriver
-from .vector.azure_mongodb_vector_store_driver import AzureMongoDbVectorStoreDriver
-from .vector.dummy_vector_store_driver import DummyVectorStoreDriver
-from .vector.qdrant_vector_store_driver import QdrantVectorStoreDriver
-from .vector.astradb_vector_store_driver import AstraDbVectorStoreDriver
-from .vector.griptape_cloud_vector_store_driver import GriptapeCloudVectorStoreDriver
-
-from .sql.base_sql_driver import BaseSqlDriver
-from .sql.amazon_redshift_sql_driver import AmazonRedshiftSqlDriver
-from .sql.snowflake_sql_driver import SnowflakeSqlDriver
+from .prompt import BasePromptDriver
+from .prompt.openai import OpenAiChatPromptDriver
+from .prompt.openai import AzureOpenAiChatPromptDriver
+from .prompt.cohere import CoherePromptDriver
+from .prompt.huggingface_pipeline import HuggingFacePipelinePromptDriver
+from .prompt.huggingface_hub import HuggingFaceHubPromptDriver
+from .prompt.anthropic import AnthropicPromptDriver
+from .prompt.amazon_sagemaker_jumpstart import AmazonSageMakerJumpstartPromptDriver
+from .prompt.amazon_bedrock import AmazonBedrockPromptDriver
+from .prompt.google import GooglePromptDriver
+from .prompt.dummy import DummyPromptDriver
+from .prompt.ollama import OllamaPromptDriver
+
+from .memory.conversation import BaseConversationMemoryDriver
+from .memory.conversation.local import LocalConversationMemoryDriver
+from .memory.conversation.amazon_dynamodb import AmazonDynamoDbConversationMemoryDriver
+from .memory.conversation.redis import RedisConversationMemoryDriver
+from .memory.conversation.griptape_cloud import GriptapeCloudConversationMemoryDriver
+
+from .embedding import BaseEmbeddingDriver
+from .embedding.openai import OpenAiEmbeddingDriver
+from .embedding.openai import AzureOpenAiEmbeddingDriver
+from .embedding.amazon_sagemaker_jumpstart import AmazonSageMakerJumpstartEmbeddingDriver
+from .embedding.amazon_bedrock import AmazonBedrockTitanEmbeddingDriver, AmazonBedrockCohereEmbeddingDriver
+from .embedding.voyageai import VoyageAiEmbeddingDriver
+from .embedding.huggingface_hub import HuggingFaceHubEmbeddingDriver
+from .embedding.google import GoogleEmbeddingDriver
+from .embedding.dummy import DummyEmbeddingDriver
+from .embedding.cohere import CohereEmbeddingDriver
+from .embedding.ollama import OllamaEmbeddingDriver
+
+from .vector import BaseVectorStoreDriver
+from .vector.local import LocalVectorStoreDriver
+from .vector.pinecone import PineconeVectorStoreDriver
+from .vector.marqo import MarqoVectorStoreDriver
+from .vector.mongodb_atlas import MongoDbAtlasVectorStoreDriver
+from .vector.redis import RedisVectorStoreDriver
+from .vector.opensearch import OpenSearchVectorStoreDriver
+from .vector.amazon_opensearch import AmazonOpenSearchVectorStoreDriver
+from .vector.pgvector import PgVectorVectorStoreDriver
+from .vector.azure_mongodb import AzureMongoDbVectorStoreDriver
+from .vector.dummy import DummyVectorStoreDriver
+from .vector.qdrant import QdrantVectorStoreDriver
+from .vector.astradb import AstraDbVectorStoreDriver
+from .vector.griptape_cloud import GriptapeCloudVectorStoreDriver
+
+from .sql import BaseSqlDriver
 from .sql.sql_driver import SqlDriver
+from .sql.amazon_redshift import AmazonRedshiftSqlDriver
+from .sql.snowflake import SnowflakeSqlDriver
 
-from .image_generation_model.base_image_generation_model_driver import BaseImageGenerationModelDriver
-from .image_generation_model.bedrock_stable_diffusion_image_generation_model_driver import (
-    BedrockStableDiffusionImageGenerationModelDriver,
-)
-from .image_generation_model.bedrock_titan_image_generation_model_driver import BedrockTitanImageGenerationModelDriver
+from .image_generation_model import BaseImageGenerationModelDriver
+from .image_generation_model.bedrock_stable_diffusion import BedrockStableDiffusionImageGenerationModelDriver
+from .image_generation_model.bedrock_titan import BedrockTitanImageGenerationModelDriver
 
-from .image_generation_pipeline.base_image_generation_pipeline_driver import (
-    BaseDiffusionImageGenerationPipelineDriver,
-)
-from .image_generation_pipeline.stable_diffusion_3_image_generation_pipeline_driver import (
-    StableDiffusion3ImageGenerationPipelineDriver,
-)
-from .image_generation_pipeline.stable_diffusion_3_img_2_img_image_generation_pipeline_driver import (
-    StableDiffusion3Img2ImgImageGenerationPipelineDriver,
-)
-from .image_generation_pipeline.stable_diffusion_3_controlnet_image_generation_pipeline_driver import (
+from .image_generation_pipeline import BaseDiffusionImageGenerationPipelineDriver
+from .image_generation_pipeline.stable_diffusion_3 import StableDiffusion3ImageGenerationPipelineDriver
+from .image_generation_pipeline.stable_diffusion_3_img_2_img import StableDiffusion3Img2ImgImageGenerationPipelineDriver
+from .image_generation_pipeline.stable_diffusion_3_controlnet import (
     StableDiffusion3ControlNetImageGenerationPipelineDriver,
 )
 
-from .image_generation.base_image_generation_driver import BaseImageGenerationDriver
-from .image_generation.base_multi_model_image_generation_driver import BaseMultiModelImageGenerationDriver
-from .image_generation.openai_image_generation_driver import OpenAiImageGenerationDriver
-from .image_generation.leonardo_image_generation_driver import LeonardoImageGenerationDriver
-from .image_generation.amazon_bedrock_image_generation_driver import AmazonBedrockImageGenerationDriver
-from .image_generation.azure_openai_image_generation_driver import AzureOpenAiImageGenerationDriver
-from .image_generation.dummy_image_generation_driver import DummyImageGenerationDriver
-from .image_generation.huggingface_pipeline_image_generation_driver import (
-    HuggingFacePipelineImageGenerationDriver,
-)
-
-from .web_scraper.base_web_scraper_driver import BaseWebScraperDriver
-from .web_scraper.trafilatura_web_scraper_driver import TrafilaturaWebScraperDriver
-from .web_scraper.markdownify_web_scraper_driver import MarkdownifyWebScraperDriver
-from .web_scraper.proxy_web_scraper_driver import ProxyWebScraperDriver
-
-from .web_search.base_web_search_driver import BaseWebSearchDriver
-from .web_search.google_web_search_driver import GoogleWebSearchDriver
-from .web_search.duck_duck_go_web_search_driver import DuckDuckGoWebSearchDriver
-from .web_search.exa_web_search_driver import ExaWebSearchDriver
-from .web_search.tavily_web_search_driver import TavilyWebSearchDriver
-
-from .event_listener.base_event_listener_driver import BaseEventListenerDriver
-from .event_listener.amazon_sqs_event_listener_driver import AmazonSqsEventListenerDriver
-from .event_listener.webhook_event_listener_driver import WebhookEventListenerDriver
-from .event_listener.aws_iot_core_event_listener_driver import AwsIotCoreEventListenerDriver
-from .event_listener.griptape_cloud_event_listener_driver import GriptapeCloudEventListenerDriver
-from .event_listener.pusher_event_listener_driver import PusherEventListenerDriver
-
-from .file_manager.base_file_manager_driver import BaseFileManagerDriver
-from .file_manager.local_file_manager_driver import LocalFileManagerDriver
-from .file_manager.amazon_s3_file_manager_driver import AmazonS3FileManagerDriver
-from .file_manager.griptape_cloud_file_manager_driver import GriptapeCloudFileManagerDriver
-
-from .rerank.base_rerank_driver import BaseRerankDriver
-from .rerank.cohere_rerank_driver import CohereRerankDriver
-
-from .ruleset.base_ruleset_driver import BaseRulesetDriver
-from .ruleset.local_ruleset_driver import LocalRulesetDriver
-from .ruleset.griptape_cloud_ruleset_driver import GriptapeCloudRulesetDriver
-
-from .text_to_speech.base_text_to_speech_driver import BaseTextToSpeechDriver
-from .text_to_speech.dummy_text_to_speech_driver import DummyTextToSpeechDriver
-from .text_to_speech.elevenlabs_text_to_speech_driver import ElevenLabsTextToSpeechDriver
-from .text_to_speech.openai_text_to_speech_driver import OpenAiTextToSpeechDriver
-from .text_to_speech.azure_openai_text_to_speech_driver import AzureOpenAiTextToSpeechDriver
-
-from .structure_run.base_structure_run_driver import BaseStructureRunDriver
-from .structure_run.griptape_cloud_structure_run_driver import GriptapeCloudStructureRunDriver
-from .structure_run.local_structure_run_driver import LocalStructureRunDriver
-
-from .audio_transcription.base_audio_transcription_driver import BaseAudioTranscriptionDriver
-from .audio_transcription.dummy_audio_transcription_driver import DummyAudioTranscriptionDriver
-from .audio_transcription.openai_audio_transcription_driver import OpenAiAudioTranscriptionDriver
-
-from .observability.base_observability_driver import BaseObservabilityDriver
-from .observability.no_op_observability_driver import NoOpObservabilityDriver
-from .observability.open_telemetry_observability_driver import OpenTelemetryObservabilityDriver
-from .observability.griptape_cloud_observability_driver import GriptapeCloudObservabilityDriver
-from .observability.datadog_observability_driver import DatadogObservabilityDriver
-
-from .assistant.base_assistant_driver import BaseAssistantDriver
-from .assistant.griptape_cloud_assistant_driver import GriptapeCloudAssistantDriver
-from .assistant.openai_assistant_driver import OpenAiAssistantDriver
+from .image_generation import BaseImageGenerationDriver
+from .image_generation import BaseMultiModelImageGenerationDriver
+from .image_generation.openai import OpenAiImageGenerationDriver, AzureOpenAiImageGenerationDriver
+from .image_generation.leonardo import LeonardoImageGenerationDriver
+from .image_generation.amazon_bedrock import AmazonBedrockImageGenerationDriver
+from .image_generation.dummy import DummyImageGenerationDriver
+from .image_generation.huggingface_pipeline import HuggingFacePipelineImageGenerationDriver
+
+from .web_scraper import BaseWebScraperDriver
+from .web_scraper.trafilatura import TrafilaturaWebScraperDriver
+from .web_scraper.markdownify import MarkdownifyWebScraperDriver
+from .web_scraper.proxy import ProxyWebScraperDriver
+
+from .web_search import BaseWebSearchDriver
+from .web_search.google import GoogleWebSearchDriver
+from .web_search.duck_duck_go import DuckDuckGoWebSearchDriver
+from .web_search.exa import ExaWebSearchDriver
+from .web_search.tavily import TavilyWebSearchDriver
+
+from .event_listener import BaseEventListenerDriver
+from .event_listener.amazon_sqs import AmazonSqsEventListenerDriver
+from .event_listener.webhook import WebhookEventListenerDriver
+from .event_listener.aws_iot_core import AwsIotCoreEventListenerDriver
+from .event_listener.griptape_cloud import GriptapeCloudEventListenerDriver
+from .event_listener.pusher import PusherEventListenerDriver
+
+from .file_manager import BaseFileManagerDriver
+from .file_manager.local import LocalFileManagerDriver
+from .file_manager.amazon_s3 import AmazonS3FileManagerDriver
+from .file_manager.griptape_cloud import GriptapeCloudFileManagerDriver
+
+from .rerank import BaseRerankDriver
+from .rerank.cohere import CohereRerankDriver
+
+from .ruleset import BaseRulesetDriver
+from .ruleset.local import LocalRulesetDriver
+from .ruleset.griptape_cloud import GriptapeCloudRulesetDriver
+
+from .text_to_speech import BaseTextToSpeechDriver
+from .text_to_speech.dummy import DummyTextToSpeechDriver
+from .text_to_speech.elevenlabs import ElevenLabsTextToSpeechDriver
+from .text_to_speech.openai import OpenAiTextToSpeechDriver, AzureOpenAiTextToSpeechDriver
+
+from .structure_run import BaseStructureRunDriver
+from .structure_run.griptape_cloud import GriptapeCloudStructureRunDriver
+from .structure_run.local import LocalStructureRunDriver
+
+from .audio_transcription import BaseAudioTranscriptionDriver
+from .audio_transcription.dummy import DummyAudioTranscriptionDriver
+from .audio_transcription.openai import OpenAiAudioTranscriptionDriver
+
+from .observability import BaseObservabilityDriver
+from .observability.no_op import NoOpObservabilityDriver
+from .observability.open_telemetry import OpenTelemetryObservabilityDriver
+from .observability.griptape_cloud import GriptapeCloudObservabilityDriver
+from .observability.datadog import DatadogObservabilityDriver
+
+from .assistant import BaseAssistantDriver
+from .assistant.griptape_cloud import GriptapeCloudAssistantDriver
+from .assistant.openai import OpenAiAssistantDriver
 
 __all__ = [
     "BasePromptDriver",

griptape/drivers/assistant/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.assistant.base_assistant_driver import BaseAssistantDriver
+
+__all__ = ["BaseAssistantDriver"]

griptape/drivers/assistant/griptape_cloud/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.assistant.griptape_cloud_assistant_driver import GriptapeCloudAssistantDriver
+
+__all__ = ["GriptapeCloudAssistantDriver"]

griptape/drivers/assistant/openai/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.assistant.openai_assistant_driver import OpenAiAssistantDriver
+
+__all__ = ["OpenAiAssistantDriver"]

griptape/drivers/audio_transcription/__init__.py

@@ -0,0 +1,3 @@
+from .base_audio_transcription_driver import BaseAudioTranscriptionDriver
+
+__all__ = ["BaseAudioTranscriptionDriver"]

griptape/drivers/audio_transcription/dummy/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.audio_transcription.dummy_audio_transcription_driver import DummyAudioTranscriptionDriver
+
+__all__ = ["DummyAudioTranscriptionDriver"]

griptape/drivers/audio_transcription/openai/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.audio_transcription.openai_audio_transcription_driver import OpenAiAudioTranscriptionDriver
+
+__all__ = ["OpenAiAudioTranscriptionDriver"]

griptape/drivers/embedding/__init__.py

@@ -0,0 +1,3 @@
+from .base_embedding_driver import BaseEmbeddingDriver
+
+__all__ = ["BaseEmbeddingDriver"]

griptape/drivers/embedding/amazon_bedrock/__init__.py

@@ -0,0 +1,5 @@
+from griptape.drivers.embedding.amazon_bedrock_titan_embedding_driver import AmazonBedrockTitanEmbeddingDriver
+from griptape.drivers.embedding.amazon_bedrock_cohere_embedding_driver import AmazonBedrockCohereEmbeddingDriver
+
+
+__all__ = ["AmazonBedrockTitanEmbeddingDriver", "AmazonBedrockCohereEmbeddingDriver"]

griptape/drivers/embedding/amazon_sagemaker_jumpstart/__init__.py

@@ -0,0 +1,5 @@
+from griptape.drivers.embedding.amazon_sagemaker_jumpstart_embedding_driver import (
+    AmazonSageMakerJumpstartEmbeddingDriver,
+)
+
+__all__ = ["AmazonSageMakerJumpstartEmbeddingDriver"]

griptape/drivers/embedding/azure_openai_embedding_driver.py

@@ -5,7 +5,7 @@
 import openai
 from attrs import Factory, define, field
 
-from griptape.drivers import OpenAiEmbeddingDriver
+from griptape.drivers.embedding.openai import OpenAiEmbeddingDriver
 from griptape.tokenizers import OpenAiTokenizer
 from griptape.utils.decorators import lazy_property
 

griptape/drivers/embedding/cohere/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.cohere_embedding_driver import CohereEmbeddingDriver
+
+__all__ = ["CohereEmbeddingDriver"]

griptape/drivers/embedding/dummy/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.dummy_embedding_driver import DummyEmbeddingDriver
+
+__all__ = ["DummyEmbeddingDriver"]

griptape/drivers/embedding/google/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.google_embedding_driver import GoogleEmbeddingDriver
+
+__all__ = ["GoogleEmbeddingDriver"]

griptape/drivers/embedding/huggingface_hub/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.huggingface_hub_embedding_driver import HuggingFaceHubEmbeddingDriver
+
+__all__ = ["HuggingFaceHubEmbeddingDriver"]

griptape/drivers/embedding/ollama/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.ollama_embedding_driver import OllamaEmbeddingDriver
+
+__all__ = ["OllamaEmbeddingDriver"]

griptape/drivers/embedding/openai/__init__.py

@@ -0,0 +1,4 @@
+from griptape.drivers.embedding.openai_embedding_driver import OpenAiEmbeddingDriver
+from griptape.drivers.embedding.azure_openai_embedding_driver import AzureOpenAiEmbeddingDriver
+
+__all__ = ["OpenAiEmbeddingDriver", "AzureOpenAiEmbeddingDriver"]

griptape/drivers/embedding/openai_embedding_driver.py

@@ -5,7 +5,7 @@
 import openai
 from attrs import Factory, define, field
 
-from griptape.drivers import BaseEmbeddingDriver
+from griptape.drivers.embedding import BaseEmbeddingDriver
 from griptape.tokenizers import OpenAiTokenizer
 from griptape.utils.decorators import lazy_property
 

griptape/drivers/embedding/voyageai/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.embedding.voyageai_embedding_driver import VoyageAiEmbeddingDriver
+
+__all__ = ["VoyageAiEmbeddingDriver"]

griptape/drivers/event_listener/__init__.py

@@ -0,0 +1,3 @@
+from .base_event_listener_driver import BaseEventListenerDriver
+
+__all__ = ["BaseEventListenerDriver"]

griptape/drivers/event_listener/amazon_sqs/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.event_listener.amazon_sqs_event_listener_driver import AmazonSqsEventListenerDriver
+
+__all__ = ["AmazonSqsEventListenerDriver"]

griptape/drivers/event_listener/aws_iot_core/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.event_listener.aws_iot_core_event_listener_driver import AwsIotCoreEventListenerDriver
+
+__all__ = ["AwsIotCoreEventListenerDriver"]

griptape/drivers/event_listener/griptape_cloud/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.event_listener.griptape_cloud_event_listener_driver import GriptapeCloudEventListenerDriver
+
+__all__ = ["GriptapeCloudEventListenerDriver"]

griptape/drivers/event_listener/pusher/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.event_listener.pusher_event_listener_driver import PusherEventListenerDriver
+
+__all__ = ["PusherEventListenerDriver"]

griptape/drivers/event_listener/webhook/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.event_listener.webhook_event_listener_driver import WebhookEventListenerDriver
+
+__all__ = ["WebhookEventListenerDriver"]

griptape/drivers/file_manager/__init__.py

@@ -0,0 +1,3 @@
+from .base_file_manager_driver import BaseFileManagerDriver
+
+__all__ = ["BaseFileManagerDriver"]

griptape/drivers/file_manager/amazon_s3/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.file_manager.amazon_s3_file_manager_driver import AmazonS3FileManagerDriver
+
+__all__ = ["AmazonS3FileManagerDriver"]

griptape/drivers/file_manager/griptape_cloud/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.file_manager.griptape_cloud_file_manager_driver import GriptapeCloudFileManagerDriver
+
+__all__ = ["GriptapeCloudFileManagerDriver"]

griptape/drivers/file_manager/local/__init__.py

@@ -0,0 +1,3 @@
+from griptape.drivers.file_manager.local_file_manager_driver import LocalFileManagerDriver
+
+__all__ = ["LocalFileManagerDriver"]