model-compose 0.2.2__tar.gz → 0.2.4__tar.gz

{model_compose-0.2.2/src/model_compose.egg-info → model_compose-0.2.4}/PKG-INFO

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

{model_compose-0.2.2 → model_compose-0.2.4}/pyproject.toml

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

{model_compose-0.2.2 → model_compose-0.2.4}/src/mindor/core/component/engine/http_client.py

@@ -38,9 +38,6 @@ class HttpClientPollingCompletion(HttpClientCompletion):
             context.register_source("result", response)
 
             status = (await context.render_variable(self.config.status)) if self.config.status else None
-            if self.config.status and not status:
-                raise RuntimeError(f"Polling failed: no status found in response.")
-
             if status:
                 if status in self.config.success_when or []:
                     return response
@@ -63,7 +60,7 @@ class HttpClientPollingCompletion(HttpClientCompletion):
         return await context.render_variable(self.config.endpoint)
 
 class HttpClientCallbackCompletion(HttpClientCompletion):
-    async def run(self, context: ComponentActionContext) -> Any:
+    async def run(self, context: ComponentActionContext, client: HttpClient) -> Any:
         callback_id = await context.render_variable(self.config.wait_for)
         future: asyncio.Future = asyncio.get_event_loop().create_future()
         HttpCallbackListener.register_pending_future(callback_id, future)

{model_compose-0.2.2 → model_compose-0.2.4}/src/mindor/core/utils/http_client.py

@@ -134,5 +134,5 @@ class HttpClient:
 
         return (HttpStreamResource(response, content_type, filename), content_type)
 
-async def create_stream_with_url(url: str):
+async def create_stream_with_url(url: str) -> HttpStreamResource:
     return await HttpClient.get_shared_instance().request(url)

{model_compose-0.2.2 → model_compose-0.2.4}/src/mindor/core/utils/renderer.py

@@ -1,5 +1,5 @@
 from typing import Callable, Dict, List, Optional, Awaitable, Any
-from .streaming import StreamResource, Base64StreamResource
+from .streaming import StreamResource, UploadFileStreamResource, Base64StreamResource
 from .streaming import encode_stream_to_base64, save_stream_to_temporary_file
 from .http_request import create_upload_file
 from .http_client import create_stream_with_url
@@ -110,6 +110,8 @@ class VariableRenderer:
             return []
 
         if type == "base64":
+            if isinstance(value, UploadFile):
+                return await encode_stream_to_base64(UploadFileStreamResource(value))
             if isinstance(value, StreamResource):
                 return await encode_stream_to_base64(value)
             return base64.b64encode(value)

{model_compose-0.2.2 → model_compose-0.2.4}/src/mindor/core/utils/streaming.py

@@ -1,7 +1,7 @@
 from typing import Optional, AsyncIterator
 from abc import ABC, abstractmethod
 from tempfile import NamedTemporaryFile
-from aiofiles.threadpool.binary import AsyncBufferedReader
+from starlette.datastructures import UploadFile
 import aiofiles, os, io, base64
 
 class StreamResource(ABC):
@@ -49,6 +49,25 @@ class FileStreamResource(StreamResource):
                 break
             yield chunk
 
+class UploadFileStreamResource(StreamResource):
+    def __init__(self, file: UploadFile):
+        super().__init__(file.content_type, file.filename)
+
+        self.file: UploadFile = file
+
+    async def __aenter__(self):
+        return self
+
+    async def close(self) -> None:
+        await self.file.close()
+
+    async def _iterate_stream(self) -> AsyncIterator[bytes]:
+        while True:
+            chunk = await self.file.read(8192)
+            if not chunk:
+                break
+            yield chunk
+
 class Base64StreamResource(StreamResource):
     def __init__(self, encoded: str, content_type: Optional[str] = None, filename: Optional[str] = None):
         super().__init__(content_type, filename)

{model_compose-0.2.2 → model_compose-0.2.4/src/model_compose.egg-info}/PKG-INFO

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