fix: ci

Chainlit · Feb 18, 2025 · 3ae371a · 3ae371a
1 parent 89e4908
commit 3ae371a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/literalai/observability/step.py b/literalai/observability/step.py
@@ -512,6 +512,7 @@ def __call__(self, func):
             parent_id=self.parent_id,
             thread_id=self.thread_id,
             root_run_id=self.root_run_id,
+            **self.kwargs,
         )
 
     async def __aenter__(self):

diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
@@ -276,12 +276,12 @@ async def test_ingestion(
                 step.metadata = {"foo": "bar"}
                 assert async_client.event_processor.event_queue._qsize() == 0
                 stack = active_steps_var.get()
-                assert len(stack) == 1
+                assert stack is not None and len(stack) == 1
 
             assert async_client.event_processor.event_queue._qsize() == 1
 
         stack = active_steps_var.get()
-        assert len(stack) == 0
+        assert stack is not None and len(stack) == 0
 
     @pytest.mark.timeout(5)
     async def test_thread_decorator(
@@ -666,14 +666,14 @@ async def test_prompt(self, async_client: AsyncLiteralClient):
     async def test_prompt_cache(self, async_client: AsyncLiteralClient):
         prompt = await async_client.api.get_prompt(name="Default", version=0)
         assert prompt is not None
-        
+
         original_key = async_client.api.api_key
         async_client.api.api_key = "invalid-api-key"
-        
+
         cached_prompt = await async_client.api.get_prompt(name="Default", version=0)
         assert cached_prompt is not None
         assert cached_prompt.id == prompt.id
-        
+
         async_client.api.api_key = original_key
 
     @pytest.mark.timeout(5)