lmnr 0.4.36__py3-none-any.whl → 0.4.37__py3-none-any.whl

lmnr/traceloop_sdk/tests/cassettes/test_workflows/test_streaming_workflow.yaml

@@ -1,167 +0,0 @@
-interactions:
-- request:
-    body: '{"messages": [{"role": "user", "content": "Tell me a joke about OpenTelemetry"}],
-      "model": "gpt-3.5-turbo", "stream": true}'
-    headers:
-      accept:
-      - application/json
-      accept-encoding:
-      - gzip, deflate
-      connection:
-      - keep-alive
-      content-length:
-      - '123'
-      content-type:
-      - application/json
-      host:
-      - api.openai.com
-      user-agent:
-      - OpenAI/Python 1.29.0
-      x-stainless-arch:
-      - arm64
-      x-stainless-async:
-      - 'false'
-      x-stainless-lang:
-      - python
-      x-stainless-os:
-      - MacOS
-      x-stainless-package-version:
-      - 1.29.0
-      x-stainless-runtime:
-      - CPython
-      x-stainless-runtime-version:
-      - 3.9.5
-    method: POST
-    uri: https://api.openai.com/v1/chat/completions
-  response:
-    body:
-      string: 'data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"role":"assistant","content":""},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"Why"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        did"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        the"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        Open"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"Te"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"lemetry"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        project"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        go"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        to"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        therapy"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"?"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        Because"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        it"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        had"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        too"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        many"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        mixed"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"
-        signals"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"!"},"logprobs":null,"finish_reason":null}]}
-
-
-        data: {"id":"chatcmpl-9OmDmBFHEMbZ31rnA9lGQVDOehR90","object":"chat.completion.chunk","created":1715692262,"model":"gpt-3.5-turbo-0125","system_fingerprint":null,"choices":[{"index":0,"delta":{},"logprobs":null,"finish_reason":"stop"}]}
-
-
-        data: [DONE]
-
-
-        '
-    headers:
-      CF-Cache-Status:
-      - DYNAMIC
-      CF-RAY:
-      - 883b21bd48433757-MXP
-      Connection:
-      - keep-alive
-      Content-Type:
-      - text/event-stream; charset=utf-8
-      Date:
-      - Tue, 14 May 2024 13:11:02 GMT
-      Server:
-      - cloudflare
-      Set-Cookie:
-      - __cf_bm=.XvQIxrHkdjS_lppgd.ONqVkLX2zudBgR.99K5inab0-1715692262-1.0.1.1-4NFlLxxPE4GhJUwcqVuShu7r29Wsv6mCxObuGBm7WaBuVBW22_MYG6uT0eJ1DUdmuZ9nykZ7bwQeYmQRkGgCdA;
-        path=/; expires=Tue, 14-May-24 13:41:02 GMT; domain=.api.openai.com; HttpOnly;
-        Secure; SameSite=None
-      - _cfuvid=l3N15ZuTdlc6cNEordeiaV5FrYGVfteXAquEnUzpYpk-1715692262541-0.0.1.1-604800000;
-        path=/; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
-      Transfer-Encoding:
-      - chunked
-      alt-svc:
-      - h3=":443"; ma=86400
-      openai-organization:
-      - traceloop
-      openai-processing-ms:
-      - '268'
-      openai-version:
-      - '2020-10-01'
-      strict-transport-security:
-      - max-age=15724800; includeSubDomains
-      x-ratelimit-limit-requests:
-      - '5000'
-      x-ratelimit-limit-tokens:
-      - '160000'
-      x-ratelimit-remaining-requests:
-      - '4999'
-      x-ratelimit-remaining-tokens:
-      - '159974'
-      x-ratelimit-reset-requests:
-      - 12ms
-      x-ratelimit-reset-tokens:
-      - 9ms
-      x-request-id:
-      - req_1bf87984992f16d43b7eb6e9338a0d05
-    status:
-      code: 200
-      message: OK
-version: 1

lmnr/traceloop_sdk/tests/conftest.py

