Skip to content

Commit 2fee559

Browse files
committed
Clean up changes in user_message
1 parent 5ddc958 commit 2fee559

2 files changed

Lines changed: 4 additions & 6 deletions

File tree

chat_server/sio/handlers/user_message.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,6 @@ async def user_message(sid, data):
4949
"""
5050
LOG.debug(f"Received user message data: {data}")
5151
try:
52-
data.setdefault("sid", "") # TODO: This is patching clients that exclude `sid`
5352
message = UserMessage(**data)
5453
is_bot = message.is_bot == "1"
5554
is_proctor = message.username.startswith("proctor")
@@ -132,7 +131,7 @@ async def user_message(sid, data):
132131
new_prompt_data = message.to_new_prompt_message()
133132
new_prompt_data.context = prompt_data.get("context", {})
134133
LOG.info(f"Emitting new_prompt_message: {new_prompt_data.model_dump()}")
135-
# TODO: Consider backwards-compat. patching of `user_id` handling
134+
# HACK: backwards-compat. patching of `userID` handling
136135
await sio.emit(
137136
"new_prompt_message",
138137
data={**new_prompt_data.model_dump(), "userID": message.user_uid},
@@ -148,9 +147,7 @@ async def user_message(sid, data):
148147
)
149148

150149
message.bound_service = cid_data.get("bound_service", "")
151-
# keys_diff = set(data.keys()).difference(set(message.model_dump().keys()))
152-
# LOG.info(f"Removed keys={keys_diff}")
153-
# TODO: Consider backwards-compat. patching of `user_id` handling
150+
# HACK: backwards-compat. patching of `userID` handling
154151
await sio.emit(
155152
"new_message",
156153
data={**message.model_dump(), "userID": message.user_uid},

requirements/requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@ fastapi==0.115.6
55
httpx==0.28.1 # required by FastAPI
66
klatchat-utils~=0.0,>=0.0.1a2
77
kubernetes==29.0.0
8-
neon-data-models~=0.0,>=0.0.2a13
8+
#neon-data-models~=0.0,>=0.0.2a13
9+
neon-data-models@git+https://github.com/neongeckocom/neon-data-models@FIX_UserMessageSid
910
neon-sftp~=0.1
1011
neon_utils[sentry]~=1.12
1112
pydantic~=2.7

0 commit comments

Comments
 (0)