markdown to html benchmark

benchmarks/markdown_to_html/.gitignore

@@ -0,0 +1 @@
+.caribou/deployment-packages/

benchmarks/markdown_to_html/app.py

@@ -0,0 +1,44 @@
+from typing import Any
+import markdown
+import base64
+import boto3
+import json
+from tempfile import TemporaryDirectory
+from caribou.deployment.client import CaribouWorkflow
+
+s3_bucket_name = "caribou-markdown-to-html"
+s3_bucket_region_name = "us-east-1"
+
+workflow = CaribouWorkflow(name="markdown_to_html", version="0.0.1")
+
+@workflow.serverless_function(
+    name="markdown_to_html",
+    entry_point=True,
+)
+def markdown_to_html(event: dict[str, Any]) -> dict[str, Any]:
+
+    if isinstance(event, str):
+        event = json.loads(event)
+
+    if "filename" in event:
+        filename = event["filename"]
+    else:
+        raise ValueError("No filename provided")
+
+    s3 = boto3.client("s3", region_name=s3_bucket_region_name)
+
+    with TemporaryDirectory() as tmp_dir:
+        s3.download_file(s3_bucket_name, filename, f"{tmp_dir}/{filename}")
+
+        with open(f"{tmp_dir}/{filename}", "r") as f:
+            markdown_text = f.read()
+
+        decoded_text = base64.b64decode(markdown_text).decode()
+        html_text = markdown.markdown(decoded_text)
+
+        with open(f"{tmp_dir}/{filename}.html", "w") as f:
+            f.write(html_text)
+
+        s3.upload_file(f"{tmp_dir}/{filename}.html", s3_bucket_name, f"output/{filename}.html")
+
+    return {"status": 200}

benchmarks/markdown_to_html/requirements.txt

@@ -0,0 +1,6 @@
+markdown
+boto3==1.35.3
+
+pyyaml==6.0.2
+
+pytz==2024.1