@@ -1,111 +0,0 @@
-# """Unit tests configuration module."""
-
-# import os
-# import pytest
-# from lmnr.traceloop_sdk import Traceloop
-# from lmnr.traceloop_sdk.instruments import Instruments
-# from lmnr.traceloop_sdk.tracing.tracing import TracerWrapper
-# from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-# from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
-
-# # pytest_plugins = []
-
-
-# @pytest.fixture(scope="session")
-# def exporter():
-#     exporter = InMemorySpanExporter()
-#     Traceloop.init(
-#         app_name="test",
-#         resource_attributes={"something": "yes"},
-#         disable_batch=True,
-#         exporter=exporter,
-#     )
-#     return exporter
-
-
-# @pytest.fixture(autouse=True)
-# def clear_exporter(exporter):
-#     exporter.clear()
-
-
-# @pytest.fixture(autouse=True)
-# def environment():
-#     if "OPENAI_API_KEY" not in os.environ:
-#         os.environ["OPENAI_API_KEY"] = "test_api_key"
-
-
-# @pytest.fixture(scope="module")
-# def vcr_config():
-#     return {
-#         "filter_headers": ["authorization"],
-#         "ignore_hosts": ["openaipublic.blob.core.windows.net"],
-#     }
-
-
-# @pytest.fixture
-# def exporter_with_custom_span_processor():
-#     # Clear singleton if existed
-#     if hasattr(TracerWrapper, "instance"):
-#         _trace_wrapper_instance = TracerWrapper.instance
-#         del TracerWrapper.instance
-
-#     class CustomSpanProcessor(SimpleSpanProcessor):
-#         def on_start(self, span, parent_context=None):
-#             span.set_attribute("custom_span", "yes")
-
-#     exporter = InMemorySpanExporter()
-#     Traceloop.init(
-#         exporter=exporter,
-#         processor=CustomSpanProcessor(exporter),
-#     )
-
-#     yield exporter
-
-#     # Restore singleton if any
-#     if _trace_wrapper_instance:
-#         TracerWrapper.instance = _trace_wrapper_instance
-
-
-# @pytest.fixture
-# def exporter_with_custom_instrumentations():
-#     # Clear singleton if existed
-#     if hasattr(TracerWrapper, "instance"):
-#         _trace_wrapper_instance = TracerWrapper.instance
-#         del TracerWrapper.instance
-
-#     exporter = InMemorySpanExporter()
-#     Traceloop.init(
-#         exporter=exporter,
-#         disable_batch=True,
-#         instruments=[i for i in Instruments],
-#     )
-
-#     yield exporter
-
-#     # Restore singleton if any
-#     if _trace_wrapper_instance:
-#         TracerWrapper.instance = _trace_wrapper_instance
-
-
-# @pytest.fixture
-# def exporter_with_no_metrics():
-#     # Clear singleton if existed
-#     if hasattr(TracerWrapper, "instance"):
-#         _trace_wrapper_instance = TracerWrapper.instance
-#         del TracerWrapper.instance
-
-#     os.environ["TRACELOOP_METRICS_ENABLED"] = "false"
-
-#     exporter = InMemorySpanExporter()
-
-#     Traceloop.init(
-#         exporter=exporter,
-#         disable_batch=True,
-#     )
-
-#     yield exporter
-
-#     # Restore singleton if any
-#     if _trace_wrapper_instance:
-#         TracerWrapper.instance = _trace_wrapper_instance
-#         os.environ["TRACELOOP_METRICS_ENABLED"] = "true"

lmnr/traceloop_sdk/tests/test_association_properties.py

