respan-instrumentation-replicate 0.1.0__tar.gz

respan_instrumentation_replicate-0.1.0/PKG-INFO

@@ -0,0 +1,66 @@
+Metadata-Version: 2.4
+Name: respan-instrumentation-replicate
+Version: 0.1.0
+Summary: Respan instrumentation plugin for the Replicate Python SDK
+License: Apache 2.0
+Author: Respan
+Author-email: team@respan.ai
+Requires-Python: >=3.11,<3.14
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: opentelemetry-semantic-conventions-ai (>=0.4.1)
+Requires-Dist: replicate (>=1.0.0)
+Requires-Dist: respan-sdk (>=2.6.1)
+Requires-Dist: respan-tracing (>=2.17.0,<3.0.0)
+Description-Content-Type: text/markdown
+
+# Respan Replicate Instrumentation
+
+Trace the official `replicate` Python SDK with Respan.
+
+The instrumentation patches the Replicate client lifecycle and emits canonical
+Respan spans for `run`, `async_run`, `stream`, `async_stream`, prediction
+creation, prediction waiting, and prediction lookup/cancel operations.
+
+## Install
+
+```bash
+pip install respan-ai respan-instrumentation-replicate replicate
+```
+
+## Usage
+
+```python
+import os
+
+import replicate
+from respan import Respan, workflow
+from respan_instrumentation_replicate import ReplicateInstrumentor
+
+respan = Respan(
+    api_key=os.environ["RESPAN_API_KEY"],
+    instrumentations=[ReplicateInstrumentor()],
+)
+
+
+@workflow(name="replicate_quickstart.workflow")
+def run_prediction() -> str:
+    output = replicate.run(
+        "meta/meta-llama-3-8b-instruct",
+        input={"prompt": "Reply with one concise sentence about tracing."},
+    )
+    return "".join(str(chunk) for chunk in output) if not isinstance(output, str) else output
+
+
+print(run_prediction())
+respan.flush()
+respan.shutdown()
+```
+
+Use `Respan(..., customer_identifier=..., thread_identifier=..., metadata=...)`
+or `respan.propagate_attributes(...)` to attach Respan attributes to Replicate
+spans.
+

respan_instrumentation_replicate-0.1.0/README.md

@@ -0,0 +1,46 @@
+# Respan Replicate Instrumentation
+
+Trace the official `replicate` Python SDK with Respan.
+
+The instrumentation patches the Replicate client lifecycle and emits canonical
+Respan spans for `run`, `async_run`, `stream`, `async_stream`, prediction
+creation, prediction waiting, and prediction lookup/cancel operations.
+
+## Install
+
+```bash
+pip install respan-ai respan-instrumentation-replicate replicate
+```
+
+## Usage
+
+```python
+import os
+
+import replicate
+from respan import Respan, workflow
+from respan_instrumentation_replicate import ReplicateInstrumentor
+
+respan = Respan(
+    api_key=os.environ["RESPAN_API_KEY"],
+    instrumentations=[ReplicateInstrumentor()],
+)
+
+
+@workflow(name="replicate_quickstart.workflow")
+def run_prediction() -> str:
+    output = replicate.run(
+        "meta/meta-llama-3-8b-instruct",
+        input={"prompt": "Reply with one concise sentence about tracing."},
+    )
+    return "".join(str(chunk) for chunk in output) if not isinstance(output, str) else output
+
+
+print(run_prediction())
+respan.flush()
+respan.shutdown()
+```
+
+Use `Respan(..., customer_identifier=..., thread_identifier=..., metadata=...)`
+or `respan.propagate_attributes(...)` to attach Respan attributes to Replicate
+spans.

respan_instrumentation_replicate-0.1.0/pyproject.toml

@@ -0,0 +1,29 @@
+[tool.poetry]
+name = "respan-instrumentation-replicate"
+version = "0.1.0"
+description = "Respan instrumentation plugin for the Replicate Python SDK"
+authors = ["Respan <team@respan.ai>"]
+license = "Apache 2.0"
+readme = "README.md"
+packages = [
+    { include = "respan_instrumentation_replicate", from = "./src" },
+]
+
+[tool.poetry.dependencies]
+python = ">=3.11,<3.14"
+respan-tracing = "^2.17.0"
+respan-sdk = ">=2.6.1"
+replicate = ">=1.0.0"
+opentelemetry-semantic-conventions-ai = ">=0.4.1"
+
+[tool.poetry.plugins."respan.instrumentations"]
+replicate = "respan_instrumentation_replicate:ReplicateInstrumentor"
+
+[tool.pytest.ini_options]
+markers = [
+    "integration: live network tests (requires API keys)",
+]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

respan_instrumentation_replicate-0.1.0/src/respan_instrumentation_replicate/__init__.py

@@ -0,0 +1,5 @@
+"""Respan instrumentation for Replicate."""
+
+from respan_instrumentation_replicate._instrumentation import ReplicateInstrumentor
+
+__all__ = ["ReplicateInstrumentor"]

respan_instrumentation_replicate-0.1.0/src/respan_instrumentation_replicate/_constants.py

@@ -0,0 +1,47 @@
+"""Replicate instrumentation constants."""
+
+REPLICATE_INSTRUMENTATION_NAME = "replicate"
+REPLICATE_SYSTEM_NAME = "replicate"
+
+REPLICATE_RUN_SPAN_NAME = "replicate.run"
+REPLICATE_STREAM_SPAN_NAME = "replicate.stream"
+REPLICATE_PREDICTION_CREATE_SPAN_NAME = "replicate.predictions.create"
+REPLICATE_PREDICTION_WAIT_SPAN_NAME = "replicate.prediction.wait"
+
+ASYNC_PREFIX = "async_"
+
+DEPLOYMENT_KEY = "deployment"
+ERROR_KEY = "error"
+ID_KEY = "id"
+INPUT_KEY = "input"
+LOGS_KEY = "logs"
+METRICS_KEY = "metrics"
+MODEL_KEY = "model"
+OUTPUT_KEY = "output"
+PREDICTION_KEY = "prediction"
+PROMPT_KEY = "prompt"
+REF_KEY = "ref"
+RESPAN_PARAMS_KEY = "respan_params"
+RESPAN_PARAMS_MODEL_KEY = "model"
+STATUS_KEY = "status"
+STREAM_KEY = "stream"
+VERSION_KEY = "version"
+PREDICTION_RESPAN_MODEL_ATTR = "_respan_reported_model"
+
+ASSISTANT_ROLE = "assistant"
+USER_ROLE = "user"
+
+MAX_STREAM_CHUNKS = 200
+MAX_TEXT_LENGTH = 16_000
+
+OFF_CONTRACT_ALIASES = {
+    "completion_tokens",
+    "has_tool_calls",
+    "model",
+    "parallel_tool_calls",
+    "prompt_tokens",
+    "span_tools",
+    "tool_calls",
+    "tools",
+    "total_request_tokens",
+}