feat: take qstash url from environment (#25)

upstash · Jan 21, 2025 · 87b45cf · 87b45cf
1 parent 560656e
commit 87b45cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/upstash_workflow/asyncio/serve/options.py b/upstash_workflow/asyncio/serve/options.py
@@ -73,6 +73,7 @@ def _initial_payload_parser(initial_request: str) -> TInitialPayload:
         qstash_client=qstash_client
         or AsyncQStash(
             cast(str, environment.get("QSTASH_TOKEN")),
+            base_url=(environment.get("QSTASH_URL")),
         ),
         on_step_finish=on_step_finish or _on_step_finish,
         initial_payload_parser=initial_payload_parser or _initial_payload_parser,

diff --git a/upstash_workflow/serve/options.py b/upstash_workflow/serve/options.py
@@ -87,6 +87,7 @@ def _initial_payload_parser(initial_request: str) -> TInitialPayload:
         qstash_client=qstash_client
         or QStash(
             cast(str, environment.get("QSTASH_TOKEN")),
+            base_url=(environment.get("QSTASH_URL")),
         ),
         on_step_finish=on_step_finish or _on_step_finish,
         initial_payload_parser=initial_payload_parser or _initial_payload_parser,