poetry lock --no-update

griptape-ai · Sep 25, 2024 · 1182b67 · 1182b67
1 parent 8084eab
commit 1182b67
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/griptape/drivers/video_generation/dream_machine_video_generation_driver.py b/griptape/drivers/video_generation/dream_machine_video_generation_driver.py
@@ -13,6 +13,7 @@
 
 if TYPE_CHECKING:
     from lumaai import LumaAI
+
 logger = logging.getLogger(__name__)
 
 

diff --git a/griptape/tools/base_video_generation_tool.py b/griptape/tools/base_video_generation_tool.py
@@ -0,0 +1,11 @@
+from attrs import define
+
+from griptape.mixins.artifact_file_output_mixin import ArtifactFileOutputMixin
+from griptape.tools import BaseTool
+
+
+@define
+class BaseVideoGenerationTool(ArtifactFileOutputMixin, BaseTool):
+    """A base class for tools that generate videos from text prompts."""
+
+    prompt_description = "Features and qualities to include in the generated video, descriptive and succinct."
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -64,14 +64,14 @@ diffusers = {version = ">=0.29.1,<0.31.0", optional = true}
 accelerate = {version = ">=0.32.1,<0.35.0", optional = true}
 sentencepiece = {version = "^0.2.0", optional = true}
 torch = {version = "^2.3.1", optional = true}
+lumaai = {version ="^1.0.2", optional = true}
 
 # loaders
 pandas = {version = "^1.3", optional = true}
 pypdf = {version = "^3.9", optional = true}
 pillow = {version = "^10.2.0", optional = true}
 mail-parser = {version = "^3.15.0", optional = true}
 filetype = {version = "^1.2", optional = true}
-lumaai = "^1.0.2"
 
 [tool.poetry.extras]
 drivers-prompt-cohere = ["cohere"]
@@ -150,6 +150,7 @@ drivers-image-generation-huggingface = [
     "torch",
     "pillow",
 ]
+drivers-video-generation-lumaai=["lumaai"]
 
 loaders-dataframe = ["pandas"]
 loaders-pdf = ["pypdf"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -13,6 +13,7 @@

		if TYPE_CHECKING:
		from lumaai import LumaAI

		logger = logging.getLogger(__name__)


Expand Down