fix: move to new definition

BIGmama-technology · Mar 9, 2024 · 0c8e25c · 0c8e25c
1 parent 0d199ad
commit 0c8e25c
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 8 deletions.
diff --git a/hyko_toolkit/functions/web/web_scraping/beautifulsoup_transformer/metadata.py b/hyko_toolkit/functions/web/web_scraping/beautifulsoup_transformer/metadata.py
@@ -1,10 +1,12 @@
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.models import CoreModel
 
-func = SDKFunction(
-    description="Scrape HTML content from URLs and convert it to plain text"
+func = ToolkitFunction(
+    name="beautifulsoup_transformer",
+    task="web_scraping",
+    description="Scrape HTML content from URLs and convert it to plain text",
 )
 
 

diff --git a/hyko_toolkit/functions/web/web_scraping/html2text_transformer/metadata.py b/hyko_toolkit/functions/web/web_scraping/html2text_transformer/metadata.py
@@ -1,10 +1,12 @@
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.models import CoreModel
 
-func = SDKFunction(
-    description="Scrape HTML content from URLs and convert it to plain text"
+func = ToolkitFunction(
+    name="html2text_transformer",
+    task="web_scraping",
+    description="Scrape HTML content from URLs and convert it to plain text",
 )
 
 

diff --git a/hyko_toolkit/functions/web/web_scraping/youtube_transcript/metadata.py b/hyko_toolkit/functions/web/web_scraping/youtube_transcript/metadata.py
@@ -2,7 +2,7 @@
 
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.models import CoreModel
 
 
@@ -12,7 +12,11 @@ class SupportedLanguages(str, Enum):
     french = "fr"
 
 
-func = SDKFunction(description="Transcript extraction from youtube video.")
+func = ToolkitFunction(
+    name="youtube_transcript",
+    task="web_scraping",
+    description="Transcript extraction from youtube video.",
+)
 
 
 @func.set_input