@@ -1,229 +0,0 @@
-# import pytest
-# from langchain_openai import ChatOpenAI
-# from langchain.prompts import ChatPromptTemplate
-# from opentelemetry.semconv_ai import SpanAttributes
-# from lmnr.traceloop_sdk import Traceloop
-# from lmnr.traceloop_sdk.decorators import task, workflow
-
-
-# def test_association_properties(exporter):
-#     @workflow(name="test_workflow")
-#     def test_workflow():
-#         return test_task()
-
-#     @task(name="test_task")
-#     def test_task():
-#         return
-
-#     Traceloop.set_association_properties({"user_id": 1, "user_name": "John Doe"})
-#     test_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == [
-#         "test_task.task",
-#         "test_workflow.workflow",
-#     ]
-
-#     some_task_span = spans[0]
-#     some_workflow_span = spans[1]
-#     assert (
-#         some_workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == 1
-#     )
-#     assert (
-#         some_workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_name"
-#         ]
-#         == "John Doe"
-#     )
-#     assert (
-#         some_task_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == 1
-#     )
-#     assert (
-#         some_task_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_name"
-#         ]
-#         == "John Doe"
-#     )
-
-
-# def test_association_properties_within_workflow(exporter):
-#     @workflow(name="test_workflow_within")
-#     def test_workflow():
-#         Traceloop.set_association_properties({"session_id": 15})
-#         return
-
-#     test_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == [
-#         "test_workflow_within.workflow",
-#     ]
-
-#     some_workflow_span = spans[0]
-#     assert (
-#         some_workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 15
-#     )
-
-
-# @pytest.mark.vcr
-# def test_langchain_association_properties(exporter):
-#     prompt = ChatPromptTemplate.from_messages(
-#         [("system", "You are helpful assistant"), ("user", "{input}")]
-#     )
-#     model = ChatOpenAI(model="gpt-3.5-turbo")
-
-#     chain = prompt | model
-#     chain.invoke(
-#         {"input": "tell me a short joke"},
-#         {"metadata": {"user_id": "1234", "session_id": 456}},
-#     )
-
-#     spans = exporter.get_finished_spans()
-
-#     assert [
-#         "ChatPromptTemplate.task",
-#         "ChatOpenAI.chat",
-#         "RunnableSequence.workflow",
-#     ] == [span.name for span in spans], [span.name for span in spans]
-
-#     workflow_span = next(
-#         span for span in spans if span.name == "RunnableSequence.workflow"
-#     )
-#     prompt_span = next(span for span in spans if span.name == "ChatPromptTemplate.task")
-#     chat_span = next(span for span in spans if span.name == "ChatOpenAI.chat")
-
-#     assert (
-#         workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-#     assert (
-#         chat_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         chat_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-#     assert (
-#         prompt_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         prompt_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-
-
-# @pytest.mark.vcr
-# def test_langchain_and_external_association_properties(exporter):
-#     @workflow(name="test_workflow_external")
-#     def test_workflow_external():
-#         Traceloop.set_association_properties({"workspace_id": "789"})
-
-#         prompt = ChatPromptTemplate.from_messages(
-#             [("system", "You are helpful assistant"), ("user", "{input}")]
-#         )
-#         model = ChatOpenAI(model="gpt-3.5-turbo")
-
-#         chain = prompt | model
-#         chain.invoke(
-#             {"input": "tell me a short joke"},
-#             {"metadata": {"user_id": "1234", "session_id": 456}},
-#         )
-
-#     test_workflow_external()
-
-#     spans = exporter.get_finished_spans()
-
-#     assert [
-#         "ChatPromptTemplate.task",
-#         "ChatOpenAI.chat",
-#         "RunnableSequence.workflow",
-#         "test_workflow_external.workflow",
-#     ] == [span.name for span in spans], [span.name for span in spans]
-
-#     workflow_span = next(
-#         span for span in spans if span.name == "RunnableSequence.workflow"
-#     )
-#     prompt_span = next(span for span in spans if span.name == "ChatPromptTemplate.task")
-#     chat_span = next(span for span in spans if span.name == "ChatOpenAI.chat")
-
-#     assert (
-#         workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-#     assert (
-#         workflow_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.workspace_id"
-#         ]
-#         == "789"
-#     )
-#     assert (
-#         chat_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         chat_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-#     assert (
-#         chat_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.workspace_id"
-#         ]
-#         == "789"
-#     )
-#     assert (
-#         prompt_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.user_id"
-#         ]
-#         == "1234"
-#     )
-#     assert (
-#         prompt_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.session_id"
-#         ]
-#         == 456
-#     )
-#     assert (
-#         prompt_span.attributes[
-#             f"{SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.workspace_id"
-#         ]
-#         == "789"
-#     )

lmnr/traceloop_sdk/tests/test_manual.py

