lmnr 0.6.15__py3-none-any.whl → 0.6.17__py3-none-any.whl

lmnr/opentelemetry_lib/opentelemetry/instrumentation/skyvern/__init__.py

@@ -0,0 +1,193 @@
+from lmnr.opentelemetry_lib.decorators import json_dumps
+from lmnr.sdk.browser.utils import with_tracer_wrapper
+from lmnr.sdk.utils import get_input_from_func_args
+from lmnr.version import __version__
+
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from opentelemetry.instrumentation.utils import unwrap
+from opentelemetry.trace import get_tracer, Tracer
+from typing import Collection
+from wrapt import wrap_function_wrapper
+import pydantic
+
+try:
+    from skyvern import Skyvern
+except ImportError as e:
+    raise ImportError(
+        f"Attempted to import {__file__}, but it is designed "
+        "to patch Skyvern, which is not installed. Use `pip install skyvern` "
+        "to install Skyvern or remove this import."
+    ) from e
+
+_instruments = ("skyvern >= 0.1.0",)
+
+WRAPPED_METHODS = [
+    {
+        "package": "skyvern.library.skyvern",
+        "object": "Skyvern",  # Class name
+        "method": "run_task", # Method name
+        "span_name": "Skyvern.run_task",
+        "span_type": "DEFAULT",
+    },
+    {
+        "package": "skyvern.webeye.scraper.scraper",
+        # No "object" field for module-level functions
+        "method": "get_interactable_element_tree", # Function name
+        "span_name": "get_interactable_element_tree",
+        "span_type": "DEFAULT",
+    },
+    {
+        "package": "skyvern.forge.agent",
+        "object": "ForgeAgent",
+        "method": "execute_step",
+        "span_name": "ForgeAgent.execute_step",
+        "span_type": "DEFAULT",
+    },
+    {  
+        "package": "skyvern.services.task_v2_service",  
+        "method": "initialize_task_v2",  
+        "span_name": "initialize_task_v2",  
+        "span_type": "DEFAULT",  
+    },
+    {  
+        "package": "skyvern.services.task_v2_service",  
+        "method": "run_task_v2_helper",  
+        "span_name": "run_task_v2_helper",  
+        "span_type": "DEFAULT",  
+    },
+    {  
+        "package": "skyvern.forge.sdk.workflow.models.block",
+        "object": "Block",
+        "method": "_generate_workflow_run_block_description",  
+        "span_name": "Block._generate_workflow_run_block_description",  
+        "span_type": "DEFAULT",  
+    },       
+    {  
+        "package": "skyvern.webeye.actions.handler",  
+        "method": "extract_information_for_navigation_goal",  
+        "span_name": "extract_information_for_navigation_goal",  
+        "span_type": "DEFAULT",  
+    },  
+]
+
+
+@with_tracer_wrapper
+async def _wrap(tracer: Tracer, to_wrap, wrapped, instance, args, kwargs):
+    span_name = to_wrap.get("span_name")
+    attributes = {
+        "lmnr.span.type": to_wrap.get("span_type"),
+    }
+    
+    attributes["lmnr.span.input"] = json_dumps(
+        get_input_from_func_args(wrapped, True, args, kwargs)
+    )
+    
+    with tracer.start_as_current_span(span_name, attributes=attributes) as span:
+        try:  
+            result = await wrapped(*args, **kwargs)  
+                
+            to_serialize = result  
+            serialized = (  
+                to_serialize.model_dump_json()  
+                if isinstance(to_serialize, pydantic.BaseModel)  
+                else json_dumps(to_serialize)  
+            )  
+            span.set_attribute("lmnr.span.output", serialized)  
+            return result  
+                
+        except Exception as e:  
+            span.record_exception(e)  
+            raise
+
+
+def instrument_llm_handler(tracer: Tracer):  
+    from skyvern.forge import app  
+      
+    # Store the original handler  
+    original_handler = app.LLM_API_HANDLER  
+      
+    async def wrapped_llm_handler(*args, **kwargs):  
+
+        prompt_name = kwargs.get("prompt_name", "")
+
+        if prompt_name:
+            span_name = f"{prompt_name}"
+        else:
+            span_name = "app.LLM_API_HANDLER"
+
+        attributes = {  
+            "lmnr.span.type": "DEFAULT",  
+        }  
+
+        with tracer.start_as_current_span(span_name, attributes=attributes) as span:  
+            try:  
+                result = await original_handler(*args, **kwargs)  
+
+                to_serialize = result
+                serialized = (
+                    to_serialize.model_dump_json()
+                    if isinstance(to_serialize, pydantic.BaseModel)
+                    else json_dumps(to_serialize)
+                )
+                span.set_attribute("lmnr.span.output", serialized)  
+                return result  
+            except Exception as e:  
+                span.record_exception(e)  
+                raise
+      
+    # Replace the global handler  
+    app.LLM_API_HANDLER = wrapped_llm_handler
+
+
+class SkyvernInstrumentor(BaseInstrumentor):
+    def __init__(self):
+        super().__init__()
+
+    def instrumentation_dependencies(self) -> Collection[str]:
+        return _instruments
+
+    def _instrument(self, **kwargs):
+
+        tracer_provider = kwargs.get("tracer_provider")
+        tracer = get_tracer(__name__, __version__, tracer_provider)
+
+        instrument_llm_handler(tracer)
+
+        for wrapped_method in WRAPPED_METHODS:
+            wrap_package = wrapped_method.get("package")
+            wrap_object = wrapped_method.get("object")
+            wrap_method = wrapped_method.get("method")
+
+            # For class methods: "Class.method", for module functions: just "function_name"
+            if wrap_object:
+                target = f"{wrap_object}.{wrap_method}"
+            else:
+                target = wrap_method
+
+            try:
+                wrap_function_wrapper(
+                    wrap_package,
+                    target,
+                    _wrap(
+                        tracer,
+                        wrapped_method,
+                    ),
+                )
+            except ModuleNotFoundError:
+                pass  # that's ok, we're not instrumenting everything
+
+    def _uninstrument(self, **kwargs):
+
+        for wrapped_method in WRAPPED_METHODS:
+            wrap_package = wrapped_method.get("package")
+            wrap_object = wrapped_method.get("object")
+            wrap_method = wrapped_method.get("method")
+
+            # For class methods: "package.Class", for module functions: just "package"
+            if wrap_object:
+                module_path = f"{wrap_package}.{wrap_object}"
+            else:
+                module_path = wrap_package
+
+            unwrap(module_path, wrap_method)
+

