monocle-apptrace 0.2.0__py3-none-any.whl → 0.3.0b1__py3-none-any.whl

{monocle_apptrace-0.2.0.dist-info → monocle_apptrace-0.3.0b1.dist-info}/RECORD

@@ -1,20 +1,20 @@
 monocle_apptrace/README.md,sha256=T5NFC01bF8VR0oVnAX_n0bhsEtttwqfTxDNAe5Y_ivE,3765
 monocle_apptrace/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-monocle_apptrace/constants.py,sha256=wjObbmMTFL201x-bf3EOXevYygwkFH_1ng5dDrpE3z0,810
-monocle_apptrace/instrumentor.py,sha256=FMQ8yVNYGRBTmyUjC2578G8RzSRpHY5HtVN5WK9ndVE,5291
-monocle_apptrace/utils.py,sha256=l-affXaMX6U_wG1rWRE2GPB6dDGtha7kL0MQK4PEptI,6395
-monocle_apptrace/wrap_common.py,sha256=ux2Ob7g1FYDtt1gF4g8XKQhl6Ed5X_iR4CP4GHX4upM,18178
-monocle_apptrace/wrapper.py,sha256=8bsMaCCCbaEJN8hW5YoYkn2XLJ7vFI1HJk6QHihbkto,830
-monocle_apptrace/exporters/base_exporter.py,sha256=1UvywEiKZ0nilcXiy-mZue012yVvYXgwib1x1qQvpBc,1647
+monocle_apptrace/constants.py,sha256=Q1rREmswflNn_ZeNhDGCOCpHXY6msaOqesS_LoXXBdg,1633
+monocle_apptrace/instrumentor.py,sha256=bTgN3pWz5cGyOBBEbiNt8O4W5QJAprUCk7JhvLdd_C8,5346
+monocle_apptrace/message_processing.py,sha256=NEg_F1epAAGNtwN_ht6TCKPnTMgx1da95-bamm-dZeE,3025
+monocle_apptrace/utils.py,sha256=Png2ucLNww_0_UFQxzdPs5xaEMqyezhIbjpbb5gy-GI,9764
+monocle_apptrace/wrap_common.py,sha256=DTb1SsqKNYWDodsbISJOy6WagRJxcN4h1F5KS3rXnW8,22465
+monocle_apptrace/wrapper.py,sha256=GVgImJasbHfrNxT8FLoMeZmGB_JRp09RjSs6Nn7OouE,904
+monocle_apptrace/botocore/__init__.py,sha256=SkRoVm6zFUHFG8JVFKhBXWsMHk2yOLpAxgq3mD-euY4,398
+monocle_apptrace/exporters/base_exporter.py,sha256=Gov_QKp5fonVZ-YdNM2ynoPot7GCaSNmKbCHIP3bDlE,1680
 monocle_apptrace/exporters/exporter_processor.py,sha256=BTcBgMuFLHCdCgVvc9TKIo9y8g1BvShI0L4vX6Q-cmk,393
 monocle_apptrace/exporters/file_exporter.py,sha256=gN9pJ_X5pcstVVsyivgHsjWhr443eRa6Y6Hx1rGLQAM,2280
 monocle_apptrace/exporters/monocle_exporters.py,sha256=mvVoGr4itvUBuakH17zNEuXYpHtqRhq2CBH5JXhleoM,1418
-monocle_apptrace/exporters/aws/s3_exporter.py,sha256=S_w2OBkM8eEYK6bBDbfcRV4LdojU1QygOBqjdhcjAl4,6730
-monocle_apptrace/exporters/azure/blob_exporter.py,sha256=dtL8MPISyP1eKM0gGpSoqTUmFuGEfycqF9OCjFF6H6E,5716
+monocle_apptrace/exporters/aws/s3_exporter.py,sha256=1WhZ_xCk1_c1vD8u2d_6z2QIedM89A-wsHGHLQsnGtY,6618
+monocle_apptrace/exporters/azure/blob_exporter.py,sha256=m7Hsw3OXlP2GOCQcdxf8LM6Fe12fZmih45x82Z12dbI,5597
 monocle_apptrace/exporters/okahu/okahu_exporter.py,sha256=p2rjStwo0OMEdHWQt_QvREpUWXbDm5jGx3qXeYai4_M,4407
 monocle_apptrace/haystack/__init__.py,sha256=zMvF6Dh1-vVIAQD__HC1ubT5bs-EupUghg7HNhDv7_c,448
