Skip to content
5 changes: 5 additions & 0 deletions livekit-protocol/generate_proto.sh
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ protoc \
$API_PROTOCOL/livekit_analytics.proto \
$API_PROTOCOL/livekit_rtc.proto \
$API_PROTOCOL/agent/livekit_agent_session.proto \
$API_PROTOCOL/agent/livekit_agent_text.proto \
$API_PROTOCOL/logger/options.proto \
$API_PROTOCOL/livekit_connector_whatsapp.proto \
$API_PROTOCOL/livekit_connector_twilio.proto \
Expand Down Expand Up @@ -86,6 +87,8 @@ mv "$API_OUT_PYTHON/livekit_connector_pb2.pyi" "$API_OUT_PYTHON/connector.pyi"
mkdir -p "$API_OUT_PYTHON/agent_pb"
mv "$API_OUT_PYTHON/agent/livekit_agent_session_pb2.py" "$API_OUT_PYTHON/agent_pb/agent_session.py"
mv "$API_OUT_PYTHON/agent/livekit_agent_session_pb2.pyi" "$API_OUT_PYTHON/agent_pb/agent_session.pyi"
mv "$API_OUT_PYTHON/agent/livekit_agent_text_pb2.py" "$API_OUT_PYTHON/agent_pb/agent_text.py"
mv "$API_OUT_PYTHON/agent/livekit_agent_text_pb2.pyi" "$API_OUT_PYTHON/agent_pb/agent_text.pyi"

mkdir -p "$API_OUT_PYTHON/logger_pb"
mv "$API_OUT_PYTHON/logger/options_pb2.py" "$API_OUT_PYTHON/logger_pb/options.py"
Expand All @@ -97,5 +100,7 @@ perl -i -pe 's|livekit_(\w+)_pb2|${1}|g' "$API_OUT_PYTHON"/**.py "$API_OUT_PYTHO

perl -i -pe 's|from logger import options_pb2 as ([^ ]+)|from .logger_pb import options as $1|g' "$API_OUT_PYTHON"/**.py "$API_OUT_PYTHON"/**.pyi

perl -i -pe 's|from agent import livekit_agent_(\w+)_pb2 as ([^ ]+)|from . import agent_$1 as $2|g' "$API_OUT_PYTHON"/agent_pb/*.py "$API_OUT_PYTHON"/agent_pb/*.pyi

# fixes - error: ClassVar can only be used for assignments in class body [misc]
perl -i -pe 's|^(\w+_FIELD_NUMBER): _ClassVar\[int\]|$1: int|g' "$API_OUT_PYTHON/logger_pb/options.pyi"
78 changes: 36 additions & 42 deletions livekit-protocol/livekit/protocol/agent.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading