render_sdk 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

render_sdk/__init__.py

@@ -5,4 +5,4 @@ This package provides:
 2. REST API Client (render_sdk.client) for interacting with Render's API
 """
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"

render_sdk/client/client.py

@@ -40,6 +40,11 @@ class Client:
         # Set default values
         if token is None:
             self.token = os.getenv("RENDER_API_KEY", "")
+            if self.token == "":
+                raise ValueError(
+                    "Either provide a token or set the RENDER_API_KEY "
+                    + "environment variable"
+                )
         else:
             self.token = token
 

render_sdk/workflows/executor.py

@@ -44,7 +44,7 @@ class TaskExecutor:
 
     async def execute(self, task_name: str, input_args: list[Any]) -> Any:
         """Execute a task by name with the given input."""
-        logger.info(f"Starting execution of task: {task_name}")
+        logger.debug(f"Starting execution of task: {task_name}")
 
         sent_error = False
 

render_sdk/workflows/runner.py

@@ -25,18 +25,16 @@ async def run_async(socket_path: str) -> None:
 
     It gets the input from the server, executes the task, and sends the result back.
     """
-    logger.info("Starting task runner")
+    logger.debug("Starting task runner")
 
     # Create client
     client = UDSClient(socket_path)
 
     try:
         # Get input from server
-        logger.info("Getting input from server")
+        logger.debug("Getting task input")
         input_response = await client.get_input()
 
-        logger.info(f"Input response: {input_response}")
-
         task_name = input_response.task_name
         raw_input = input_response.input_
 
@@ -50,18 +48,13 @@ async def run_async(socket_path: str) -> None:
         else:
             input_data = []
 
-        logger.info(f"Received input - task: {task_name}, input: {input_data}")
-
         # Create executor and execute task
         task_registry = get_task_registry()
         executor = TaskExecutor(task_registry, client)
 
-        result = await executor.execute(task_name, input_data)
-
-        logger.info(f"Task executed successfully with result: {result}")
-
-    except Exception as e:
-        logger.error(f"Task execution failed: {e}")
+        logger.debug(f"Executing task: {task_name}")
+        await executor.execute(task_name, input_data)
+    except Exception:
         raise
 
 
@@ -76,7 +69,7 @@ async def register_async(socket_path: str) -> None:
     """
     Register all tasks with the server asynchronously.
     """
-    logger.info("Registering tasks")
+    logger.debug("Registering tasks")
 
     # Create client
     client = UDSClient(socket_path)
@@ -105,10 +98,10 @@ async def register_async(socket_path: str) -> None:
             tasks.append(task_def)
 
         # Register tasks with server
-        logger.info(f"Registering {len(tasks)} tasks: {[t.name for t in tasks]}")
+        logger.debug(f"Registering {len(tasks)} tasks: {[t.name for t in tasks]}")
         await client.register_tasks(Tasks(tasks=tasks))
 
-        logger.info("Tasks registered successfully")
+        logger.debug("Tasks registered successfully")
 
     except Exception as e:
         logger.error(f"Task registration failed: {e}")
@@ -139,8 +132,6 @@ def start() -> None:
     if not socket_path:
         raise ValueError("RENDER_SDK_SOCKET_PATH environment variable is required")
 
-    logger.info(f"Starting in mode: {mode}")
-
     if mode == "run":
         run(socket_path)
     elif mode == "register":

render_sdk/workflows/task.py

@@ -1,5 +1,6 @@
 """Task decorator and related functionality."""
 
+import asyncio
 import contextvars
 import functools
 from abc import ABC, abstractmethod
@@ -104,28 +105,19 @@ class TaskRegistry:
         return task_info.func
 
 
-class TaskCallable:
-    """A callable that can be awaited to run as a subtask."""
+class TaskInstance:
+    """Represents a single task execution that can be awaited."""
 
-    def __init__(self, func, name):
-        self._func = func
+    def __init__(self, name: str, future: asyncio.Task):
         self._name = name
-        # Copy function attributes for introspection
-        functools.update_wrapper(self, func)
-
-    def __call__(self, *args, **kwargs):
-        # Store args for potential await
-        self._args = args
-        self._kwargs = kwargs
-        return self
+        self._future = future
 
     def __await__(self):
-        """Run as a subtask when awaited."""
+        """Await the task execution."""
 
         async def run_subtask():
             try:
