monocle-apptrace 0.3.0b3__py3-none-any.whl → 0.3.0b5__py3-none-any.whl

monocle_apptrace/instrumentation/metamodel/openai/entities/retrieval.py

@@ -0,0 +1,24 @@
+from monocle_apptrace.instrumentation.metamodel.openai import (
+    _helper,
+)
+from monocle_apptrace.instrumentation.common.utils import resolve_from_alias
+
+RETRIEVAL = {
+    "type": "retrieval",
+    "attributes": [
+        [
+            {
+                "_comment": "LLM Model",
+                "attribute": "name",
+                "accessor": lambda arguments: resolve_from_alias(arguments['kwargs'], ['model', 'model_name', 'endpoint_name', 'deployment_name'])
+            },
+            {
+                "attribute": "type",
+                "accessor": lambda arguments: 'model.embedding.' + resolve_from_alias(arguments['kwargs'], ['model', 'model_name', 'endpoint_name', 'deployment_name'])
+            }
+        ]
+    ],
+    "events": [
+
+    ]
+}

monocle_apptrace/instrumentation/metamodel/openai/methods.py

@@ -0,0 +1,25 @@
+from monocle_apptrace.instrumentation.common.wrapper import atask_wrapper, task_wrapper
+from monocle_apptrace.instrumentation.metamodel.openai.entities.inference import (
+    INFERENCE,
+)
+from monocle_apptrace.instrumentation.metamodel.openai.entities.retrieval import (
+    RETRIEVAL,
+)
+
+OPENAI_METHODS = [
+    {
+        "package": "openai.resources.chat.completions",
+        "object": "Completions",
+        "method": "create",
+        "wrapper_method": task_wrapper,
+        "span_name": "openai_inference",
+        "output_processor": INFERENCE
+    },
+    {
+        "package": "openai.resources.embeddings",
+        "object": "Embeddings",
+        "method": "create",
+        "wrapper_method": task_wrapper,
+        "output_processor": RETRIEVAL
+    }
+]

monocle_apptrace/instrumentation/metamodel/requests/__init__.py

@@ -0,0 +1,2 @@
+from os import environ
+allowed_urls = environ.get('MONOCLE_TRACE_PROPAGATATION_URLS', ' ').split(',')

monocle_apptrace/instrumentation/metamodel/requests/_helper.py

@@ -0,0 +1,31 @@
+import os
+from  monocle_apptrace.instrumentation.metamodel.requests import allowed_urls
+from opentelemetry.propagate import inject
+from monocle_apptrace.instrumentation.common.span_handler import SpanHandler
+
+def request_pre_task_processor(kwargs):
+    # add traceparent to the request headers in kwargs
+    if 'headers' not in kwargs:
+        headers = {}
+    else:
+        headers = kwargs['headers'].copy()
+    inject(headers)
+    kwargs['headers'] = headers
+
+def request_skip_span(kwargs) -> bool:
+    # add traceparent to the request headers in kwargs
+    if 'url' in kwargs:
+        url:str = kwargs['url']
+        for allowed_url in allowed_urls:
+            if url.startswith(allowed_url.strip()):
+                return False
+    return True
+
+class RequestSpanHandler(SpanHandler):
+
+    def pre_task_processing(self, to_wrap, wrapped, instance, args,kwargs, span):
+        request_pre_task_processor(kwargs)
+        super().pre_task_processing(to_wrap, wrapped, instance, args,kwargs,span)
+
+    def skip_span(self, to_wrap, wrapped, instance, args, kwargs) -> bool:
+        return request_skip_span(kwargs)

monocle_apptrace/instrumentation/metamodel/requests/methods.py

@@ -0,0 +1,12 @@
+from monocle_apptrace.instrumentation.common.wrapper import task_wrapper
+
+REQUESTS_METHODS = [
+    {
+        "package": "requests.sessions",
+        "object": "Session",
+        "method": "request",
+        "span_name": "http_requests",
+        "wrapper_method": task_wrapper,
+        "span_handler":"request_handler",
+    }
+]

