wip

Author: theomonnom

livekit-api/livekit/api/__init__.py

@@ -36,7 +36,14 @@
 
 from .twirp_client import TwirpError, TwirpErrorCode
 from .livekit_api import LiveKitAPI
-from .access_token import VideoGrants, SIPGrants, AccessToken, TokenVerifier
+from .access_token import (
+    InferenceGrants,
+    ObservabilityGrants,
+    VideoGrants,
+    SIPGrants,
+    AccessToken,
+    TokenVerifier,
+)
 from .webhook import WebhookReceiver
 from .version import __version__
 
@@ -47,6 +54,8 @@
     "ingress_service",
     "sip_service",
     "agent_dispatch_service",
+    "InferenceGrants",
+    "ObservabilityGrants",
     "VideoGrants",
     "SIPGrants",
     "AccessToken",

livekit-protocol/generate_proto.sh

@@ -34,7 +34,8 @@ protoc \
     $API_PROTOCOL/livekit_agent_dispatch.proto \
     $API_PROTOCOL/livekit_metrics.proto \
     $API_PROTOCOL/livekit_sip.proto \
-    $API_PROTOCOL/livekit_analytics.proto
+    $API_PROTOCOL/livekit_analytics.proto \
+    $API_PROTOCOL/agent/livekit_agent_session.proto
 
 
 touch -a "$API_OUT_PYTHON/__init__.py"
@@ -68,7 +69,9 @@ mv "$API_OUT_PYTHON/livekit_sip_pb2.py" "$API_OUT_PYTHON/sip.py"
 mv "$API_OUT_PYTHON/livekit_sip_pb2.pyi" "$API_OUT_PYTHON/sip.pyi"
 mv "$API_OUT_PYTHON/livekit_metrics_pb2.py" "$API_OUT_PYTHON/metrics.py"
 mv "$API_OUT_PYTHON/livekit_metrics_pb2.pyi" "$API_OUT_PYTHON/metrics.pyi"
+mv "$API_OUT_PYTHON/agent/livekit_agent_session_pb2.py" "$API_OUT_PYTHON/agent/agent_session.py"
+mv "$API_OUT_PYTHON/agent/livekit_agent_session_pb2.pyi" "$API_OUT_PYTHON/agent/agent_session.pyi"
 
-perl -i -pe 's|^(import (livekit_egress_pb2\|livekit_room_pb2\|livekit_webhook_pb2\|livekit_ingress_pb2\|livekit_models_pb2\|livekit_agent_pb2\|livekit_agent_dispatch_pb2\|livekit_analytics_pb2\|livekit_sip_pb2\|livekit_metrics_pb2))|from . $1|g' "$API_OUT_PYTHON"/*.py "$API_OUT_PYTHON"/*.pyi
+perl -i -pe 's|^(import (livekit_egress_pb2\|livekit_room_pb2\|livekit_webhook_pb2\|livekit_ingress_pb2\|livekit_models_pb2\|livekit_agent_pb2\|livekit_agent_dispatch_pb2\|livekit_analytics_pb2\|livekit_sip_pb2\|livekit_metrics_pb2\|livekit_agent_session_pb2))|from . $1|g' "$API_OUT_PYTHON"/**.py "$API_OUT_PYTHON"/**.pyi
 
-perl -i -pe 's|livekit_(\w+)_pb2|${1}|g' "$API_OUT_PYTHON"/*.py "$API_OUT_PYTHON"/*.pyi
+perl -i -pe 's|livekit_(\w+)_pb2|${1}|g' "$API_OUT_PYTHON"/**.py "$API_OUT_PYTHON"/**.pyi