restore context

Chainlit · Nov 17, 2023 · ae291aa · ae291aa
1 parent a4548d1
commit ae291aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py
@@ -1,4 +1,3 @@
-import asyncio
 from typing import Any, Dict, List, Optional, Union
 from uuid import UUID
 
@@ -402,6 +401,7 @@ def __init__(
  self.to_keep = to_keep
 
  def _run_sync(self, co):
+ context_var.set(self.context)
  self.context.loop.create_task(co)
 
  def _persist_run(self, run: Run) -> None:

diff --git a/backend/pyproject.toml b/backend/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chainlit"
-version = "0.7.604rc1"
+version = "0.7.604rc2"
 keywords = ['LLM', 'Agents', 'gen ai', 'chat ui', 'chatbot ui', 'langchain']
 description = "A faster way to build chatbot UIs."
 authors = ["Chainlit"]