Spaces:
Runtime error
Runtime error
File size: 5,595 Bytes
c6878c7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 | # import asyncio
# import socketio
# import logging
# # Configure logging
# logging.basicConfig(level=logging.DEBUG)
# logger = logging.getLogger(__name__)
# class RasaSocketIOClient:
# def __init__(self, uri):
# self.sio = socketio.AsyncClient(logger=True, engineio_logger=True) # Enable internal logging
# self.uri = uri
# self.response = None
# @self.sio.on('bot_uttered') # Adjust the event name as per your Rasa setup
# async def on_message(data):
# logger.info("Received response: %s", data)
# self.response = data # Store the latest response
# async def connect(self):
# try:
# await self.sio.connect(self.uri)
# logger.info("Successfully connected to the server.")
# except Exception as e:
# logger.error("Failed to connect to the server: %s", e)
# async def send_message(self, message):
# self.response = None # Reset previous response
# try:
# await self.sio.emit('user_uttered', message) # Adjust the event name as per your Rasa setup
# logger.debug("Message sent: %s", message)
# await asyncio.sleep(1) # Wait for the response to be received
# except Exception as e:
# logger.error("Failed to send message: %s", e)
# return self.response
# async def disconnect(self):
# try:
# await self.sio.disconnect()
# logger.info("Disconnected from the server.")
# except Exception as e:
# logger.error("Failed to disconnect: %s", e)
# import asyncio
# import socketio
# import logging
# from custom_logging import IMPORTANT_LEVEL_NUM
# logger = logging.getLogger(__name__)
# class RasaSocketIOClient:
# def __init__(self, uri):
# self.sio = socketio.AsyncClient(logger=False, engineio_logger=False)
# self.response_event = asyncio.Event()
# self.uri = uri
# self.response = None
# @self.sio.on('bot_uttered') # Adjust the event name as per your Rasa setup
# async def on_message(data):
# # Use the new logging level
# logger.important("Socket ID: %s Received response: %s", self.sio.sid, data)
# self.response = data # Store the latest response
# async def connect(self):
# try:
# await self.sio.connect(self.uri)
# # Use the new logging level
# logger.important("Successfully connected to the server.")
# except Exception as e:
# # Use the new logging level for errors too, for consistency
# logger.important("Failed to connect to the server: %s", e)
# return "Failed to connect to the server."
# async def send_message(self, message):
# self.response = None # Reset previous response
# try:
# await self.sio.emit('user_uttered', message) # Adjust the event name as per your Rasa setup
# # Since this is more of a debug level, consider if you want to keep it under IMPORTANT or not
# logger.important("Message sent: %s", message)
# await asyncio.sleep(1) # Wait for the response to be received
# except Exception as e:
# logger.important("Failed to send message: %s", e)
# return self.response
# async def disconnect(self):
# try:
# await self.sio.disconnect()
# logger.important("Disconnected from the server.")
# except Exception as e:
# logger.important("Failed to disconnect: %s", e)
import asyncio
import socketio
from custom_logging import get_logger
logger = get_logger(__name__)
class RasaSocketIOClient:
def __init__(self, uri, update_chat_history_callback=None):
self.sio = socketio.AsyncClient(logger=True, engineio_logger=True)
self.uri = uri
self.message_queue = asyncio.Queue()
self.register_event_handlers()
def register_event_handlers(self):
@self.sio.event
async def connect():
logger.important("Connected to the server.")
@self.sio.event
async def disconnect():
logger.important("Disconnected from the server.")
@self.sio.on('bot_uttered')
async def on_message(data):
logger.debug(f"Socket ID: {self.sio.sid} Received response: {data}")
await self.message_queue.put(data) # Put received message into the queue
async def connect(self):
try:
await self.sio.connect(self.uri)
logger.important("Successfully connected to the server.")
except Exception as e:
logger.error(f"Failed to connect to the server: {e}")
async def send_message(self, user_message):
try:
await self.sio.emit('user_uttered', {'message': user_message})
logger.important(f"Message sent: {user_message}")
timeout = 4.0 # Timeout in seconds after the last message is received
while True:
try:
data = await asyncio.wait_for(self.message_queue.get(), timeout)
yield data
except asyncio.TimeoutError:
break
except Exception as e:
logger.error(f"Failed to send message: {e}")
async def disconnect(self):
try:
await self.sio.disconnect()
logger.important("Disconnected from the server.")
except Exception as e:
logger.error(f"Failed to disconnect: {e}")
|