-                client = _current_client.get()
-                return await client.run_subtask(self._name, list(self._args))
+                return await self._future
             except LookupError as e:
                 raise RuntimeError(
                     f"Cannot run {self._name} as subtask \
@@ -135,6 +127,23 @@ class TaskCallable:
         return run_subtask().__await__()
 
 
+class TaskCallable:
+    """A callable that can be awaited to run as a subtask."""
+
+    def __init__(self, func, name):
+        self._func = func
+        self._name = name
+        # Copy function attributes for introspection
+        functools.update_wrapper(self, func)
+
+    def __call__(self, *args, **kwargs):
+        # Create a new TaskInstance for each call
+        client = _current_client.get()
+        # Start execution immediately
+        future = asyncio.create_task(client.run_subtask(self._name, list(args)))
+        return TaskInstance(self._name, future)
+
+
 def create_task_decorator(registry: TaskRegistry) -> Callable:
     """
     Create a task decorator bound to a specific registry.

{render_sdk-0.1.0.dist-info → render_sdk-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: render_sdk
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python SDK for Render Workflows
 Author: Render
 Author-email: support@render.com

{render_sdk-0.1.0.dist-info → render_sdk-0.1.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
-render_sdk/__init__.py,sha256=Fl_M6R2YTiaSK_ki8zNklB6veGHkwQlXfH-LdCAtj48,215
+render_sdk/__init__.py,sha256=fXivApwYZa4k0p3j5j4EbyfHIx2wzlWmYfxmrxszeJA,215
 render_sdk/client/__init__.py,sha256=OtkFIaTnPhU6_1yveJwZFRzL3TemdEjZcRvtoYCVhTE,630
-render_sdk/client/client.py,sha256=tdSSyxSvUqaOCnm8fdP87xef7soCbQ5aeE-VF6y70Yo,2284
+render_sdk/client/client.py,sha256=Relr4rc8qVbEkROBom6VvACKA8AalTZkYby3DPc2-vE,2486
 render_sdk/client/errors.py,sha256=Y1WZV_AOpqvf0_HT5XBQ1JD5jNiEjHrgSiyq8wBfM5U,758
 render_sdk/client/sse.py,sha256=Y-3uWEjfHwfl2g17L6uW2ox5sW1cQKmSojjIVWXuCVA,4419
 render_sdk/client/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -561,14 +561,14 @@ render_sdk/workflows/callback_api/models/tasks.py,sha256=mb3M6wKIb47y3tzl2uompmt
 render_sdk/workflows/callback_api/py.typed,sha256=8PjyZ1aVoQpRVvt71muvuq5qE-jTFZkK-GLHkhdebmc,26
 render_sdk/workflows/callback_api/types.py,sha256=AX4orxQZQJat3vZrgjJ-TYb2sNBL8kNo9yqYDT-n8y8,1391
 render_sdk/workflows/client.py,sha256=P-2RRJydCrQopiVrms1Qoi_mS2z4L1HjDELgjgyZNSI,8448
-render_sdk/workflows/executor.py,sha256=3EevvPeTbYGqmjcJztZUzq85-7EHGErwK91ChnRZj54,3027
-render_sdk/workflows/runner.py,sha256=vJV1vK6sW1zffmCSEBIb1Pg_HwngIZD3TUC1CRoHB6w,4230
-render_sdk/workflows/task.py,sha256=iicK96R40cQ7XUiJh7sBVqnyTqBrNQ1SbAfNb5j7CVQ,5047
+render_sdk/workflows/executor.py,sha256=whdE_q2plZsTt5da-9C5c6Ho7PohVVg2_d7XQy9xQ0c,3028
+render_sdk/workflows/runner.py,sha256=hGeoACuITT96zP8PVfygUdB80T_TtBpUZBbf9yND5r0,3955
+render_sdk/workflows/task.py,sha256=Wh742_vC5JxvY8ifb8cWzefeEioGwgUGjB_bbplUGSA,5315
 render_sdk/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 render_sdk/workflows/tests/test_end_to_end.py,sha256=eEasR0oCcZWgxZC8H0566I3HP501uK4nzawoEoRVFso,5183
 render_sdk/workflows/tests/test_executor.py,sha256=8l5RtrAJKB3ZiWUr7aE6tVWJgCWWEHfadPvtqc3uvzs,4685
 render_sdk/workflows/tests/test_registration.py,sha256=l0_xUnknCEaE-oO55HYlsbaXQIe9KfUo8DbmJQ7fssw,4004
-render_sdk-0.1.0.dist-info/LICENSE,sha256=YQ01omJ4YGRAeEslBDXIx9tgO1VbIFYsusdCjQwjwW4,11336
-render_sdk-0.1.0.dist-info/METADATA,sha256=PRAlkpANkeWdlfxomH1E1q1tlPF9QflG3FbgbGqWGSk,2707
-render_sdk-0.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-render_sdk-0.1.0.dist-info/RECORD,,
+render_sdk-0.1.2.dist-info/LICENSE,sha256=YQ01omJ4YGRAeEslBDXIx9tgO1VbIFYsusdCjQwjwW4,11336
+render_sdk-0.1.2.dist-info/METADATA,sha256=F1SxdFvdcP1LZ-3oe-Nnr51NZca0muPM9LiyVyEKkKM,2707
+render_sdk-0.1.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+render_sdk-0.1.2.dist-info/RECORD,,