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

lmnr/traceloop_sdk/tests/test_sdk_initialization.py

@@ -1,57 +0,0 @@
-# import pytest
-# from openai import OpenAI
-# from lmnr.traceloop_sdk.decorators import workflow
-
-
-# @pytest.fixture
-# def openai_client():
-#     return OpenAI()
-
-
-# @pytest.mark.vcr
-# def test_resource_attributes(exporter, openai_client):
-#     openai_client.chat.completions.create(
-#         model="gpt-3.5-turbo",
-#         messages=[{"role": "user", "content": "Tell me a joke about opentelemetry"}],
-#     )
-
-#     spans = exporter.get_finished_spans()
-#     open_ai_span = spans[0]
-#     assert open_ai_span.resource.attributes["something"] == "yes"
-#     assert open_ai_span.resource.attributes["service.name"] == "test"
-
-
-# def test_custom_span_processor(exporter_with_custom_span_processor):
-#     @workflow()
-#     def run_workflow():
-#         pass
-
-#     run_workflow()
-
-#     spans = exporter_with_custom_span_processor.get_finished_spans()
-#     workflow_span = spans[0]
-#     assert workflow_span.attributes["custom_span"] == "yes"
-
-
-# def test_instruments(exporter_with_custom_instrumentations):
-#     @workflow()
-#     def run_workflow():
-#         pass
-
-#     run_workflow()
-
-#     spans = exporter_with_custom_instrumentations.get_finished_spans()
-#     workflow_span = spans[0]
-#     assert workflow_span
-
-
-# def test_no_metrics(exporter_with_no_metrics):
-#     @workflow()
-#     def run_workflow():
-#         pass
-
-#     run_workflow()
-
-#     spans = exporter_with_no_metrics.get_finished_spans()
-#     workflow_span = spans[0]
-#     assert workflow_span

lmnr/traceloop_sdk/tests/test_tasks.py

@@ -1,32 +0,0 @@
-# import json
-# import pytest
-
-# from langchain_openai import ChatOpenAI
-# from lmnr.traceloop_sdk.decorators import task
-# from opentelemetry.semconv_ai import SpanAttributes
-
-
-# @pytest.mark.vcr
-# def test_task_io_serialization_with_langchain(exporter):
-#     @task(name="answer_question")
-#     def answer_question():
-#         chat = ChatOpenAI(temperature=0)
-
-#         return chat.invoke("Is Berlin the capital of Germany? Answer with yes or no")
-
-#     answer_question()
-
-#     spans = exporter.get_finished_spans()
-
-#     assert [span.name for span in spans] == [
-#         "ChatOpenAI.chat",
-#         "answer_question.task",
-#     ]
-
-#     task_span = next(span for span in spans if span.name == "answer_question.task")
-#     assert (
-#         json.loads(task_span.attributes.get(SpanAttributes.TRACELOOP_ENTITY_OUTPUT))[
-#             "kwargs"
-#         ]["content"]
-#         == "Yes"
-#     )

lmnr/traceloop_sdk/tests/test_workflows.py

