Skip to content

Commit 51b5e08

Browse files
generate text mode protos (#598)
Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 98f7a91 commit 51b5e08

5 files changed

Lines changed: 135 additions & 2 deletions

File tree

livekit-protocol/generate_proto.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
set -e
2020

21+
2122
API_PROTOCOL=./protocol/protobufs
2223
API_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"
8688
mkdir -p "$API_OUT_PYTHON/agent_pb"
8789
mv "$API_OUT_PYTHON/agent/livekit_agent_session_pb2.py" "$API_OUT_PYTHON/agent_pb/agent_session.py"
8890
mv "$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

9094
mkdir -p "$API_OUT_PYTHON/logger_pb"
9195
mv "$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)
102106
find "$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]
105112
perl -i -pe 's|^(\w+_FIELD_NUMBER): _ClassVar\[int\]|$1: int|g' "$API_OUT_PYTHON/logger_pb/options.pyi"

livekit-protocol/livekit/protocol/agent_pb/__init__.py

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

livekit-protocol/livekit/protocol/agent_pb/agent_text.py

Lines changed: 47 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

livekit-protocol/livekit/protocol/agent_pb/agent_text.pyi

Lines changed: 79 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

livekit-protocol/livekit/protocol/version.py

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)