Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-10-16 09:02:14 +02:00
parent dcb13a7e6f
commit 4ca7689f31
3 changed files with 34 additions and 22 deletions

View file

@ -106,13 +106,16 @@ var sessionLock sync.Mutex
func RegisterRealtime(cl *config.BackendConfigLoader, ml *model.ModelLoader, appConfig *config.ApplicationConfig) func(c *websocket.Conn) {
return func(c *websocket.Conn) {
log.Debug().Msgf("WebSocket connection established with '%s'", c.RemoteAddr().String())
// Generate a unique session ID
sessionID := generateSessionID()
modelFile, input, err := readWSRequest(c, cl, ml, appConfig, true)
if err != nil {
return fmt.Errorf("failed reading parameters from request:%w", err)
}
// modelFile, input, err := readWSRequest(c, cl, ml, appConfig, true)
// if err != nil {
// return fmt.Errorf("failed reading parameters from request:%w", err)
// }
session := &Session{
ID: sessionID,