-monocle_apptrace/haystack/wrap_node.py,sha256=IK07Wn3Lk1Os9URsyrmB1HXOH2FNdzK9fNLlR8TZdYc,908
-monocle_apptrace/haystack/wrap_openai.py,sha256=Yp916DhOl0WI6virRi3L43snfsQm7PhI28wlDsg19v8,1536
 monocle_apptrace/haystack/wrap_pipeline.py,sha256=yZAw7Hdv7FXe6rrM7gA2y5SjaZYQZCAi0q-R-uqUEvk,2254
 monocle_apptrace/langchain/__init__.py,sha256=3yhbdw1h9I1nVEfnOOPKz9yD5NqwdLSZsxtXbMplRkw,400
 monocle_apptrace/llamaindex/__init__.py,sha256=TgV1ZM8Cz113pZ2aAgpYwsCJyHA2aHOvNoW1QMBp0mM,709
@@ -26,19 +26,23 @@ monocle_apptrace/metamodel/entities/inference_types.json,sha256=Gkq7qNw5Qn91tuq-
 monocle_apptrace/metamodel/entities/model_types.json,sha256=Oz9DzuX_ptpLEVTc1Epv4_Y80TrYxR7cyc9kM0Zk780,785
 monocle_apptrace/metamodel/entities/vector_store_types.json,sha256=EA0KayHMOG7wMIkec54en03_3yT1qpdGh5TiDrAoh4g,462
 monocle_apptrace/metamodel/entities/workflow_types.json,sha256=eD0W3_FocKrzrrW0bks0hIJb9Kj7w8c1dpXzxmGLOwk,386
-monocle_apptrace/metamodel/maps/haystack_methods.json,sha256=JmngkaKICAzOyrWNTWEOLYFrp99l5wcERYKE_SQRNxE,698
+monocle_apptrace/metamodel/maps/botocore_methods.json,sha256=FvobptTS-whE5fI14-3QGqGErLijmUshIvzJWF1-6fE,338
+monocle_apptrace/metamodel/maps/haystack_methods.json,sha256=yWOs2LHgrDpwhGC2aVJqNOLt0kPwlzZdzMb2yzHP1QI,1611
 monocle_apptrace/metamodel/maps/langchain_methods.json,sha256=lfU8nd6td5qzTI01glM063Vg0UoZM9HGPirwN4OZcKc,4285
-monocle_apptrace/metamodel/maps/llamaindex_methods.json,sha256=pZ4d1DS4D6Wgpz9EBOrEuJUFN58jtu75E8ELyGRBHyM,2636
-monocle_apptrace/metamodel/maps/attributes/inference/langchain_entities.json,sha256=E_K3eHv_e9Va2AcjNLHcXeAgePB1uolTMdRnkxxtNqU,1152
-monocle_apptrace/metamodel/maps/attributes/inference/llamaindex_entities.json,sha256=LuAa9ZKl3mO2VUiXMVGZNEfSHOZ0n5qsVQLqWRHczVM,1122
-monocle_apptrace/metamodel/maps/attributes/retrieval/langchain_entities.json,sha256=vSe2kbeJ7izmj5TBdQpX5pJ3hbjdY4SkcXvWPiam7oo,718
-monocle_apptrace/metamodel/maps/attributes/retrieval/llamaindex_entities.json,sha256=BtWeSXUPzKR9-vuhZUUhn2Gz9Y4SXBZoFC81K1q0XwQ,712
+monocle_apptrace/metamodel/maps/llamaindex_methods.json,sha256=dqfDQ4H8qB-GvctBzrgH0ldjdarbTRV0ID5a4LerN5I,3328
+monocle_apptrace/metamodel/maps/attributes/inference/botocore_entities.json,sha256=GCFQPlTTlE64dFc9DPM1sf3Uk1x6g0UdsrABlyFwl6k,936
+monocle_apptrace/metamodel/maps/attributes/inference/haystack_entities.json,sha256=e-0SOcP6b4A1hRYtDBQT3Mx94J73xqQNqdvqWyJSFmo,1756
+monocle_apptrace/metamodel/maps/attributes/inference/langchain_entities.json,sha256=QZJkP4JZ0VmWWtegCxFXASO8bJSYtJTz2ObpWzSyF7o,1826
+monocle_apptrace/metamodel/maps/attributes/inference/llamaindex_entities.json,sha256=-0ARREkmWRQLbFAufiBIAbPhunuUYVBTUoP5lNJrk90,1755
+monocle_apptrace/metamodel/maps/attributes/retrieval/haystack_entities.json,sha256=_SNwWIoO4anww-UDrTFIvDeFaHJKmg3R8IzZLbRGBD8,1035
+monocle_apptrace/metamodel/maps/attributes/retrieval/langchain_entities.json,sha256=-Dj7U0of2TmPjhYZ2SAMztQBHeeN81H-_4rTCfoAzyQ,916
+monocle_apptrace/metamodel/maps/attributes/retrieval/llamaindex_entities.json,sha256=opfmuUWizZvfTD7FVxG21OrpHXc8HcL32En9ftC3eRA,903
 monocle_apptrace/metamodel/spans/README.md,sha256=_uMkLLaWitQ_rPh7oQbW5Oe7uGSv2h_QA6YwxHRJi74,5433
 monocle_apptrace/metamodel/spans/span_example.json,sha256=7ZAUssL1UYXlbKHre8PfeNlIhHnM28e2CH9EyHDTOt8,4402
 monocle_apptrace/metamodel/spans/span_format.json,sha256=GhfioGgMhG7St0DeYA1fgNtMkbr9wiQ1L2hovekRQ24,1512
 monocle_apptrace/metamodel/spans/span_types.json,sha256=jwVyPbYMhEf2Ea6Egmb3m1Za28ap9dgZIpJDhdE1BlY,361
