fix: retries option in serve (#31)

Author: yunusemreozdemir

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "upstash-workflow"
-version = "0.1.0"
+version = "0.1.1"
 description = "Python SDK for Upstash Workflow"
 license = "MIT"
 authors = ["Upstash <[email protected]>"]

upstash_workflow/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 from upstash_workflow.context.context import WorkflowContext
 from upstash_workflow.serve.serve import serve

upstash_workflow/asyncio/context/context.py

@@ -58,7 +58,7 @@ def __init__(
         self.headers: Dict[str, str] = headers
         self.request_payload: TInitialPayload = initial_payload
         self.env: Dict[str, Optional[str]] = env or {}
-        self.retries: int = retries or DEFAULT_RETRIES
+        self.retries: int = DEFAULT_RETRIES if retries is None else retries
         self._executor: _AutoExecutor = _AutoExecutor(self, self._steps)
 
     async def run(

upstash_workflow/asyncio/serve/options.py

@@ -92,7 +92,7 @@ def _initial_payload_parser(initial_request: str) -> TInitialPayload:
         ),
         base_url=base_url or environment.get("UPSTASH_WORKFLOW_URL"),
         env=environment,
-        retries=retries or DEFAULT_RETRIES,
+        retries=DEFAULT_RETRIES if retries is None else retries,
         url=url,
     )
 

upstash_workflow/context/context.py

@@ -57,7 +57,7 @@ def __init__(
         self.headers: Dict[str, str] = headers
         self.request_payload: TInitialPayload = initial_payload
         self.env: Dict[str, Optional[str]] = env or {}
-        self.retries: int = retries or DEFAULT_RETRIES
+        self.retries: int = DEFAULT_RETRIES if retries is None else retries
         self._executor: _AutoExecutor = _AutoExecutor(self, self._steps)
 
     def run(

upstash_workflow/serve/options.py

@@ -106,7 +106,7 @@ def _initial_payload_parser(initial_request: str) -> TInitialPayload:
         ),
         base_url=base_url or environment.get("UPSTASH_WORKFLOW_URL"),
         env=environment,
-        retries=retries or DEFAULT_RETRIES,
+        retries=DEFAULT_RETRIES if retries is None else retries,
         url=url,
     )
 

upstash_workflow/workflow_requests.py

@@ -299,7 +299,7 @@ def _get_headers(
                 )
 
     content_type = user_headers.get("Content-Type") if user_headers else None
-    content_type = content_type or DEFAULT_CONTENT_TYPE
+    content_type = DEFAULT_CONTENT_TYPE if content_type is None else content_type
 
     if step and step.call_headers is not None:
         forwarded_headers = {