From 3fd4d20181ae5a6844aa9df1c5075155e0ebfa94 Mon Sep 17 00:00:00 2001 From: Collin Dutter Date: Mon, 6 Jan 2025 11:07:36 -0800 Subject: [PATCH] PR cleanup --- griptape/tasks/prompt_task.py | 3 +-- griptape/tools/structured_output/tool.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/griptape/tasks/prompt_task.py b/griptape/tasks/prompt_task.py index b70b1eac7..d2de27ac6 100644 --- a/griptape/tasks/prompt_task.py +++ b/griptape/tasks/prompt_task.py @@ -7,8 +7,7 @@ from attrs import NOTHING, Attribute, Factory, NothingType, define, field from griptape import utils -from griptape.artifacts import ActionArtifact, BaseArtifact, ErrorArtifact, ListArtifact, TextArtifact -from griptape.artifacts.json_artifact import JsonArtifact +from griptape.artifacts import ActionArtifact, BaseArtifact, ErrorArtifact, JsonArtifact, ListArtifact, TextArtifact from griptape.common import PromptStack, ToolAction from griptape.configs import Defaults from griptape.memory.structure import Run diff --git a/griptape/tools/structured_output/tool.py b/griptape/tools/structured_output/tool.py index 89e638f59..54ef5a9c6 100644 --- a/griptape/tools/structured_output/tool.py +++ b/griptape/tools/structured_output/tool.py @@ -13,7 +13,7 @@ class StructuredOutputTool(BaseTool): @activity( config={ "description": "Used to provide the final response which ends this conversation.", - "schema": lambda self: self.output_schema, + "schema": lambda _self: _self.output_schema, } ) def provide_output(self, params: dict) -> BaseArtifact: