fix: move converters to new definition

BIGmama-technology · Mar 9, 2024 · 10551bc · 10551bc
1 parent 51e33f7
commit 10551bc
Show file tree

Hide file tree

Showing 7 changed files with 22 additions and 12 deletions.
diff --git a/hyko_toolkit/functions/converters/markdown_to_pdf/main.py b/hyko_toolkit/functions/converters/markdown_to_pdf/main.py
@@ -10,7 +10,7 @@
 from pygments.util import ClassNotFound
 
 from hyko_sdk.io import PDF
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 def format_code_block(code: str, language: str):

diff --git a/hyko_toolkit/functions/converters/markdown_to_pdf/metadata.py b/hyko_toolkit/functions/converters/markdown_to_pdf/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.io import PDF
 from hyko_sdk.models import CoreModel
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="markdown_to_pdf",
+    task="converters",
     description="Convert Markdown content to PDF format.",
 )
 

diff --git a/hyko_toolkit/functions/converters/ocr_pdf_to_text/metadata.py b/hyko_toolkit/functions/converters/ocr_pdf_to_text/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.io import PDF
 from hyko_sdk.models import CoreModel
 
@@ -13,7 +13,9 @@ class SupportedLanguages(str, Enum):
     french = "fra"
 
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="ocr_pdf_to_text",
+    task="converters",
     description="Perform OCR (Optical Character Recognition) on a PDF document",
 )
 

diff --git a/hyko_toolkit/functions/converters/pdf_to_text/metadata.py b/hyko_toolkit/functions/converters/pdf_to_text/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.io import PDF
 from hyko_sdk.models import CoreModel
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="pdf_to_text",
+    task="converters",
     description="Extracts text from pdf.",
 )
 

diff --git a/hyko_toolkit/functions/converters/video_converter/metadata.py b/hyko_toolkit/functions/converters/video_converter/metadata.py
@@ -2,11 +2,13 @@
 
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.io import Video
 from hyko_sdk.models import CoreModel, Ext
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="video_converter",
+    task="converters",
     description="Convert a video from one format to another.",
 )
 

diff --git a/hyko_toolkit/functions/converters/video_to_audio/main.py b/hyko_toolkit/functions/converters/video_to_audio/main.py
@@ -4,7 +4,7 @@
 from metadata import Inputs, Outputs, Params, func
 
 from hyko_sdk.io import Audio
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 @func.on_execute

diff --git a/hyko_toolkit/functions/converters/video_to_audio/metadata.py b/hyko_toolkit/functions/converters/video_to_audio/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.io import Audio, Video
 from hyko_sdk.models import CoreModel
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="video_to_audio",
+    task="converters",
     description="Convert a video type to audio type (takes only the audio data)",
 )