mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
104 lines
4.3 KiB
Python
104 lines
4.3 KiB
Python
import json
|
|
|
|
import quart
|
|
|
|
from ... import group
|
|
|
|
|
|
@group.group_class('webchat', '/api/v1/pipelines/<pipeline_uuid>/chat')
|
|
class WebChatDebugRouterGroup(group.RouterGroup):
|
|
async def initialize(self) -> None:
|
|
@self.route('/send', methods=['POST'])
|
|
async def send_message(pipeline_uuid: str) -> str:
|
|
"""Send a message to the pipeline for debugging"""
|
|
|
|
async def stream_generator(generator):
|
|
async for message in generator:
|
|
yield rf"data:{json.dumps({'message': message})}\n\n"
|
|
yield "data:{type: end}\n\n''"
|
|
try:
|
|
data = await quart.request.get_json()
|
|
session_type = data.get('session_type', 'person')
|
|
message_chain_obj = data.get('message', [])
|
|
is_stream = data.get('is_stream', False)
|
|
|
|
if not message_chain_obj:
|
|
return self.http_status(400, -1, 'message is required')
|
|
|
|
if session_type not in ['person', 'group']:
|
|
return self.http_status(400, -1, 'session_type must be person or group')
|
|
|
|
webchat_adapter = self.ap.platform_mgr.webchat_proxy_bot.adapter
|
|
|
|
if not webchat_adapter:
|
|
return self.http_status(404, -1, 'WebChat adapter not found')
|
|
|
|
if is_stream:
|
|
|
|
generator = webchat_adapter.send_webchat_message(pipeline_uuid, session_type, message_chain_obj, is_stream)
|
|
|
|
return quart.Response(
|
|
stream_generator(generator),
|
|
mimetype='text/event-stream'
|
|
)
|
|
|
|
else:
|
|
# result = await webchat_adapter.send_webchat_message(pipeline_uuid, session_type, message_chain_obj)
|
|
result = None
|
|
async for message in webchat_adapter.send_webchat_message(pipeline_uuid, session_type, message_chain_obj):
|
|
result = message
|
|
if result is not None:
|
|
return self.success(
|
|
data={
|
|
'message': result,
|
|
}
|
|
)
|
|
else:
|
|
return self.http_status(400, -1, 'message is required')
|
|
|
|
|
|
except Exception as e:
|
|
return self.http_status(500, -1, f'Internal server error: {str(e)}')
|
|
|
|
@self.route('/messages/<session_type>', methods=['GET'])
|
|
async def get_messages(pipeline_uuid: str, session_type: str) -> str:
|
|
"""Get the message history of the pipeline for debugging"""
|
|
try:
|
|
if session_type not in ['person', 'group']:
|
|
return self.http_status(400, -1, 'session_type must be person or group')
|
|
|
|
webchat_adapter = self.ap.platform_mgr.webchat_proxy_bot.adapter
|
|
|
|
if not webchat_adapter:
|
|
return self.http_status(404, -1, 'WebChat adapter not found')
|
|
|
|
messages = webchat_adapter.get_webchat_messages(pipeline_uuid, session_type)
|
|
|
|
return self.success(data={'messages': messages})
|
|
|
|
except Exception as e:
|
|
return self.http_status(500, -1, f'Internal server error: {str(e)}')
|
|
|
|
@self.route('/reset/<session_type>', methods=['POST'])
|
|
async def reset_session(session_type: str) -> str:
|
|
"""Reset the debug session"""
|
|
try:
|
|
if session_type not in ['person', 'group']:
|
|
return self.http_status(400, -1, 'session_type must be person or group')
|
|
|
|
webchat_adapter = None
|
|
for bot in self.ap.platform_mgr.bots:
|
|
if hasattr(bot.adapter, '__class__') and bot.adapter.__class__.__name__ == 'WebChatAdapter':
|
|
webchat_adapter = bot.adapter
|
|
break
|
|
|
|
if not webchat_adapter:
|
|
return self.http_status(404, -1, 'WebChat adapter not found')
|
|
|
|
webchat_adapter.reset_debug_session(session_type)
|
|
|
|
return self.success(data={'message': 'Session reset successfully'})
|
|
|
|
except Exception as e:
|
|
return self.http_status(500, -1, f'Internal server error: {str(e)}')
|