From c500614733e09fc5eae06822cff8b7a09d75b979 Mon Sep 17 00:00:00 2001
From: Sondre Olsen <sools@equinor.com>
Date: Thu, 26 Sep 2024 12:38:40 +0200
Subject: [PATCH] Specify supported media connection types

---
 src/robot_interface/telemetry/media_connection_type.py | 5 +++++
 src/robot_interface/telemetry/payloads.py              | 3 ++-
 2 files changed, 7 insertions(+), 1 deletion(-)
 create mode 100644 src/robot_interface/telemetry/media_connection_type.py

diff --git a/src/robot_interface/telemetry/media_connection_type.py b/src/robot_interface/telemetry/media_connection_type.py
new file mode 100644
index 00000000..6530adcf
--- /dev/null
+++ b/src/robot_interface/telemetry/media_connection_type.py
@@ -0,0 +1,5 @@
+from enum import Enum
+
+
+class MediaConnectionType(str, Enum):
+    LiveKit: str = "LiveKit"
diff --git a/src/robot_interface/telemetry/payloads.py b/src/robot_interface/telemetry/payloads.py
index 18207d1d..e175606d 100644
--- a/src/robot_interface/telemetry/payloads.py
+++ b/src/robot_interface/telemetry/payloads.py
@@ -6,6 +6,7 @@
 from transitions import State
 
 from robot_interface.models.mission.status import RobotStatus
+from robot_interface.telemetry.media_connection_type import MediaConnectionType
 
 
 @dataclass
@@ -59,7 +60,7 @@ class VideoStream:
 class MediaConfig(TelemetryPayload):
     url: str
     token: str
-    media_connection_type: str
+    media_connection_type: MediaConnectionType
 
 
 @dataclass