take care of RUN_BY_PROCESSOR

khvn26 · khvn26 · commit c0ae170af89a · 2025-03-27T13:44:17.000Z
diff --git a/src/common/core/main.py b/src/common/core/main.py
@@ -41,6 +41,10 @@ def ensure_cli_env() -> typing.Generator[None, None, None]:
         )
         os.environ["PROMETHEUS_MULTIPROC_DIR"] = prometheus_multiproc_dir_name
 
+    if "task-processor" in sys.argv:
+        # A bit hacky way to signal we're not running the API
+        os.environ["RUN_BY_PROCESSOR"] = "true"
+
     with ctx:
         yield
 

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,10 @@ def ensure_cli_env() -> typing.Generator[None, None, None]:`
`41`	`41`	`)`
`42`	`42`	`os.environ["PROMETHEUS_MULTIPROC_DIR"] = prometheus_multiproc_dir_name`
`43`	`43`
	`44`	`+ if "task-processor" in sys.argv:`
	`45`	`+ # A bit hacky way to signal we're not running the API`
	`46`	`+ os.environ["RUN_BY_PROCESSOR"] = "true"`
	`47`	`+`
`44`	`48`	`with ctx:`
`45`	`49`	`yield`
`46`	`50`