1818
1919set -e
2020
21+
2122API_PROTOCOL=./protocol/protobufs
2223API_OUT_PYTHON=./livekit/protocol
2324
@@ -37,6 +38,7 @@ protoc \
3738 $API_PROTOCOL /livekit_analytics.proto \
3839 $API_PROTOCOL /livekit_rtc.proto \
3940 $API_PROTOCOL /agent/livekit_agent_session.proto \
41+ $API_PROTOCOL /agent/livekit_agent_text.proto \
4042 $API_PROTOCOL /logger/options.proto \
4143 $API_PROTOCOL /livekit_connector_whatsapp.proto \
4244 $API_PROTOCOL /livekit_connector_twilio.proto \
@@ -86,6 +88,8 @@ mv "$API_OUT_PYTHON/livekit_connector_pb2.pyi" "$API_OUT_PYTHON/connector.pyi"
8688mkdir -p " $API_OUT_PYTHON /agent_pb"
8789mv " $API_OUT_PYTHON /agent/livekit_agent_session_pb2.py" " $API_OUT_PYTHON /agent_pb/agent_session.py"
8890mv " $API_OUT_PYTHON /agent/livekit_agent_session_pb2.pyi" " $API_OUT_PYTHON /agent_pb/agent_session.pyi"
91+ mv " $API_OUT_PYTHON /agent/livekit_agent_text_pb2.py" " $API_OUT_PYTHON /agent_pb/agent_text.py"
92+ mv " $API_OUT_PYTHON /agent/livekit_agent_text_pb2.pyi" " $API_OUT_PYTHON /agent_pb/agent_text.pyi"
8993
9094mkdir -p " $API_OUT_PYTHON /logger_pb"
9195mv " $API_OUT_PYTHON /logger/options_pb2.py" " $API_OUT_PYTHON /logger_pb/options.py"
@@ -101,5 +105,8 @@ find "$API_OUT_PYTHON" -maxdepth 1 -name '*.py' -o -name '*.pyi' | xargs perl -i
101105# fix logger imports for files in subdirectories (need parent-relative import)
102106find " $API_OUT_PYTHON " -mindepth 2 -name ' *.py' -o -name ' *.pyi' | xargs perl -i -pe ' s|from logger import options_pb2 as ([^ ]+)|from ..logger_pb import options as $1|g'
103107
108+ # fix `from agent import agent_xxx as xxx` to `from . import agent_xxx as xxx`
109+ find " $API_OUT_PYTHON " /agent_pb -name ' *.py' -o -name ' *.pyi' | xargs perl -i -pe ' s|from agent import (agent_\w+) as ([^ ]+)|from . import $1 as $2|g'
110+
104111# fixes - error: ClassVar can only be used for assignments in class body [misc]
105112perl -i -pe ' s|^(\w+_FIELD_NUMBER): _ClassVar\[int\]|$1: int|g' " $API_OUT_PYTHON /logger_pb/options.pyi"
0 commit comments