From d4de7758bf753888b0eb771b2a993566d5b51fad Mon Sep 17 00:00:00 2001
From: Razvan Dinu <rdinu@nvidia.com>
Date: Thu, 23 Nov 2023 11:09:27 +0200
Subject: [PATCH] Fix ChatGenerationChunk issue.

Fixes the issue with the latest version of LangChain.
---
 nemoguardrails/streaming.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nemoguardrails/streaming.py b/nemoguardrails/streaming.py
index fdc31d87f..de75eeb15 100644
--- a/nemoguardrails/streaming.py
+++ b/nemoguardrails/streaming.py
@@ -191,6 +191,8 @@ async def push_chunk(
             chunk = chunk.text
         elif isinstance(chunk, AIMessageChunk):
             chunk = chunk.content
+        elif isinstance(chunk, ChatGenerationChunk):
+            chunk = chunk.text
         elif isinstance(chunk, str) or chunk is None:
             pass
         else: