From fd41ddf733f6dd2958ff03f21215477113851b8e Mon Sep 17 00:00:00 2001
From: dltn <6599399+dltn@users.noreply.github.com>
Date: Wed, 9 Oct 2024 11:28:11 -0700
Subject: [PATCH] bidirectional wrapper

---
 .../providers/impls/meta_reference/inference/parallel_utils.py  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py b/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
index 5b5372240..0fc883bec 100644
--- a/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
+++ b/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
@@ -124,7 +124,7 @@ def mp_rank_0() -> bool:
 
 
 def encode_msg(msg: ProcessingMessage) -> bytes:
-    return msg.model_dump_json().encode("utf-8")
+    return ProcessingMessageWrapper(payload=msg).model_dump_json().encode("utf-8")
 
 
 def retrieve_requests(reply_socket_url: str):