ChatFAQ
diff --git a/‎back/back/apps/language_model/consumers/__init__.py‎
Lines changed: 4 additions & 4 deletions b/‎back/back/apps/language_model/consumers/__init__.py‎
Lines changed: 4 additions & 4 deletions
diff --git a/‎sdk/chatfaq_sdk/__init__.py‎
Lines changed: 3 additions & 2 deletions b/‎sdk/chatfaq_sdk/__init__.py‎
Lines changed: 3 additions & 2 deletions
@@ -141,7 +141,7 @@ def merge_contents(existing: List[Content], new: List[Content]) -> List[Content]
                     content=aggregated_contents,
                     usage=None,
                     stop_reason="end_turn"
-                )
+                ).model_dump()
             )
             # Start a new group for the new role.
             current_role = role
@@ -155,7 +155,7 @@ def merge_contents(existing: List[Content], new: List[Content]) -> List[Content]
                 content=aggregated_contents,
                 usage=None,
                 stop_reason="end_turn"
-            )
+            ).model_dump()
         )
 
     return aggregated_messages
@@ -283,9 +283,9 @@ async def query_llm(
         if messages: # In case the fsm sends messages
             if messages[0]["role"] == AgentType.system.value:
                 if prev_messages[0].role == AgentType.system.value:
-                    new_messages[0] = Message(**messages[0])  # replace the original system message with the new one from the fsm
+                    new_messages[0] = messages[0].copy()  # replace the original system message with the new one from the fsm
                 else:
-                    new_messages.insert(0, Message(**messages[0]))  # or add the fsm system message
+                    new_messages.insert(0, messages[0].copy())  # or add the fsm system message
 
                 # pop the system message
                 messages = messages[1:]
 
@@ -13,6 +13,7 @@
 import httpx
 import sentry_sdk
 import websockets
+from websockets.protocol import State
 
 from chatfaq_sdk import settings
 from chatfaq_sdk.conditions import Condition
@@ -183,7 +184,7 @@ async def producer(self, actions, consumer_route):
         ws_attrs = [attr for attr in dir(self) if attr.startswith("ws_")]
         while True:
             if any(
-                getattr(self, ws_attr) is None or not getattr(self, ws_attr).open
+                getattr(self, ws_attr) is None or getattr(self, ws_attr).state != State.OPEN
                 for ws_attr in ws_attrs
             ):
                 await asyncio.sleep(0.01)
@@ -245,7 +246,7 @@ async def _disconnect(self):
         logger.info("Shutting Down...")
         wss = [getattr(self, attr) for attr in dir(self) if attr.startswith("ws_")]
         for ws in wss:
-            if ws is not None and ws.open:
+            if ws is not None and ws.state == State.OPEN:
                 await ws.close()
 
     async def rpc_request_callback(self, payload):