sinapsis-huggingface 0.2.1__tar.gz → 0.2.3__tar.gz

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-huggingface
-Version: 0.2.1
+Version: 0.2.3
 Summary: Package for HuggingFace-based templates
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-huggingface
-Version: 0.2.1
+Version: 0.2.3
 Summary: Package for HuggingFace-based templates
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_diffusers/src/sinapsis_huggingface_diffusers/templates/base_diffusers.py

@@ -11,8 +11,10 @@ from pydantic import BaseModel, ConfigDict
 from sinapsis_core.data_containers.data_packet import DataContainer, ImagePacket
 from sinapsis_core.template_base import Template
 from sinapsis_core.template_base.base_models import (
+    OutputTypes,
     TemplateAttributes,
     TemplateAttributeType,
+    UIPropertiesMetadata,
 )
 from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 
@@ -64,7 +66,7 @@ class BaseDiffusers(Template, ABC):
     """
 
     AttributesBaseModel = BaseDiffusersAttributes
-    CATEGORY = "Diffusers"
+    UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
     TORCH_DTYPE: dict = TorchTypes().model_dump()
 
     def __init__(self, attributes: TemplateAttributeType) -> None:

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/hugging_face_embedding_extractor.py

@@ -6,8 +6,10 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base import Template
 from sinapsis_core.template_base.base_models import (
+    OutputTypes,
     TemplateAttributes,
     TemplateAttributeType,
+    UIPropertiesMetadata,
 )
 
 
@@ -44,7 +46,7 @@ class HuggingFaceEmbeddingExtractor(Template):
     """
 
     AttributesBaseModel = HuggingFaceEmbeddingExtractorAttributes
-    CATEGORY = "Embeddings"
+    UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.TEXT)
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_embeddings/src/sinapsis_huggingface_embeddings/templates/speaker_embedding_from_dataset.py

@@ -5,7 +5,7 @@ from typing import Literal
 from datasets import load_dataset
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base import Template
-from sinapsis_core.template_base.base_models import TemplateAttributes
+from sinapsis_core.template_base.base_models import OutputTypes, TemplateAttributes, UIPropertiesMetadata
 from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 
 
@@ -63,7 +63,7 @@ class SpeakerEmbeddingFromDataset(Template):
     """
 
     AttributesBaseModel = SpeakerEmbeddingFromDatasetAttributes
-    CATEGORY = "Embeddings"
+    UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.AUDIO)
 
     def execute(self, container: DataContainer) -> DataContainer:
         """Retrieve and attach speaker embeddings to specified packets in a DataContainer.

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_grounding_dino/src/sinapsis_huggingface_grounding_dino/templates/grounding_dino.py

@@ -7,7 +7,12 @@ import torch
 from sinapsis_core.data_containers.annotations import BoundingBox, ImageAnnotations
 from sinapsis_core.data_containers.data_packet import DataContainer, ImagePacket
 from sinapsis_core.template_base import Template
-from sinapsis_core.template_base.base_models import TemplateAttributes, TemplateAttributeType
+from sinapsis_core.template_base.base_models import (
+    OutputTypes,
+    TemplateAttributes,
+    TemplateAttributeType,
+    UIPropertiesMetadata,
+)
 from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 from transformers import (
     AutoModelForZeroShotObjectDetection,
@@ -75,7 +80,7 @@ class GroundingDINO(Template):
 
     """
 
-    CATEGORY = "Grounding DINO"
+    UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
     KEYS = GroundingDINOKeys()
 
     class AttributesBaseModel(GroundingBaseAttributes):

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/base_transformers.py

@@ -8,10 +8,7 @@ import torch
 from pydantic import Field
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base import Template
-from sinapsis_core.template_base.base_models import (
-    TemplateAttributes,
-    TemplateAttributeType,
-)
+from sinapsis_core.template_base.base_models import TemplateAttributes, TemplateAttributeType, UIPropertiesMetadata
 from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 from transformers import AutoProcessor, pipeline
 from transformers.pipelines import Pipeline
@@ -56,7 +53,7 @@ class TransformersBase(Template):
     """
 
     AttributesBaseModel = TransformersBaseAttributes
-    CATEGORY = "Transformers"
+    UIProperties = UIPropertiesMetadata(category="Transformers")
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/packages/sinapsis_huggingface_transformers/src/sinapsis_huggingface_transformers/templates/pali_gemma/pali_gemma_base.py

@@ -6,8 +6,10 @@ import torch
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base import Template
 from sinapsis_core.template_base.base_models import (
+    OutputTypes,
     TemplateAttributes,
     TemplateAttributeType,
+    UIPropertiesMetadata,
 )
 from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
@@ -41,7 +43,7 @@ class PaliGemmaBase(Template):
     """Base class for PaliGemma implementations."""
 
     AttributesBaseModel = PaliGemmaBaseAttributes
-    CATEGORY = "Transformers"
+    UIProperties = UIPropertiesMetadata(category="HuggingFace", output_type=OutputTypes.IMAGE)
     _TORCH_DTYPE: ClassVar[dict[str, Any]] = {"float16": torch.float16, "float32": torch.float32}
 
     def __init__(self, attributes: TemplateAttributeType) -> None:

{sinapsis_huggingface-0.2.1 → sinapsis_huggingface-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-huggingface"
-version = "0.2.1"
+version = "0.2.3"
 description = "Package for HuggingFace-based templates"
 authors = [
     {name = "SinapsisAI", email = "dev@sinapsis.tech"},