Spaces:
Building
Building
Update websocket_handler.py
Browse files- websocket_handler.py +2 -2
websocket_handler.py
CHANGED
@@ -352,9 +352,9 @@ async def process_user_input(websocket: WebSocket, conversation: ConversationMan
|
|
352 |
|
353 |
# Get response based on session state
|
354 |
if conversation.session.state == "await_param":
|
355 |
-
response_text = await
|
356 |
else:
|
357 |
-
response_text = await
|
358 |
|
359 |
# Add response to history
|
360 |
conversation.session.add_turn("assistant", response_text)
|
|
|
352 |
|
353 |
# Get response based on session state
|
354 |
if conversation.session.state == "await_param":
|
355 |
+
response_text = await handle_parameter_followup(conversation.session, user_text)
|
356 |
else:
|
357 |
+
response_text = await handle_new_message(conversation.session, user_text)
|
358 |
|
359 |
# Add response to history
|
360 |
conversation.session.add_turn("assistant", response_text)
|