@@ -1,262 +0,0 @@
-# import json
-
-# import pytest
-# from openai import OpenAI, AsyncOpenAI
-# from opentelemetry.semconv_ai import SpanAttributes
-# from lmnr.traceloop_sdk import Traceloop
-# from lmnr.traceloop_sdk.decorators import workflow, task, aworkflow, atask
-
-
-# @pytest.fixture
-# def openai_client():
-#     return OpenAI()
-
-
-# @pytest.fixture
-# def async_openai_client():
-#     return AsyncOpenAI()
-
-
-# # commented out because of the version parameter in task decorator
-# # @pytest.mark.vcr
-# # def test_simple_workflow(exporter, openai_client):
-# #     @task(name="something_creator", version=2)
-# #     def create_something(what: str, subject: str):
-# #         Traceloop.set_prompt(
-# #             "Tell me a {what} about {subject}", {"what": what, "subject": subject}, 5
-# #         )
-# #         completion = openai_client.chat.completions.create(
-# #             model="gpt-3.5-turbo",
-# #             messages=[{"role": "user", "content": f"Tell me a {what} about {subject}"}],
-# #         )
-# #         return completion.choices[0].message.content
-
-# #     @workflow(name="pirate_joke_generator", version=1)
-# #     def joke_workflow():
-# #         return create_something("joke", subject="OpenTelemetry")
-
-# #     joke = joke_workflow()
-
-# #     spans = exporter.get_finished_spans()
-# #     assert [span.name for span in spans] == [
-# #         "openai.chat",
-# #         "something_creator.task",
-# #         "pirate_joke_generator.workflow",
-# #     ]
-# #     open_ai_span = next(span for span in spans if span.name == "openai.chat")
-# #     assert (
-# #         open_ai_span.attributes[f"{SpanAttributes.LLM_PROMPTS}.0.content"]
-# #         == "Tell me a joke about OpenTelemetry"
-# #     )
-# #     assert open_ai_span.attributes.get(f"{SpanAttributes.LLM_COMPLETIONS}.0.content")
-# #     assert (
-# #         open_ai_span.attributes.get("traceloop.prompt.template")
-# #         == "Tell me a {what} about {subject}"
-# #     )
-# #     assert (
-# #         open_ai_span.attributes.get("traceloop.prompt.template_variables.what")
-# #         == "joke"
-# #     )
-# #     assert (
-# #         open_ai_span.attributes.get("traceloop.prompt.template_variables.subject")
-# #         == "OpenTelemetry"
-# #     )
-# #     assert open_ai_span.attributes.get("traceloop.prompt.version") == 5
-
-# #     workflow_span = next(
-# #         span for span in spans if span.name == "pirate_joke_generator.workflow"
-# #     )
-# #     task_span = next(span for span in spans if span.name == "something_creator.task")
-# #     assert json.loads(task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_INPUT]) == {
-# #         "args": ["joke"],
-# #         "kwargs": {"subject": "OpenTelemetry"},
-# #     }
-
-# #     assert (
-# #         json.loads(task_span.attributes.get(SpanAttributes.TRACELOOP_ENTITY_OUTPUT))
-# #         == joke
-# #     )
-# #     assert task_span.parent.span_id == workflow_span.context.span_id
-# #     assert (
-# #         workflow_span.attributes[SpanAttributes.TRACELOOP_ENTITY_NAME]
-# #         == "pirate_joke_generator"
-# #     )
-# #     assert workflow_span.attributes[SpanAttributes.TRACELOOP_ENTITY_VERSION] == 1
-# #     assert task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_VERSION] == 2
-
-
-# @pytest.mark.vcr
-# @pytest.mark.asyncio
-# async def test_simple_aworkflow(exporter, async_openai_client):
-#     @atask(name="something_creator", version=2)
-#     async def create_something(what: str, subject: str):
-#         Traceloop.set_prompt(
-#             "Tell me a {what} about {subject}", {"what": what, "subject": subject}, 5
-#         )
-#         completion = await async_openai_client.chat.completions.create(
-#             model="gpt-3.5-turbo",
-#             messages=[{"role": "user", "content": f"Tell me a {what} about {subject}"}],
-#         )
-#         return completion.choices[0].message.content
-
-#     @aworkflow(name="pirate_joke_generator", version=1)
-#     async def joke_workflow():
-#         return await create_something("joke", subject="OpenTelemetry")
-
-#     joke = await joke_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == [
-#         "openai.chat",
-#         "something_creator.task",
-#         "pirate_joke_generator.workflow",
-#     ]
-#     open_ai_span = next(span for span in spans if span.name == "openai.chat")
-#     assert (
-#         open_ai_span.attributes[f"{SpanAttributes.LLM_PROMPTS}.0.content"]
-#         == "Tell me a joke about OpenTelemetry"
-#     )
-#     assert open_ai_span.attributes.get(f"{SpanAttributes.LLM_COMPLETIONS}.0.content")
-#     assert (
-#         open_ai_span.attributes.get("traceloop.prompt.template")
-#         == "Tell me a {what} about {subject}"
-#     )
-#     assert (
-#         open_ai_span.attributes.get("traceloop.prompt.template_variables.what")
-#         == "joke"
-#     )
-#     assert (
-#         open_ai_span.attributes.get("traceloop.prompt.template_variables.subject")
-#         == "OpenTelemetry"
-#     )
-#     assert open_ai_span.attributes.get("traceloop.prompt.version") == 5
-
-#     workflow_span = next(
-#         span for span in spans if span.name == "pirate_joke_generator.workflow"
-#     )
-#     task_span = next(span for span in spans if span.name == "something_creator.task")
-#     assert json.loads(task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_INPUT]) == {
-#         "args": ["joke"],
-#         "kwargs": {"subject": "OpenTelemetry"},
-#     }
-
-#     assert (
-#         json.loads(task_span.attributes.get(SpanAttributes.TRACELOOP_ENTITY_OUTPUT))
-#         == joke
-#     )
-#     assert task_span.parent.span_id == workflow_span.context.span_id
-#     assert (
-#         workflow_span.attributes[SpanAttributes.TRACELOOP_ENTITY_NAME]
-#         == "pirate_joke_generator"
-#     )
-#     assert workflow_span.attributes[SpanAttributes.TRACELOOP_ENTITY_VERSION] == 1
-#     assert task_span.attributes[SpanAttributes.TRACELOOP_ENTITY_VERSION] == 2
-
-
-# @pytest.mark.vcr
-# def test_streaming_workflow(exporter, openai_client):
-
-#     @task(name="pirate_joke_generator")
-#     def joke_task():
-#         response_stream = openai_client.chat.completions.create(
-#             model="gpt-3.5-turbo",
-#             messages=[
-#                 {"role": "user", "content": "Tell me a joke about OpenTelemetry"}
-#             ],
-#             stream=True,
-#         )
-#         for chunk in response_stream:
-#             yield chunk
-
-#     @task(name="joke_runner")
-#     def joke_runner():
-#         res = joke_task()
-#         return res
-
-#     @workflow(name="joke_manager")
-#     def joke_workflow():
-#         res = joke_runner()
-#         for chunk in res:
-#             pass
-
-#     joke_workflow()
-
-#     spans = exporter.get_finished_spans()
-#     assert set([span.name for span in spans]) == set(
-#         [
-#             "openai.chat",
-#             "pirate_joke_generator.task",
-#             "joke_runner.task",
-#             "joke_manager.workflow",
-#         ]
-#     )
-#     generator_span = next(
-#         span for span in spans if span.name == "pirate_joke_generator.task"
-#     )
-#     runner_span = next(span for span in spans if span.name == "joke_runner.task")
-#     manager_span = next(span for span in spans if span.name == "joke_manager.workflow")
-#     openai_span = next(span for span in spans if span.name == "openai.chat")
-
-#     assert openai_span.parent.span_id == generator_span.context.span_id
-#     assert generator_span.parent.span_id == runner_span.context.span_id
-#     assert runner_span.parent.span_id == manager_span.context.span_id
-#     assert openai_span.end_time <= manager_span.end_time
-
-
-# def test_unrelated_entities(exporter):
-#     @workflow(name="workflow_1")
-#     def workflow_1():
-#         return
-
-#     @task(name="task_1")
-#     def task_1():
-#         return
-
-#     workflow_1()
-#     task_1()
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == ["workflow_1.workflow", "task_1.task"]
-
-#     workflow_1_span = spans[0]
-#     task_1_span = spans[1]
-
-#     assert (
-#         workflow_1_span.attributes[SpanAttributes.TRACELOOP_ENTITY_NAME] == "workflow_1"
-#     )
-#     assert workflow_1_span.attributes[SpanAttributes.TRACELOOP_SPAN_KIND] == "workflow"
-
-#     assert task_1_span.attributes[SpanAttributes.TRACELOOP_ENTITY_NAME] == "task_1"
-#     assert task_1_span.attributes[SpanAttributes.TRACELOOP_SPAN_KIND] == "task"
-#     assert task_1_span.parent is None
-
-
-# def test_unserializable_workflow(exporter):
-#     @task(name="unserializable_task")
-#     def unserializable_task(obj: object):
-#         return object()
-
-#     @workflow(name="unserializable_workflow")
-#     def unserializable_workflow(obj: object):
-#         return unserializable_task(obj)
-
-#     unserializable_task(object())
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == ["unserializable_task.task"]
-
-
-# @pytest.mark.asyncio
-# async def test_unserializable_async_workflow(exporter):
-#     @atask(name="unserializable_task")
-#     async def unserializable_task(obj: object):
-#         return object()
-
-#     @aworkflow(name="unserializable_workflow")
-#     async def unserializable_workflow(obj: object):
-#         return await unserializable_task(obj)
-
-#     await unserializable_task(object())
-
-#     spans = exporter.get_finished_spans()
-#     assert [span.name for span in spans] == ["unserializable_task.task"]

