chore: temporal evals & perf benchmarks

.vscode/settings.json

@@ -62,6 +62,8 @@
     "cli",
     "client",
     "examples",
+    "evaluations",
+    "benchmarks",
     "others"
   ]
 }

python/.gitignore

@@ -181,3 +181,11 @@ pyrightconfig.json
 
 # data files
 *.rrd
+
+# benchmark artifacts
+benchmarks/payload
+benchmarks/results
+benchmarks/scenarios/**/payload
+
+# evaluation artifacts
+!evaluations/**/*.rrd

python/Dockerfile

@@ -0,0 +1,35 @@
+FROM python:3.12-slim-bookworm AS base
+
+ENV VIRTUAL_ENV=/app/.venv \
+  PATH="/app/.venv/bin:$PATH"
+
+FROM base AS builder
+
+ENV POETRY_NO_INTERACTION=1 \
+  POETRY_VIRTUALENVS_IN_PROJECT=1 \
+  POETRY_VIRTUALENVS_CREATE=1 \
+  POETRY_CACHE_DIR=/tmp/poetry_cache
+
+WORKDIR /app
+
+RUN pip install --no-cache-dir poetry==1.8.4
+
+COPY pyproject.toml poetry.lock ./
+RUN poetry install --without dev --no-root && rm -rf $POETRY_CACHE_DIR;
+
+
+FROM base AS runtime
+
+COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
+
+
+WORKDIR /app
+
+COPY arflow ./arflow
+COPY cakelab ./cakelab
+
+EXPOSE 8500
+
+ENV PORT=8500
+
+ENTRYPOINT ["python", "-m", "arflow._cli", "view"]

python/arflow/__init__.py

@@ -7,7 +7,20 @@
     SessionStream as SessionStream,
 )
 from cakelab.arflow_grpc.v1.ar_frame_pb2 import ARFrame as ARFrame
+from cakelab.arflow_grpc.v1.audio_frame_pb2 import AudioFrame as AudioFrame
+from cakelab.arflow_grpc.v1.color_frame_pb2 import ColorFrame as ColorFrame
+from cakelab.arflow_grpc.v1.depth_frame_pb2 import DepthFrame as DepthFrame
 from cakelab.arflow_grpc.v1.device_pb2 import Device as Device
+from cakelab.arflow_grpc.v1.gyroscope_frame_pb2 import GyroscopeFrame as GyroscopeFrame
+from cakelab.arflow_grpc.v1.mesh_detection_frame_pb2 import (
+    MeshDetectionFrame as MeshDetectionFrame,
+)
+from cakelab.arflow_grpc.v1.plane_detection_frame_pb2 import (
+    PlaneDetectionFrame as PlaneDetectionFrame,
+)
+from cakelab.arflow_grpc.v1.point_cloud_detection_frame_pb2 import (
+    PointCloudDetectionFrame as PointCloudDetectionFrame,
+)
 from cakelab.arflow_grpc.v1.session_pb2 import Session as Session
 from cakelab.arflow_grpc.v1.transform_frame_pb2 import TransformFrame as TransformFrame
 
@@ -19,6 +32,13 @@
     "ARFlowServicer",
     "ARFrame",
     "TransformFrame",
+    "ColorFrame",
+    "DepthFrame",
+    "GyroscopeFrame",
+    "AudioFrame",
+    "PlaneDetectionFrame",
+    "PointCloudDetectionFrame",
+    "MeshDetectionFrame",
     "SessionStream",
     "Session",
     "Device",

python/arflow/_cli.py

@@ -100,7 +100,7 @@ def parse_args(
         "-p",
         "--port",
         type=int,
-        default=8500,
+        default=int(os.getenv("PORT", 8500)),
         help=f"Port to run the server on (default: %(default)s).",
     )
     view_parser.add_argument(