-monocle_apptrace-0.2.0.dist-info/METADATA,sha256=zaIOrJwa-RLX1bWeBEfTCfeBGC2cuNeXKwPHAYYNRT8,5364
-monocle_apptrace-0.2.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-monocle_apptrace-0.2.0.dist-info/licenses/LICENSE,sha256=ay9trLiP5I7ZsFXo6AqtkLYdRqe5S9r-DrPOvsNlZrg,9136
-monocle_apptrace-0.2.0.dist-info/licenses/NOTICE,sha256=9jn4xtwM_uUetJMx5WqGnhrR7MIhpoRlpokjSTlyt8c,112
-monocle_apptrace-0.2.0.dist-info/RECORD,,
+monocle_apptrace-0.3.0b1.dist-info/METADATA,sha256=PSSgPDB9-1HHTIw9WpIzuN3s2_B_a_fNgrD5EtH5UQs,5503
+monocle_apptrace-0.3.0b1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+monocle_apptrace-0.3.0b1.dist-info/licenses/LICENSE,sha256=ay9trLiP5I7ZsFXo6AqtkLYdRqe5S9r-DrPOvsNlZrg,9136
+monocle_apptrace-0.3.0b1.dist-info/licenses/NOTICE,sha256=9jn4xtwM_uUetJMx5WqGnhrR7MIhpoRlpokjSTlyt8c,112
+monocle_apptrace-0.3.0b1.dist-info/RECORD,,

{monocle_apptrace-0.2.0.dist-info → monocle_apptrace-0.3.0b1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.25.0
+Generator: hatchling 1.26.3
 Root-Is-Purelib: true
 Tag: py3-none-any

monocle_apptrace/haystack/wrap_node.py

@@ -1,27 +0,0 @@
-
-
-import logging
-from opentelemetry import context as context_api
-from opentelemetry.context import attach, set_value
-from opentelemetry.instrumentation.utils import (
-    _SUPPRESS_INSTRUMENTATION_KEY,
-)
-from monocle_apptrace.wrap_common import WORKFLOW_TYPE_MAP, with_tracer_wrapper
-
-logger = logging.getLogger(__name__)
-
-
-@with_tracer_wrapper
-def wrap(tracer, to_wrap, wrapped, instance, args, kwargs):
-    if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
-        return wrapped(*args, **kwargs)
-    name = instance.name
-    attach(set_value("workflow_name", name))
-    with tracer.start_as_current_span(f"{name}.task") as span:
-        workflow_name = span.resource.attributes.get("service.name")
-        span.set_attribute("workflow_name",workflow_name)
-        span.set_attribute("workflow_type", WORKFLOW_TYPE_MAP["haystack"])
-
-        response = wrapped(*args, **kwargs)
-
-    return response

monocle_apptrace/haystack/wrap_openai.py

@@ -1,44 +0,0 @@
-import logging
-from opentelemetry import context as context_api
-from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
-from monocle_apptrace.wrap_common import with_tracer_wrapper
-from monocle_apptrace.utils import (
-    dont_throw,
-    set_span_attribute
-)
-
-logger = logging.getLogger(__name__)
-
-@dont_throw
-def _set_input_attributes(span, kwargs, instance, args):
-    set_span_attribute(span, "llm_input", kwargs.get("prompt"))
-
-    if 'model' in instance.__dict__:
-        model_name = instance.__dict__.get("model")
-        set_span_attribute(span, "model_name", model_name)
-
-@dont_throw
-def _set_response_attributes(span, response):
-
-    if "meta" in response:
-        token_usage = response["meta"][0]["usage"]
-        set_span_attribute(span, "completion_tokens", token_usage.get("completion_tokens"))
-        set_span_attribute(span, "prompt_tokens", token_usage.get("prompt_tokens"))
-        set_span_attribute(span, "total_tokens", token_usage.get("total_tokens"))
-
-
-@with_tracer_wrapper
-def wrap_openai(tracer, to_wrap, wrapped, instance, args, kwargs):
-    if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
-        return wrapped(*args, **kwargs)
-
-    with tracer.start_as_current_span("haystack.openai") as span:
-        if span.is_recording():
-            _set_input_attributes(span, kwargs, instance, args)
-        response = wrapped(*args, **kwargs)
-
-        if response:
-            if span.is_recording():
-                _set_response_attributes(span, response)
-
-        return response