Add async source

JoinTheAlliance · Aug 1, 2023 · a705eab · a705eab
1 parent 100c671
commit a705eab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agentcomlink/server.py b/agentcomlink/server.py
@@ -99,7 +99,7 @@ def send_message(message, type="chat", source="default"):
         asyncio.run(ws.send_text(message))
 
 
-async def async_send_message(message, type="chat"):
+async def async_send_message(message, type="chat", source="default"):
     """
     Send a message to the websocket.
 
@@ -109,7 +109,7 @@ async def async_send_message(message, type="chat"):
     global loop
     if ws is not None and loop is not None:
         print("send text")
-        message = json.dumps({"type": type, "message": message})
+        message = json.dumps({"type": type, "message": message, "source": source})
         print(message)
         await ws.send_text(message)