lmnr/opentelemetry_lib/tracing/_instrument_initializers.py

@@ -345,6 +345,14 @@ class SageMakerInstrumentorInitializer(InstrumentorInitializer):
 
         return SageMakerInstrumentor()
 
+class SkyvernInstrumentorInitializer(InstrumentorInitializer):
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
+        if not is_package_installed("skyvern"):
+            return None
+
+        from ..opentelemetry.instrumentation.skyvern import SkyvernInstrumentor
+
+        return SkyvernInstrumentor()
 
 class TogetherInstrumentorInitializer(InstrumentorInitializer):
     def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:

lmnr/opentelemetry_lib/tracing/instruments.py

@@ -40,13 +40,13 @@ class Instruments(Enum):
     QDRANT = "qdrant"
     REPLICATE = "replicate"
     SAGEMAKER = "sagemaker"
+    SKYVERN = "skyvern"
     TOGETHER = "together"
     TRANSFORMERS = "transformers"
     VERTEXAI = "vertexai"
     WATSONX = "watsonx"
     WEAVIATE = "weaviate"
 
-
 INSTRUMENTATION_INITIALIZERS: dict[
     Instruments, initializers.InstrumentorInitializer
 ] = {
@@ -77,6 +77,7 @@ INSTRUMENTATION_INITIALIZERS: dict[
     Instruments.QDRANT: initializers.QdrantInstrumentorInitializer(),
     Instruments.REPLICATE: initializers.ReplicateInstrumentorInitializer(),
     Instruments.SAGEMAKER: initializers.SageMakerInstrumentorInitializer(),
+    Instruments.SKYVERN: initializers.SkyvernInstrumentorInitializer(),
     Instruments.TOGETHER: initializers.TogetherInstrumentorInitializer(),
     Instruments.TRANSFORMERS: initializers.TransformersInstrumentorInitializer(),
     Instruments.VERTEXAI: initializers.VertexAIInstrumentorInitializer(),

lmnr/sdk/browser/browser_use_otel.py

@@ -76,9 +76,16 @@ async def _wrap(tracer: Tracer, to_wrap, wrapped, instance, args, kwargs):
         )
     else:
         if not to_wrap.get("ignore_input"):
-            attributes["lmnr.span.input"] = json_dumps(
-                get_input_from_func_args(wrapped, True, args, kwargs)
-            )
+            inp_dict = get_input_from_func_args(wrapped, True, args, kwargs)
+            # Add task to the `agent.run` span input
+            if to_wrap.get("method") == "run" and hasattr(instance, "task"):
+                inp_dict["task"] = instance.task
+            attributes["lmnr.span.input"] = json_dumps(inp_dict)
+    if to_wrap.get("method") == "step" and to_wrap.get("object") == "Agent":
+        # Add step number to the `agent.step` span name
+        step_info = kwargs.get("step_info", args[0] if len(args) > 0 else None)
+        if step_info and hasattr(step_info, "step_number"):
+            span_name = f"agent.step.{step_info.step_number}"
     with tracer.start_as_current_span(span_name, attributes=attributes) as span:
         span.set_attributes(attributes)
         result = await wrapped(*args, **kwargs)

lmnr/version.py

@@ -3,7 +3,7 @@ import httpx
 from packaging import version
 
 
-__version__ = "0.6.15"
+__version__ = "0.6.17"
 PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}"
 
 

