model-compose 0.4.2__tar.gz → 0.4.4__tar.gz

{model_compose-0.4.2 → model_compose-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.4.2
+Version: 0.4.4
 Summary: model-compose: Declarative AI Model and Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT

{model_compose-0.4.2 → model_compose-0.4.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "model-compose"
-version = "0.4.2"
+version = "0.4.4"
 description = "model-compose: Declarative AI Model and Workflow Orchestrator"
 authors = [
     { name = "Hanyeol Cho", email = "hanyeol.cho@gmail.com" }

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/core/component/services/model/tasks/image_to_text.py

@@ -13,7 +13,7 @@ from threading import Thread
 import asyncio
 
 if TYPE_CHECKING:
-    from transformers import PreTrainedModel, PreTrainedTokenizer, ProcessorMixin, GenerationMixin, TextIteratorStreamer, StopStringCriteria
+    from transformers import PreTrainedModel, PreTrainedTokenizer, ProcessorMixin, GenerationMixin
     from torch import Tensor
     import torch
 
@@ -28,6 +28,7 @@ class ImageToTextTaskAction:
         self.device: torch.device = device
 
     async def run(self, context: ComponentActionContext, loop: asyncio.AbstractEventLoop) -> Any:
+        from transformers import TextIteratorStreamer, StopStringCriteria
         import torch
 
         image: Union[PILImage.Image, List[PILImage.Image]] = await context.render_image(self.config.image)

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/core/component/services/model/tasks/summarization.py

@@ -8,11 +8,11 @@ from mindor.core.utils.streamer import AsyncStreamer
 from mindor.core.logger import logging
 from ..base import ModelTaskService, ModelTaskType, register_model_task_service
 from ..base import ComponentActionContext
+from threading import Thread
 import asyncio
 
 if TYPE_CHECKING:
-    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin, TextIteratorStreamer, StopStringCriteria
-    from threading import Thread
+    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin
     from torch import Tensor
     import torch
 
@@ -24,6 +24,7 @@ class SummarizationTaskAction:
         self.device: torch.device = device
 
     async def run(self, context: ComponentActionContext, loop: asyncio.AbstractEventLoop) -> Any:
+        from transformers import TextIteratorStreamer, StopStringCriteria
         import torch
 
         text: Union[str, List[str]] = await context.render_variable(self.config.text)

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/core/component/services/model/tasks/text_generation.py

@@ -8,11 +8,11 @@ from mindor.core.utils.streamer import AsyncStreamer
 from mindor.core.logger import logging
 from ..base import ModelTaskService, ModelTaskType, register_model_task_service
 from ..base import ComponentActionContext
+from threading import Thread
 import asyncio
 
 if TYPE_CHECKING:
-    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin, TextIteratorStreamer, StopStringCriteria
-    from threading import Thread
+    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin
     from torch import Tensor
     import torch
 
@@ -24,6 +24,7 @@ class TextGenerationTaskAction:
         self.device: torch.device = device
 
     async def run(self, context: ComponentActionContext, loop: asyncio.AbstractEventLoop) -> Any:
+        from transformers import TextIteratorStreamer, StopStringCriteria
         import torch
 
         prompt: Union[str, List[str]] = await context.render_variable(self.config.prompt)

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/core/component/services/model/tasks/translation.py

@@ -8,11 +8,11 @@ from mindor.core.utils.streamer import AsyncStreamer
 from mindor.core.logger import logging
 from ..base import ModelTaskService, ModelTaskType, register_model_task_service
 from ..base import ComponentActionContext
+from threading import Thread
 import asyncio
 
 if TYPE_CHECKING:
-    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin, TextIteratorStreamer, StopStringCriteria
-    from threading import Thread
+    from transformers import PreTrainedModel, PreTrainedTokenizer, GenerationMixin
     from torch import Tensor
     import torch
 
@@ -24,6 +24,7 @@ class TranslationTaskAction:
         self.device: torch.device = device
 
     async def run(self, context: ComponentActionContext, loop: asyncio.AbstractEventLoop) -> Any:
+        from transformers import TextIteratorStreamer, StopStringCriteria
         import torch
 
         text: Union[str, List[str]] = await context.render_variable(self.config.text)

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/core/controller/runner/mcp_client.py

@@ -15,7 +15,7 @@ class McpControllerClient(ControllerClient):
         self.client: McpClient = McpClient(self._resolve_controller_url())
 
     async def run_workflow(self, workflow_id: Optional[str], input: Any, workflow: WorkflowSchema) -> Any:
-        return await self._call_tool(workflow_id, input, workflow)
+        return await self._call_tool(workflow_id or "__workflow__", input, workflow)
 
     async def close(self) -> None:
         await self.client.close()

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/dsl/schema/action/impl/mcp_client.py

@@ -4,6 +4,6 @@ from pydantic import model_validator
 from .common import CommonActionConfig
 
 class McpClientActionConfig(CommonActionConfig):
-    tool: str = Field(default="__tool__", description="Name of the tool to invoke.")
+    tool: str = Field(default="__workflow__", description="Name of the tool to invoke.")
     arguments: Dict[str, Any] = Field(default_factory=dict, description="Arguments to pass to the tool.")
     headers: Dict[str, str] = Field(default_factory=dict, description="Optional HTTP headers to include in the tool call.")

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/dsl/schema/action/impl/mcp_server.py

@@ -4,6 +4,6 @@ from pydantic import model_validator
 from .common import CommonActionConfig
 
 class McpServerActionConfig(CommonActionConfig):
-    tool: str = Field(default="__tool__", description="Name of the tool to invoke.")
+    tool: str = Field(default="__workflow__", description="Name of the tool to invoke.")
     arguments: Dict[str, Any] = Field(default_factory=dict, description="Arguments to pass to the tool.")
     headers: Dict[str, str] = Field(default_factory=dict, description="Optional HTTP headers to include in the tool call.")

{model_compose-0.4.2 → model_compose-0.4.4}/src/mindor/dsl/schema/component/impl/mcp_server.py

@@ -34,6 +34,7 @@ class McpServerComponentConfig(CommonComponentConfig):
     manage: McpServerManageConfig = Field(default_factory=McpServerManageConfig, description="Configuration used to manage the MCP server lifecycle.")
     port: int = Field(default=8000, ge=1, le=65535, description="Port on which the MCP server will listen for incoming requests.")
     base_path: Optional[str] = Field(default=None, description="Base path to prefix all MCP routes exposed by this component.")
+    headers: Dict[str, Any] = Field(default_factory=dict, description="Headers to be included in all outgoing requests.")
     actions: List[McpServerActionConfig] = Field(default_factory=list)
 
     @model_validator(mode="before")

{model_compose-0.4.2 → model_compose-0.4.4}/src/model_compose.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.4.2
+Version: 0.4.4
 Summary: model-compose: Declarative AI Model and Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT