From d73af56690a4f32aa78101bd5e8ee3d644955857 Mon Sep 17 00:00:00 2001
From: cybermaggedon <cybermaggedon@gmail.com>
Date: Thu, 11 Sep 2025 15:56:22 +0100
Subject: [PATCH] Fix/websocket close await (1.4) (#517)

* Vector stores will create collections on query (#511)

* Return empty embeddings list as empty list through gateway (#513)

* Await async websocket closure
---
 trustgraph-flow/trustgraph/gateway/dispatch/mux.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/trustgraph-flow/trustgraph/gateway/dispatch/mux.py b/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
index afce6b75..ddaa8ddf 100644
--- a/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
+++ b/trustgraph-flow/trustgraph/gateway/dispatch/mux.py
@@ -147,7 +147,7 @@ class Mux:
                 self.running.stop()
 
                 if self.ws:
-                    self.ws.close()
+                    await self.ws.close()
                     self.ws = None
 
                 break
@@ -165,6 +165,6 @@ class Mux:
         self.running.stop()
 
         if self.ws:
-            self.ws.close()
+            await self.ws.close()
             self.ws = None