updated tool for proper docstring

This commit is contained in:
Jason Schleifer
2024-11-16 12:39:49 +13:00
parent 714a820f3e
commit 741d4020a4
+3 -2
View File
@@ -10,7 +10,7 @@ extraction_engines = ["csv", "json"]
class gtUIExtractionTool(gtUIBaseTool):
"""
The Griptape Prompt Summary Tool
The Griptape Text Extraction Tool
"""
@classmethod
@@ -49,7 +49,8 @@ class gtUIExtractionTool(gtUIBaseTool):
def create(self, **kwargs):
prompt_driver = kwargs.get("prompt_driver", None)
extraction_type = kwargs.get("extraction_type", "json")
column_names = kwargs.get("column_names", "")
column_names_string = kwargs.get("column_names", "")
column_names = [column_name.strip() for column_name in column_names_string]
template_schema = kwargs.get("template_schema", "")
params = {}