Function set_progress

redis_config.py:41–68

(task_id: str, progress: float, message: Union[str, None] = None)

def set_progress(task_id: str, progress: float, message: Union[str, None] = None):
    if not redis_client:
        return  # Skip if connection failed
    if not task_id:
        return  # skip if not run as API
    key = get_progress_key(task_id)

    # Retrieve the existing status if message is None
    if message is None:
        try:
            raw_data = redis_client.get(key)
            if raw_data:
                existing_status = json.loads(raw_data)
                message = existing_status.get("message", "initial message")
            else:
                message = "initial message"
        except Exception as e:
            print(
                f"ERROR: Failed to retrieve existing message from Redis for {task_id}: {e}"
            )
            message = "initial message"

    status = {"progress": progress, "message": message}
    try:
        redis_client.set(key, json.dumps(status), ex=DEFAULT_EXPIRY_SECONDS)
        print(f"Task {task_id}: {progress*100:.2f}% complete - {message}")
    except Exception as e:
        print(f"ERROR: Failed to set progress in Redis for {task_id}: {e}")


def get_progress(