openinference-instrumentation 0.1.0__tar.gz

openinference_instrumentation-0.1.0/.gitignore

@@ -0,0 +1,2 @@
+# vendored virtual environments
+.venv

openinference_instrumentation-0.1.0/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.3
+Name: openinference-instrumentation
+Version: 0.1.0
+Summary: OpenInference instrumentation utilities
+Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/openinference-instrumentation
+Author-email: OpenInference Authors <oss@arize.com>
+License-Expression: Apache-2.0
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: <3.13,>=3.8
+Requires-Dist: opentelemetry-api
+Provides-Extra: test
+Requires-Dist: opentelemetry-sdk; extra == 'test'
+Description-Content-Type: text/markdown
+
+# OpenInference Mistral AI Instrumentation
+[![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-mistralai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-mistralai) 
+
+Utility functions for OpenInference instrumentation.
+
+## Installation
+
+```shell
+pip install openinference-instrumentation
+```

openinference_instrumentation-0.1.0/README.md

@@ -0,0 +1,10 @@
+# OpenInference Mistral AI Instrumentation
+[![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-mistralai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-mistralai) 
+
+Utility functions for OpenInference instrumentation.
+
+## Installation
+
+```shell
+pip install openinference-instrumentation
+```

openinference_instrumentation-0.1.0/pyproject.toml

@@ -0,0 +1,49 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "openinference-instrumentation"
+dynamic = ["version"]
+description = "OpenInference instrumentation utilities"
+readme = "README.md"
+license = "Apache-2.0"
+requires-python = ">=3.8, <3.13"
+authors = [
+  { name = "OpenInference Authors", email = "oss@arize.com" },
+]
+classifiers = [
+  "Development Status :: 5 - Production/Stable",
+  "Intended Audience :: Developers",
+  "License :: OSI Approved :: Apache Software License",
+  "Programming Language :: Python",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.8",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+  "opentelemetry-api",
+]
+
+[project.optional-dependencies]
+test = [
+  "opentelemetry-sdk",
+]
+
+[project.urls]
+Homepage = "https://github.com/Arize-ai/openinference/tree/main/python/openinference-instrumentation"
+
+[tool.hatch.version]
+path = "src/openinference/instrumentation/version.py"
+
+[tool.hatch.build.targets.sdist]
+include = [
+  "/src",
+  "/tests",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/openinference"]

openinference_instrumentation-0.1.0/src/openinference/instrumentation/__init__.py

@@ -0,0 +1,22 @@
+import contextlib
+from typing import Iterator
+
+
+@contextlib.contextmanager
+def suppress_tracing() -> Iterator[None]:
+    """
+    Context manager to pause OpenTelemetry instrumentation.
+
+    Examples:
+        with suppress_tracing():
+            # No tracing will occur within this block
+            ...
+    """
+    try:
+        from opentelemetry.context import _SUPPRESS_INSTRUMENTATION_KEY, attach, detach, set_value
+    except ImportError:
+        yield
+        return
+    token = attach(set_value(_SUPPRESS_INSTRUMENTATION_KEY, True))
+    yield
+    detach(token)

openinference_instrumentation-0.1.0/src/openinference/instrumentation/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

openinference_instrumentation-0.1.0/tests/test_suppress_tracing.py

@@ -0,0 +1,29 @@
+from typing import Iterator
+
+import pytest
+from openinference.instrumentation import suppress_tracing
+from opentelemetry.context import (
+    _SUPPRESS_INSTRUMENTATION_KEY,
+    attach,
+    detach,
+    get_value,
+    set_value,
+)
+
+
+def test_suppress_tracing(obj: object) -> None:
+    with suppress_tracing():
+        assert get_value(_SUPPRESS_INSTRUMENTATION_KEY) is True
+    assert get_value(_SUPPRESS_INSTRUMENTATION_KEY) is obj
+
+
+@pytest.fixture(autouse=True)
+def instrument(obj: object) -> Iterator[None]:
+    token = attach(set_value(_SUPPRESS_INSTRUMENTATION_KEY, obj))
+    yield
+    detach(token)
+
+
+@pytest.fixture
+def obj() -> object:
+    return object()