fix unbound reference issue (#807)

* fix unbound reference issue * bump version
Chainlit · Mar 11, 2024 · 3b6852c · 3b6852c
1 parent 3b17af9
commit 3b6852c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py
@@ -525,11 +525,13 @@ def _on_run_update(self, run: Run) -> None:
                 # find first message with prompt_id
                 for m in chat_start["input_messages"]:
                     if m.additional_kwargs.get("prompt_id"):
-                        current_step.generation.prompt_id = m.additional_kwargs["prompt_id"]
+                        current_step.generation.prompt_id = m.additional_kwargs[
+                            "prompt_id"
+                        ]
                         if custom_variables := m.additional_kwargs.get("variables"):
                             current_step.generation.variables = custom_variables
                     break
-    
+
                 current_step.language = "json"
                 current_step.output = json.dumps(message_completion)
             else:
@@ -558,8 +560,6 @@ def _on_run_update(self, run: Run) -> None:
                 self._run_sync(current_step.update())
 
             if self.final_stream and self.has_streamed_final_answer:
-                if self.final_stream.content:
-                    self.final_stream.content = completion
                 self._run_sync(self.final_stream.update())
 
             return

diff --git a/backend/pyproject.toml b/backend/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chainlit"
-version = "1.0.400"
+version = "1.0.401"
 keywords = ['LLM', 'Agents', 'gen ai', 'chat ui', 'chatbot ui', 'openai', 'copilot', 'langchain', 'conversational ai']
 description = "Build Conversational AI."
 authors = ["Chainlit"]