lmnr/traceloop_sdk/tracing/__init__.py

@@ -1 +0,0 @@
-from lmnr.traceloop_sdk.tracing.context_manager import get_tracer

lmnr-0.4.36.dist-info/RECORD

@@ -1,52 +0,0 @@
-lmnr/__init__.py,sha256=bIkXw003GaigTxNfGE_gwVtTDGdSdAPUr6CS_nN51oA,430
-lmnr/cli.py,sha256=Ptvm5dsNLKUY5lwnN8XkT5GtCYjzpRNi2WvefknB3OQ,1079
-lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lmnr/sdk/datasets.py,sha256=w8U9E6fvetAo65Cb2CbYzlfhY8CfXAR-VysrakG6-4I,1591
-lmnr/sdk/decorators.py,sha256=ZSDaEZyjo-RUzRCltsNbe6x0t9SKl2xRQ2q4uaKvXtk,2250
-lmnr/sdk/evaluations.py,sha256=5Vfyp0aIjuGpqfuM3cqsaaLpcoO7z6lcOOKxnyHCNHk,16264
-lmnr/sdk/laminar.py,sha256=r8X5SNHr3hvF3VIFRH5HhEzqdi4D6ZsXpv5WyMtPy3c,27429
-lmnr/sdk/log.py,sha256=cZBeUoSK39LMEV-X4-eEhTWOciULRfHaKfRK8YqIM8I,1532
-lmnr/sdk/types.py,sha256=qGD1tkGszd-_sZJaZ_Zx9U_CdUYzoDkUeN2g-o48Gls,5588
-lmnr/sdk/utils.py,sha256=Uk8y15x-sd5tP2ERONahElLDJVEy_3dA_1_5g9A6auY,3358
-lmnr/traceloop_sdk/.flake8,sha256=bCxuDlGx3YQ55QHKPiGJkncHanh9qGjQJUujcFa3lAU,150
-lmnr/traceloop_sdk/.python-version,sha256=9OLQBQVbD4zE4cJsPePhnAfV_snrPSoqEQw-PXgPMOs,6
-lmnr/traceloop_sdk/__init__.py,sha256=AVTqhuom40hx34zNB8TvqtfPliKYo7EGi08fw3Rs670,2384
-lmnr/traceloop_sdk/config/__init__.py,sha256=DliMGp2NjYAqRFLKpWQPUKjGMHRO8QsVfazBA1qENQ8,248
-lmnr/traceloop_sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lmnr/traceloop_sdk/decorators/base.py,sha256=SeImd8SAeqdC8LV1bvoQUq0d_TLkOLyd5HCe7vXKhNU,5559
-lmnr/traceloop_sdk/instruments.py,sha256=CGGUEELldrXkQwAzAkDeAtDq07_pjhz7i14a92P7C_E,1036
-lmnr/traceloop_sdk/tests/__init__.py,sha256=RYnG0-8zbXL0-2Ste1mEBf5sN4d_rQjGTCgPBuaZC74,20
-lmnr/traceloop_sdk/tests/cassettes/test_association_properties/test_langchain_and_external_association_properties.yaml,sha256=26g0wRA0juicHg_XrhcE8H4vhs1lawDs0o0aLFn-I7w,3103
-lmnr/traceloop_sdk/tests/cassettes/test_association_properties/test_langchain_association_properties.yaml,sha256=FNlSWlYCsWc3w7UPZzfGjDnxS3gAOhL-kpsu4BTxsDE,3061
-lmnr/traceloop_sdk/tests/cassettes/test_manual/test_manual_report.yaml,sha256=iq_U_DBKNyM8mhgwGZrqw1OMalIb3g1422cqkvsrPHw,2956
-lmnr/traceloop_sdk/tests/cassettes/test_manual/test_resource_attributes.yaml,sha256=IxAYRGTo46TtVsbxwLY2MPjgNehnvBVhzrH2ErHGxNA,2936
-lmnr/traceloop_sdk/tests/cassettes/test_privacy_no_prompts/test_simple_workflow.yaml,sha256=4KUVSMVLOMzubVXqwKY7ukhrQEXTtysa0RyPXydMu0c,5861
-lmnr/traceloop_sdk/tests/cassettes/test_prompt_management/test_prompt_management.yaml,sha256=ASroB5TiDVoJ1pUIQte4rajjlOs7XclIjDxPT9EpVmA,6142
-lmnr/traceloop_sdk/tests/cassettes/test_sdk_initialization/test_resource_attributes.yaml,sha256=gUQ1mlrDG-2qmp7E6nmQ8ma7T0EeYL6Ve0nIZkmkEJ0,5877
-lmnr/traceloop_sdk/tests/cassettes/test_tasks/test_task_io_serialization_with_langchain.yaml,sha256=OLtmIMglgFwNOIFSzZ5Xye6mO6PA5gPfHf2db6U_uPQ,2852
-lmnr/traceloop_sdk/tests/cassettes/test_workflows/test_simple_aworkflow.yaml,sha256=xbol_wTn-SeEanT4kLY4Y2_HLCTZf0ZHRFkPw402gqI,2944
-lmnr/traceloop_sdk/tests/cassettes/test_workflows/test_simple_workflow.yaml,sha256=ueiJY_6pyKQwbuMpeTAqHui4Ux7kYq_KTBZSob-cAjc,5866
-lmnr/traceloop_sdk/tests/cassettes/test_workflows/test_streaming_workflow.yaml,sha256=flOkiaW0DnQfD4rn_9F9dA0nIGMFjqwR3UzPyaanVjE,7947
-lmnr/traceloop_sdk/tests/conftest.py,sha256=XG6U9T6OyQliJUS8sBTXStjIP7Bb6ZChWTt0fulvHuc,3039
-lmnr/traceloop_sdk/tests/test_association_properties.py,sha256=D_LxhiVLwyXE0eW70uztmvZh3RRbVucM0L2C7O1oaFk,6735
-lmnr/traceloop_sdk/tests/test_manual.py,sha256=YHmVYfDfFUalAzYQsz50PIzXdj3Jfjcl4erpBU4ae4A,1759
-lmnr/traceloop_sdk/tests/test_nested_tasks.py,sha256=o7GPLVdgQOI5YH5HMBjInW4sprY4Nlu4SuwjMr1dxWU,1377
-lmnr/traceloop_sdk/tests/test_privacy_no_prompts.py,sha256=lS8bwg_xOS4gcwwzrrWflMkTSn8eZWyTmAxL3h3hZA0,1478
-lmnr/traceloop_sdk/tests/test_sdk_initialization.py,sha256=fRaf6lrxFzJIN94P1Tav_z_eywOsF5JXLPWzbJPMMyQ,1477
-lmnr/traceloop_sdk/tests/test_tasks.py,sha256=xlEx8BKp4yG83SCjK5WkPGfyC33JSrx4h8VyjVwGbgw,906
-lmnr/traceloop_sdk/tests/test_workflows.py,sha256=RVcfY3WAFIDZC15-aSua21aoQyYeWE7KypDyUsm-2EM,9372
-lmnr/traceloop_sdk/tracing/__init__.py,sha256=Ckq7zCM26VdJVB5tIZv0GTPyMZKyfso_KWD5yPHaqdo,66
-lmnr/traceloop_sdk/tracing/attributes.py,sha256=h970zmb7yTszzf2oHBfOY3cDYhE6O7LhkiHLqa_7x1k,1261
-lmnr/traceloop_sdk/tracing/content_allow_list.py,sha256=3feztm6PBWNelc8pAZUcQyEGyeSpNiVKjOaDk65l2ps,846
-lmnr/traceloop_sdk/tracing/context_manager.py,sha256=csVlB6kDmbgSPsROHwnddvGGblx55v6lJMRj0wsSMQM,304
-lmnr/traceloop_sdk/tracing/tracing.py,sha256=enZ6uRH9H9ZShPl4LK9B6ThzOZrBBK1Bap4l3NoqhQ8,29145
-lmnr/traceloop_sdk/utils/__init__.py,sha256=pNhf0G3vTd5ccoc03i1MXDbricSaiqCbi1DLWhSekK8,604
-lmnr/traceloop_sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
-lmnr/traceloop_sdk/utils/json_encoder.py,sha256=dK6b_axr70IYL7Vv-bu4wntvDDuyntoqsHaddqX7P58,463
-lmnr/traceloop_sdk/utils/package_check.py,sha256=Ki74WZME-ASF0fx7RXSIMsULGYUzI86sOINi1EGrc_Y,235
-lmnr/traceloop_sdk/version.py,sha256=OlatFEFA4ttqSSIiV8jdE-sq3KG5zu2hnC4B4mzWF3s,23
-lmnr-0.4.36.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.4.36.dist-info/METADATA,sha256=Kbcvfb28HQX4Qv35xfSnAOV0q7jlfHSKVnAl0dg-S7o,10322
-lmnr-0.4.36.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-lmnr-0.4.36.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.4.36.dist-info/RECORD,,