🐛 Bug Report: Traceloop SDK does not always make an API request at the completion of a Colab/Jupyter notebook block

Which component is this bug for?

Traceloop SDK

📜 Description

If you instantiate the Traceloop SDK in Block A:

from traceloop.sdk import Traceloop
import os
# This is just for easy illustrative purposes so people don't need to set up any API keys
os.environ['TRACELOOP_API_KEY'] = 'asdasd'
os.environ['TRACELOOP_BASE_URL'] = 'asdasd'
Traceloop.init(disable_batch=True)

and you run the following code in Block B:

from openai import OpenAI

api_key = input("Please enter your OpenAI API key: ")

client = OpenAI(api_key=api_key)

completion = client.chat.completions.create(
    model='gpt-4o',
    messages=[{"role":"user","content":"What's the meaning of life?"}]
)

print(completion.choices[0].message.content)

then the Traceloop SDK should send an API request at the completion of Block B to record the OpenAI request. With appropriate logging set, you can see that this is not the case.

I see https://github.com/traceloop/openllmetry/pull/66, but it seems like this same issue might have reared its head again. I swear this used to work.

👟 Reproduction steps

I have this notebook which reproduces the issue. If you run it, you should see that there is no Traceloop API call in the OpenAI call code block.

👍 Expected behavior

The API call should be triggered at the completion of every code block.

👎 Actual Behavior with Screenshots

The only API calls logged are OpenAI API calls.

🤖 Python Version

No response

📃 Provide any additional context for the Bug.

No response

👀 Have you spent some time to check if this bug has been raised before?

• I checked and didn't find similar issue

Are you willing to submit PR?

None

Start at the Traceloop.init entry point and compare the reported behavior with pull request #66, then run the linked Colab notebook using the two-block reproduction. Done means an API request is logged at the completion of each notebook code block, including the OpenAI call in Block B.