sunholo 0.145.1__py3-none-any.whl → 0.145.3__py3-none-any.whl

sunholo/invoke/async_task_runner.py

@@ -96,6 +96,13 @@ class AsyncTaskRunner:
             >>> results = await runner.get_aggregated_results()
             >>> print(results['results'])  # {'fetch_data': 'data from api'}
             
+            # Custom task names for better clarity
+            >>> runner = AsyncTaskRunner()
+            >>> runner.add_task(fetch_data, "user_api", task_name="fetch_user_data")
+            >>> runner.add_task(fetch_data, "posts_api", task_name="fetch_posts")
+            >>> results = await runner.get_aggregated_results()
+            >>> print(results['results']['fetch_user_data'])  # User data
+            
             # Silent mode - no console output but still collects results
             >>> runner = AsyncTaskRunner(verbose=False)
             
@@ -108,6 +115,7 @@ class AsyncTaskRunner:
             >>> runner = AsyncTaskRunner(use_default_callbacks=False)
         """
         self.tasks = []
+        self.task_name_counts = {}  # Track task names to ensure uniqueness
         self.retry_enabled = retry_enabled
         self.retry_kwargs = retry_kwargs or {}
         self.timeout = timeout
@@ -209,18 +217,35 @@ class AsyncTaskRunner:
                  func: Callable[..., Any], 
                  *args: Any, 
                  task_config: Optional[TaskConfig] = None,
+                 task_name: Optional[str] = None,
                  **kwargs: Any):
         """
         Adds a task to the list of tasks to be executed, with optional per-task configuration.
+        
+        Automatically ensures task names are unique by appending a suffix if needed.
 
         Args:
             func: The function to be executed.
             *args: Positional arguments for the function.
             task_config: Optional per-task configuration for timeout, retry, and callbacks.
+            task_name: Optional custom name for the task. If not provided, uses func.__name__.
             **kwargs: Keyword arguments for the function.
         """
-        log.info(f"Adding task: {func.__name__} with args: {args}, kwargs: {kwargs}, config: {task_config}")
-        self.tasks.append((func.__name__, func, args, kwargs, task_config))
+        # Get base name from task_name or function name
+        base_name = task_name if task_name is not None else func.__name__
+        
+        # Ensure uniqueness by adding suffix if needed
+        if base_name in self.task_name_counts:
+            # Name already exists, increment count and add suffix
+            self.task_name_counts[base_name] += 1
+            name = f"{base_name}_{self.task_name_counts[base_name]}"
+        else:
+            # First occurrence of this name
+            self.task_name_counts[base_name] = 0
+            name = base_name
+        
+        log.info(f"Adding task: {name} with args: {args}, kwargs: {kwargs}, config: {task_config}")
+        self.tasks.append((name, func, args, kwargs, task_config))
 
     async def run_async_with_callbacks(self) -> AsyncGenerator[CallbackContext, None]:
         """
@@ -313,10 +338,10 @@ class AsyncTaskRunner:
         
         Example:
             >>> runner = AsyncTaskRunner()
-            >>> runner.add_task(fetch_data, "api")
-            >>> runner.add_task(process_data, "raw_data")
+            >>> runner.add_task(fetch_data, "api", task_name="api_fetch")
+            >>> runner.add_task(process_data, "raw_data", task_name="data_processing")
             >>> results = await runner.get_aggregated_results()
-            >>> print(results['results']['fetch_data'])  # Access specific result
+            >>> print(results['results']['api_fetch'])  # Access specific result
             >>> if results['errors']:  # Check for any errors
             ...     print(f"Errors occurred: {results['errors']}")
         """

{sunholo-0.145.1.dist-info → sunholo-0.145.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.145.1
+Version: 0.145.3
 Summary: AI DevOps - a package to help deploy GenAI to the Cloud.
 Author-email: Holosun ApS <multivac@sunholo.com>
 License: Apache License, Version 2.0

{sunholo-0.145.1.dist-info → sunholo-0.145.3.dist-info}/RECORD

@@ -102,7 +102,7 @@ sunholo/genai/process_funcs_cls.py,sha256=D6eNrc3vtTZzwdkacZNOSfit499N_o0C5AHspy
 sunholo/genai/safety.py,sha256=mkFDO_BeEgiKjQd9o2I4UxB6XI7a9U-oOFjZ8LGRUC4,1238
 sunholo/invoke/__init__.py,sha256=BNyovdgKPt2Zuq3cw0VmxWGzYCPF5pfgqMn9SlLjGAU,219
 sunholo/invoke/async_class.py,sha256=ZMzxKQtelbYibu9Fac7P9OU3GorH8KxawZxSMv5EO9A,12514
-sunholo/invoke/async_task_runner.py,sha256=8FjZMKAxFkb6qr0EZlSI2Likc2B7BRNUggf9jebfa9c,30407
+sunholo/invoke/async_task_runner.py,sha256=uv2v6HVvM294hVRCNiy2NXsZOkX4wcxgjIBgQA6XpIQ,31738
 sunholo/invoke/direct_vac_func.py,sha256=dACx3Zh7uZnuWLIFYiyLoyXUhh5-eUpd2RatDUd9ov8,9753
 sunholo/invoke/invoke_vac_utils.py,sha256=sJc1edHTHMzMGXjji1N67c3iUaP7BmAL5nj82Qof63M,2053
 sunholo/langfuse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -183,9 +183,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.145.1.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.145.1.dist-info/METADATA,sha256=KiLh38dP1mlkut1F1fxus_1FA1r5vxfTCACZIHPEfks,18639
-sunholo-0.145.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sunholo-0.145.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.145.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.145.1.dist-info/RECORD,,
+sunholo-0.145.3.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.145.3.dist-info/METADATA,sha256=X1gQCYnMVaouSX9h30QgR9WV6tmyVad93pscdRn49xQ,18639
+sunholo-0.145.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sunholo-0.145.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.145.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.145.3.dist-info/RECORD,,