{monocle_apptrace-0.3.0b3.dist-info → monocle_apptrace-0.3.0b5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: monocle_apptrace
-Version: 0.3.0b3
+Version: 0.3.0b5
 Summary: package with monocle genAI tracing
 Project-URL: Homepage, https://github.com/monocle2ai/monocle
 Project-URL: Issues, https://github.com/monocle2ai/monocle/issues
@@ -35,6 +35,7 @@ Requires-Dist: langchain-mistralai==0.1.13; extra == 'dev'
 Requires-Dist: langchain-openai==0.1.8; extra == 'dev'
 Requires-Dist: langchain==0.2.5; extra == 'dev'
 Requires-Dist: langchainhub==0.1.21; extra == 'dev'
+Requires-Dist: langgraph==0.2.68; extra == 'dev'
 Requires-Dist: llama-index-embeddings-huggingface==0.2.0; extra == 'dev'
 Requires-Dist: llama-index-llms-azure-openai==0.1.9; extra == 'dev'
 Requires-Dist: llama-index-llms-mistralai==0.1.20; extra == 'dev'

{monocle_apptrace-0.3.0b3.dist-info → monocle_apptrace-0.3.0b5.dist-info}/RECORD

@@ -1,28 +1,33 @@
 monocle_apptrace/README.md,sha256=T5NFC01bF8VR0oVnAX_n0bhsEtttwqfTxDNAe5Y_ivE,3765
 monocle_apptrace/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/__main__.py,sha256=wBwV0fpwIuj9XSorPRP1MpkHHkZPM9Tg-lIFj1nokkU,609
 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=WRcRnnN0gSL7khN-BZ9wonMKiWaHNFqnBZ-QyM6hMsg,2177
-monocle_apptrace/exporters/aws/s3_exporter.py,sha256=eq5SfZmaeyhtXeXYnJGAfT1vlaNGSyh1leGX9Rr7vQ4,6619
+monocle_apptrace/exporters/monocle_exporters.py,sha256=AxhZsTHjz2ZTuI-QOw1zk_bCKD899_EYyiEtCyAb1GA,2210
+monocle_apptrace/exporters/aws/s3_exporter.py,sha256=JMxtox61J6gUoEFsM1PJisBJPySMpm_U2Uv68WioKtE,7146
 monocle_apptrace/exporters/aws/s3_exporter_opendal.py,sha256=FvyW0KkAz0W_1g16C_ERmamg4fSreT-UXgLaN9URTVQ,5057
 monocle_apptrace/exporters/azure/blob_exporter.py,sha256=m7Hsw3OXlP2GOCQcdxf8LM6Fe12fZmih45x82Z12dbI,5597
 monocle_apptrace/exporters/azure/blob_exporter_opendal.py,sha256=h5xv7JU6YEXL4AKT2B1op3YsHoA0rNnLCGq8seoVRWs,6114
 monocle_apptrace/exporters/okahu/okahu_exporter.py,sha256=p2rjStwo0OMEdHWQt_QvREpUWXbDm5jGx3qXeYai4_M,4407
 monocle_apptrace/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monocle_apptrace/instrumentation/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-monocle_apptrace/instrumentation/common/constants.py,sha256=ySfwN5ZUtNFwm3ba5QWxBpLmZVuBRqVltuaBvpk-e3Y,1937
-monocle_apptrace/instrumentation/common/instrumentor.py,sha256=v79pmHDRf64G1Y0A6RMZlxbKizYRuNwlkJpRabESDiU,8386
-monocle_apptrace/instrumentation/common/span_handler.py,sha256=d5ycN3_OYkNm6_XKuAybMa-s1o6VrH92bEiR-Q_Hj24,7141
-monocle_apptrace/instrumentation/common/utils.py,sha256=nX3TErgaaYSlebYqQHuj4nqFyU51-nMrdIbQmO2vOTI,5357
-monocle_apptrace/instrumentation/common/wrapper.py,sha256=Q-7gyJpTtNpY1-khduT3vykvOewQ8qKzKQfYaZjho7s,2864
-monocle_apptrace/instrumentation/common/wrapper_method.py,sha256=rWiSdzhZrnWnNVVAZMetPboCH2HHSOZ7KgI7dPwGxV8,1629
+monocle_apptrace/instrumentation/common/constants.py,sha256=5h3UHq-5Wz9wmwETZi950XMHnV3rSH18yTDUTlQZnyw,2158
+monocle_apptrace/instrumentation/common/instrumentor.py,sha256=mHiNA2Lcb3S0tiXjI78bHYZUpzCZziw98Y-n2x1E1Ck,11100
+monocle_apptrace/instrumentation/common/span_handler.py,sha256=vvKZkrXuFXqo0JDljLgv1nlrSWg8huFamqUoUl-e9Hw,7784
+monocle_apptrace/instrumentation/common/utils.py,sha256=PiqvYlBaxoxTlJ48iW5Uw3RJ9DERVxMxIL7c6tcsiJc,9463
+monocle_apptrace/instrumentation/common/wrapper.py,sha256=jPIiPIecTQXsnR6SVs7KtPfjq37MuBi5CeeZLPVz03w,3860
+monocle_apptrace/instrumentation/common/wrapper_method.py,sha256=_j1JZ-K7Fo5UMuRypNui14n6Fc8zAMTS1zqslAfods4,2996
 monocle_apptrace/instrumentation/metamodel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monocle_apptrace/instrumentation/metamodel/botocore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-monocle_apptrace/instrumentation/metamodel/botocore/_helper.py,sha256=wK6yZ96csTOUaq5itQx0dS-vjDfyMu-AIPu3i3cXebs,4979
-monocle_apptrace/instrumentation/metamodel/botocore/methods.py,sha256=EYNYE8FolCoFgJU8q6Jkf_0CWDJ8kXIktX1eB5eTAIQ,404
+monocle_apptrace/instrumentation/metamodel/botocore/_helper.py,sha256=JIYtaN57OXKO9zPuxMZzDycJbgHgAQaQUkwuCI_SzF8,3744
+monocle_apptrace/instrumentation/metamodel/botocore/methods.py,sha256=LzmjbZjDWy7Uozc0chNjWG6PZhLngh_KJe5L6rw5rqI,452
 monocle_apptrace/instrumentation/metamodel/botocore/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monocle_apptrace/instrumentation/metamodel/botocore/entities/inference.py,sha256=JfTRmrxgU6e-b3dBbunWt5ObY_Ry_ZBYJBwKJB5UlJ8,2255
+monocle_apptrace/instrumentation/metamodel/botocore/handlers/botocore_span_handler.py,sha256=_KJMFFhuetg3HpjewS3tmwS0K__P6uExKtqhXCBT5ws,1347
+monocle_apptrace/instrumentation/metamodel/flask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/flask/_helper.py,sha256=AcQ5F6_IDmu9PXaeKKeiGIyq2I2YzA7wu1cvLzR-uyU,1175
+monocle_apptrace/instrumentation/metamodel/flask/methods.py,sha256=QkWHX4wKQf_GiJBHmiS9_JD2CiKMTCWMcig2dxAiKgU,340
 monocle_apptrace/instrumentation/metamodel/haystack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monocle_apptrace/instrumentation/metamodel/haystack/_helper.py,sha256=VgTrKn7rZMcv4OVdVEBI76G-5B0Rux4guiI6Nsso14s,4833
 monocle_apptrace/instrumentation/metamodel/haystack/methods.py,sha256=4WwhZoPQBkV42TpBvn-rXu37xtaBRrw7_VZB3MGrfxE,1434
@@ -30,19 +35,34 @@ monocle_apptrace/instrumentation/metamodel/haystack/entities/__init__.py,sha256=
 monocle_apptrace/instrumentation/metamodel/haystack/entities/inference.py,sha256=PkCaaar5hbZH7YGtWisq8dUJqBINsFGmtaUgt11UDa4,3019
 monocle_apptrace/instrumentation/metamodel/haystack/entities/retrieval.py,sha256=nq3lsk2qFxXqwrAHsBt8zrh4ZVGAJABkPtylrjUCCqc,2357
 monocle_apptrace/instrumentation/metamodel/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-monocle_apptrace/instrumentation/metamodel/langchain/_helper.py,sha256=7GixVwwzYR_Vcit1YN-xscA00HWFxOx7lcpqI94zi-w,4535
+monocle_apptrace/instrumentation/metamodel/langchain/_helper.py,sha256=g88Hz4n25ALJnjYFhdbdoIlSFUJUkN-8gho8ru7txEQ,4910
 monocle_apptrace/instrumentation/metamodel/langchain/methods.py,sha256=xEWO4uSiOnR221cvXESnVgAfC6JeExsP46ZkbK8_Yqs,3027
 monocle_apptrace/instrumentation/metamodel/langchain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monocle_apptrace/instrumentation/metamodel/langchain/entities/inference.py,sha256=wjW9hb1Qwr_aqY0lPehdPftyHtuvHinGxVmy0TVj5xo,2705
 monocle_apptrace/instrumentation/metamodel/langchain/entities/retrieval.py,sha256=r4UqTCT5vOfkbz9lwoTRoiMkUUJtPMwqOYbqo53A6K8,2039
+monocle_apptrace/instrumentation/metamodel/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/langgraph/_helper.py,sha256=-XmAbhkgqwaunFZa-BP0zWZ3e-uD-ihSszbn5Cz75yc,2043
+monocle_apptrace/instrumentation/metamodel/langgraph/methods.py,sha256=gnrKhcEPoy_qjyZWEkKZAUGTjRHvE3rqm3b4hQZoWMQ,453
+monocle_apptrace/instrumentation/metamodel/langgraph/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/langgraph/entities/inference.py,sha256=OaPeQ8pkyEP5j6ad537MTPp0BdDI7nabxf60u66Dzbk,1659
 monocle_apptrace/instrumentation/metamodel/llamaindex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-monocle_apptrace/instrumentation/metamodel/llamaindex/_helper.py,sha256=PnxHbVP_XdXTelGYpWqlPvhR4Tr5QcaVd7hdJ0LuA3I,5854
-monocle_apptrace/instrumentation/metamodel/llamaindex/methods.py,sha256=THr-nmeGRNwmHaHeYxNLAG7EQOHZTwTYX-0EdFjfFBk,2748
+monocle_apptrace/instrumentation/metamodel/llamaindex/_helper.py,sha256=5nqG-bSW3-ZEADZcwlHXIhhGZoTZu2a5Sc3Lo_AByeo,6199
+monocle_apptrace/instrumentation/metamodel/llamaindex/methods.py,sha256=r-o2zz9_ATfgKjUmIiaeSFQ774Vy4wGYueoVc-TqGMI,3061
 monocle_apptrace/instrumentation/metamodel/llamaindex/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/llamaindex/entities/agent.py,sha256=g7IEwFMLjYvxljX7iHoYSPJW6k-wC7Z3i_y2qlNEZcs,1338
 monocle_apptrace/instrumentation/metamodel/llamaindex/entities/inference.py,sha256=585hJXbdN2kFOnABv12vlzFkCbDExZln5ISvQI71EHw,2623
 monocle_apptrace/instrumentation/metamodel/llamaindex/entities/retrieval.py,sha256=QBF1nrqog5KHh925jiY2V-kejL6iVLKUowZmqUDoiJ4,1870
-monocle_apptrace-0.3.0b3.dist-info/METADATA,sha256=VVAF6k-HKzOqIX05cguZz9fmkySu6CKpNWr8ynwRo8I,6265
-monocle_apptrace-0.3.0b3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-monocle_apptrace-0.3.0b3.dist-info/licenses/LICENSE,sha256=ay9trLiP5I7ZsFXo6AqtkLYdRqe5S9r-DrPOvsNlZrg,9136
-monocle_apptrace-0.3.0b3.dist-info/licenses/NOTICE,sha256=9jn4xtwM_uUetJMx5WqGnhrR7MIhpoRlpokjSTlyt8c,112
-monocle_apptrace-0.3.0b3.dist-info/RECORD,,
+monocle_apptrace/instrumentation/metamodel/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/openai/_helper.py,sha256=nQSV8B000AfWjaIAKqkA5ZD5e-ozx3KIVTEBOYvoYxA,3066
+monocle_apptrace/instrumentation/metamodel/openai/methods.py,sha256=Ahmi0wo6UUHcOXz-6U_6EPZ2bEKXnu5o8NUVrikz0w0,759
+monocle_apptrace/instrumentation/metamodel/openai/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+monocle_apptrace/instrumentation/metamodel/openai/entities/inference.py,sha256=yu0a9DyBqGOGRQFcPgl1fOSCKu6Jc41Dxx9N0QuwVR0,2659
+monocle_apptrace/instrumentation/metamodel/openai/entities/retrieval.py,sha256=QICL56xZF0R84rCpf_Wj_1pMUOt6M8UzLRaICJnEQ7Y,755
+monocle_apptrace/instrumentation/metamodel/requests/__init__.py,sha256=A9M_SrBoVqjo2HENM9VtAdOzIme82lsoGt361bBKW84,101
+monocle_apptrace/instrumentation/metamodel/requests/_helper.py,sha256=lKU7py-M0eweHA_LWatwdyWbSGSlQNhScGZ43Xko7us,1115
+monocle_apptrace/instrumentation/metamodel/requests/methods.py,sha256=OJtosy_07xy01o5Qv-53--aCLQLkr82NZtyi2t6ZDEM,326
+monocle_apptrace-0.3.0b5.dist-info/METADATA,sha256=Tr4V0KMOc7nfKDTNDDE_OKc559DGJPULoacZEiG7EZE,6314
+monocle_apptrace-0.3.0b5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+monocle_apptrace-0.3.0b5.dist-info/licenses/LICENSE,sha256=ay9trLiP5I7ZsFXo6AqtkLYdRqe5S9r-DrPOvsNlZrg,9136
+monocle_apptrace-0.3.0b5.dist-info/licenses/NOTICE,sha256=9jn4xtwM_uUetJMx5WqGnhrR7MIhpoRlpokjSTlyt8c,112
+monocle_apptrace-0.3.0b5.dist-info/RECORD,,