@@ -1,48 +0,0 @@
-# from opentelemetry.semconv_ai import SpanAttributes
-# import pytest
-# from openai import OpenAI
-# from lmnr.traceloop_sdk.tracing.manual import LLMMessage, track_llm_call
-
-
-# @pytest.fixture
-# def openai_client():
-#     return OpenAI()
-
-
-# @pytest.mark.vcr
-# def test_manual_report(exporter, openai_client):
-#     with track_llm_call(vendor="openai", type="chat") as span:
-#         span.report_request(
-#             model="gpt-3.5-turbo",
-#             messages=[
-#                 LLMMessage(role="user", content="Tell me a joke about opentelemetry")
-#             ],
-#         )
-
-#         res = openai_client.chat.completions.create(
-#             model="gpt-3.5-turbo",
-#             messages=[
-#                 {"role": "user", "content": "Tell me a joke about opentelemetry"}
-#             ],
-#         )
-
-#         span.report_response(res.model, [text.message.content for text in res.choices])
-
-#     spans = exporter.get_finished_spans()
-#     open_ai_span = spans[0]
-#     assert open_ai_span.attributes[SpanAttributes.LLM_REQUEST_MODEL] == "gpt-3.5-turbo"
-#     assert open_ai_span.attributes[f"{SpanAttributes.LLM_PROMPTS}.0.role"] == "user"
-#     assert (
-#         open_ai_span.attributes[f"{SpanAttributes.LLM_PROMPTS}.0.content"]
-#         == "Tell me a joke about opentelemetry"
-#     )
-#     assert (
-#         open_ai_span.attributes[SpanAttributes.LLM_RESPONSE_MODEL]
-#         == "gpt-3.5-turbo-0125"
-#     )
-#     assert (
-#         open_ai_span.attributes[f"{SpanAttributes.LLM_COMPLETIONS}.0.content"]
-#         == "Why did the opentelemetry developer break up with their partner? Because they were tired"
-#         + " of constantly tracing their every move!"
-#     )
-#     assert open_ai_span.end_time > open_ai_span.start_time

lmnr/traceloop_sdk/tests/test_nested_tasks.py

@@ -1,47 +0,0 @@
-# from opentelemetry.semconv_ai import SpanAttributes
-# from lmnr.traceloop_sdk.decorators import task, workflow
-# from pytest import raises
-
-
-# def test_nested_tasks(exporter):
-#     @workflow(name="some_workflow")
-#     def some_workflow():
-#         return outer_task()
-
-#     @task(name="outer_task")
-#     def outer_task():
-#         return inner_task()
-
-#     @task(name="inner_task")
-#     def inner_task():
-#         return inner_inner_task()
-
-#     @task(name="inner_inner_task")
-#     def inner_inner_task():
-#         return
-
-#     some_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == [
-#         "inner_inner_task.task",
-#         "inner_task.task",
-#         "outer_task.task",
-#         "some_workflow.workflow",
-#     ]
-
-#     inner_inner_task_span = spans[0]
-#     inner_task_span = spans[1]
-#     outer_task_span = spans[2]
-#     some_workflow_span = spans[3]
-
-#     assert (
-#         inner_inner_task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_PATH]
-#         == "outer_task.inner_task"
-#     )
-#     assert (
-#         inner_task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_PATH] == "outer_task"
-#     )
-#     with raises(KeyError):
-#         _ = outer_task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_PATH]
-#         _ = some_workflow_span.attributes[SpanAttributes.TRACELOOP_ENTITY_PATH]

lmnr/traceloop_sdk/tests/test_privacy_no_prompts.py

@@ -1,50 +0,0 @@
-# import os
-
-# import pytest
-# from openai import OpenAI
-# from opentelemetry.semconv_ai import SpanAttributes
-# from lmnr.traceloop_sdk.decorators import workflow, task
-
-
-# @pytest.fixture(autouse=True)
-# def disable_trace_content():
-#     os.environ["TRACELOOP_TRACE_CONTENT"] = "false"
-#     yield
-#     os.environ["TRACELOOP_TRACE_CONTENT"] = "true"
-
-
-# @pytest.fixture
-# def openai_client():
-#     return OpenAI()
-
-
-# @pytest.mark.vcr
-# def test_simple_workflow(exporter, openai_client):
-#     @task(name="joke_creation")
-#     def create_joke():
-#         completion = openai_client.chat.completions.create(
-#             model="gpt-3.5-turbo",
-#             messages=[
-#                 {"role": "user", "content": "Tell me a joke about opentelemetry"}
-#             ],
-#         )
-#         return completion.choices[0].message.content
-
-#     @workflow(name="pirate_joke_generator")
-#     def joke_workflow():
-#         create_joke()
-
-#     joke_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == [
-#         "openai.chat",
-#         "joke_creation.task",
-#         "pirate_joke_generator.workflow",
-#     ]
-#     open_ai_span = spans[0]
-#     assert open_ai_span.attributes[SpanAttributes.LLM_USAGE_PROMPT_TOKENS] == 15
-#     assert not open_ai_span.attributes.get(f"{SpanAttributes.LLM_PROMPTS}.0.content")
-#     assert not open_ai_span.attributes.get(
-#         f"{SpanAttributes.LLM_PROMPTS}.0.completions"
-#     )