{lmnr-0.6.15.dist-info → lmnr-0.6.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.6.15
+Version: 0.6.17
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai
@@ -40,7 +40,6 @@ Provides-Extra: transformers
 Provides-Extra: vertexai
 Provides-Extra: watsonx
 Provides-Extra: weaviate
-Requires-Dist: argparse (>=1.0)
 Requires-Dist: grpcio (>=1)
 Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: opentelemetry-api (>=1.33.0)
@@ -58,8 +57,8 @@ Requires-Dist: opentelemetry-instrumentation-cohere (>=0.40.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-cohere (>=0.40.12) ; extra == "cohere"
 Requires-Dist: opentelemetry-instrumentation-crewai (>=0.40.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-crewai (>=0.40.12) ; extra == "crewai"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.40.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.40.12) ; extra == "google-generativeai"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (<0.40.10) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (<0.40.10) ; extra == "google-generativeai"
 Requires-Dist: opentelemetry-instrumentation-groq (>=0.40.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-groq (>=0.40.12) ; extra == "groq"
 Requires-Dist: opentelemetry-instrumentation-haystack (>=0.40.12) ; extra == "all"

{lmnr-0.6.15.dist-info → lmnr-0.6.17.dist-info}/RECORD

@@ -10,12 +10,13 @@ lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/config.py,sha2
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/utils.py,sha256=8SSBliRoJtiZME5RDEwt90CI2BadKPHQrtV4p6bDy_0,7669
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/langgraph/__init__.py,sha256=Jyv9koZdGA4-oTaB7ATB7DaX7aNOY-3YOGL4wX0c7PM,3107
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/langgraph/utils.py,sha256=nf9sJZXnnts4gYZortEiDvwYjYqYJZTAT0zutuP_R6Y,1512
+lmnr/opentelemetry_lib/opentelemetry/instrumentation/skyvern/__init__.py,sha256=dk5P6Xn7CNeCFBmjzFw66JpmZLYm75KCWfjxdck56uo,6334
 lmnr/opentelemetry_lib/tracing/__init__.py,sha256=27QogAe-aHyrVr6b56-DduUm0KEE24K1aV2e1nouTNg,6007
-lmnr/opentelemetry_lib/tracing/_instrument_initializers.py,sha256=RYSp4PxrF8yfG5qy0ALpx1EArLFBPDLQd6H6YeYw184,14567
+lmnr/opentelemetry_lib/tracing/_instrument_initializers.py,sha256=WVmcVTL8SJPN4hvATqywM6nc9Sv7NzGMR7Ew7rWv2eg,14898
 lmnr/opentelemetry_lib/tracing/attributes.py,sha256=MvowVluXfCqSIC3Cvx3tWDqB0Cpr9bpSlY91qL4Iy74,1497
 lmnr/opentelemetry_lib/tracing/context_properties.py,sha256=aWbvMdWB4Q7uqc0GGSsjcRXMTcO18aWOaIZe3QyS_aA,2314
 lmnr/opentelemetry_lib/tracing/exporter.py,sha256=avjmH9hz6PXbMV2bnx7b9GuGC6flAUDwvcxoZMbTWgM,2082
-lmnr/opentelemetry_lib/tracing/instruments.py,sha256=dsStEEAFfZ70MHn6WcmRCdDn9Yg-aCvfA6VJs_IvOOs,5267
+lmnr/opentelemetry_lib/tracing/instruments.py,sha256=3gL60IhFj1C2cvEEqZGFpP5rjjAJeYg9QePXLmKifus,5362
 lmnr/opentelemetry_lib/tracing/processor.py,sha256=YJ_2j02oU_YZD8cuDltXkCU_fN1hcACI61X5qgCZd7s,3441
 lmnr/opentelemetry_lib/tracing/tracer.py,sha256=oNC6V8eFvuK3i5IWXsKDjEMFL_axeSov3L1fPevwuWM,476
 lmnr/opentelemetry_lib/utils/__init__.py,sha256=pNhf0G3vTd5ccoc03i1MXDbricSaiqCbi1DLWhSekK8,604
@@ -24,7 +25,7 @@ lmnr/opentelemetry_lib/utils/package_check.py,sha256=ST2BvDxt0bax-ih9F9Wf0jlKAEr
 lmnr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lmnr/sdk/browser/browser_use_otel.py,sha256=I3Qoc8y1hqTxRJtzh84qJKz1sz2s_b68YNEYxI0CTV4,4577
+lmnr/sdk/browser/browser_use_otel.py,sha256=L8fm_pOAi6XkuSED4hTIItzA_BUMYj7YvN7BgFsHZDA,5093
 lmnr/sdk/browser/patchright_otel.py,sha256=O7n1dB_Mw-_L70zi0zqpnFFiKuTPEEpO7ry1ia6IXWQ,5454
 lmnr/sdk/browser/playwright_otel.py,sha256=LFg1iJXbez-BEgEIY9eaO_2T2uR6SxyfFL46fHcS7Mg,13423
 lmnr/sdk/browser/pw_utils.py,sha256=PPVdeE6RV-n7de3ApuEVqjTXrbkJxtGOsZ47Ce-Nna0,10307
@@ -52,9 +53,9 @@ lmnr/sdk/laminar.py,sha256=oOVco_c9ZstT71HsquGsgbtFumXd2Ejz0rl_qpmMlTU,33996
 lmnr/sdk/log.py,sha256=nt_YMmPw1IRbGy0b7q4rTtP4Yo3pQfNxqJPXK3nDSNQ,2213
 lmnr/sdk/types.py,sha256=ZQp5SeYJNZsK3KrbSeXPY_xn6mGjW5mSw_i0Rd_Oa4k,12328
 lmnr/sdk/utils.py,sha256=yrcHIhoADf9lWH9qJWZMmkRWYvd0DuxPSLP3mY6YFw0,4327
-lmnr/version.py,sha256=ppiL3UqkmmEoQ2_ewtXVMKuuJFfy_MHBbQZS_SfT6uA,1322
-lmnr-0.6.15.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.6.15.dist-info/METADATA,sha256=howYh4k6EGuz9J-eJ6pOzmuApJCAjrvooy-e--7YM9I,15186
-lmnr-0.6.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lmnr-0.6.15.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.6.15.dist-info/RECORD,,
+lmnr/version.py,sha256=X2cJf0gdlak_gTkSX8K7hD1iv3rMr48NxBaEODOlzko,1322
+lmnr-0.6.17.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.6.17.dist-info/METADATA,sha256=dGCAXz4gY1LcOHUTGjR-cHU3hCeROzCQM4M1q0wlFPU,15152
+lmnr-0.6.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lmnr-0.6.17.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.6.17.dist-info/RECORD,,