streamlit-octostar-utils 0.6.3.dev1__tar.gz → 0.6.4__tar.gz

{streamlit_octostar_utils-0.6.3.dev1 → streamlit_octostar_utils-0.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.6.3.dev1
+Version: 0.6.4
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.6.3.dev1 → streamlit_octostar_utils-0.6.4}/pyproject.toml

@@ -5,7 +5,7 @@ include = '\.pyi?$'
 
 [tool.poetry]
 name = "streamlit-octostar-utils"
-version = "0.6.3-dev.1"
+version = "0.6.4"
 description = ""
 license = "MIT"
 authors = ["Octostar"]

{streamlit_octostar_utils-0.6.3.dev1 → streamlit_octostar_utils-0.6.4}/streamlit_octostar_utils/api_crafter/celery.py

@@ -352,6 +352,7 @@ class CeleryQueueConfig:
         max_tasks_per_child=None,
         max_memory_per_child=None,
         warn_memory_per_child=None,
+        worker_env=None,
         **options,
     ):
         self.n_workers = n_workers
@@ -359,6 +360,7 @@ class CeleryQueueConfig:
         self.max_tasks_per_child = max_tasks_per_child
         self.max_memory_per_child = max_memory_per_child  # KiB — hard limit, worker restarted
         self.warn_memory_per_child = warn_memory_per_child  # KiB — soft limit, logs WARNING
+        self.worker_env = dict(worker_env) if worker_env else {}
         self.options = options
 
 
@@ -800,6 +802,10 @@ class CeleryExecutor(object):
                     f"PYTHONPATH={os.getcwd()}",
                     f"CELERY_WORKER_QUEUES={queue}",
                     f"CELERY_WORKER_NAME={worker_name}",
+                ]
+                for env_key, env_val in queue_config.worker_env.items():
+                    command.append(f"{env_key}={env_val}")
+                command.extend([
                     "celery",
                     "--app",
                     f"{self.filename}.{self.name}",
@@ -808,7 +814,7 @@ class CeleryExecutor(object):
                     "--concurrency=1",
                     f"--loglevel={self.loglevel}",
                     f"-n {worker_name}",
-                ]
+                ])
                 if queue_config.max_tasks_per_child is not None:
                     command.append(f"--max-tasks-per-child={queue_config.max_tasks_per_child}")
                 if queue_config.max_memory_per_child is not None: