raindrop-ai 0.0.34__py3-none-any.whl → 0.0.36__py3-none-any.whl

raindrop/analytics.py

@@ -66,6 +66,7 @@ __all__ = [
     "start_span",
     "ManualSpan",
     "set_span_properties",
+    "set_llm_span_io",
     "flush",
     "shutdown",
 ]
@@ -347,6 +348,59 @@ def _should_send_prompts():
     ).lower() == "true" or context_api.get_value("override_enable_content_tracing")
 
 
+def set_llm_span_io(
+    input: Any = None,
+    output: Any = None,
+) -> None:
+    """
+    Set LLM input/output content on the current span.
+
+    Use this to add prompt/completion content to auto-instrumented spans
+    that don't capture content automatically (e.g., Bedrock with aioboto3).
+
+    Args:
+        input: The input/prompt content (messages, text, etc.)
+        output: The output/completion content (response text, message, etc.)
+
+    Example:
+        response = await bedrock_client.converse(modelId=model, messages=messages)
+        raindrop.set_llm_span_io(
+            input=messages,
+            output=response["output"]["message"]["content"]
+        )
+    """
+    if not _should_send_prompts():
+        return
+
+    span = get_current_span()
+    if not span or not span.is_recording():
+        logger.debug("[raindrop] set_llm_span_io called but no active span found")
+        return
+
+    try:
+        if input is not None:
+            input_str = (
+                json.dumps(input, cls=JSONEncoder)
+                if not isinstance(input, str)
+                else input
+            )
+            input_str = _truncate_json_if_needed(input_str)
+            span.set_attribute("gen_ai.prompt.0.role", "user")
+            span.set_attribute("gen_ai.prompt.0.content", input_str)
+
+        if output is not None:
+            output_str = (
+                json.dumps(output, cls=JSONEncoder)
+                if not isinstance(output, str)
+                else output
+            )
+            output_str = _truncate_json_if_needed(output_str)
+            span.set_attribute("gen_ai.completion.0.role", "assistant")
+            span.set_attribute("gen_ai.completion.0.content", output_str)
+    except Exception as e:
+        logger.debug(f"[raindrop] Failed to record LLM content: {e}")
+
+
 # Signal types - This is now defined in models.py
 # SignalType = Literal["default", "feedback", "edit"]
 

raindrop/version.py

@@ -1 +1 @@
-VERSION = "0.0.34"
+VERSION = "0.0.36"

{raindrop_ai-0.0.34.dist-info → raindrop_ai-0.0.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: raindrop-ai
-Version: 0.0.34
+Version: 0.0.36
 Summary: Raindrop AI (Python SDK)
 License: MIT
 Author: Raindrop AI
@@ -11,9 +11,11 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: opentelemetry-sdk (>=1.39.0)
 Requires-Dist: pydantic (>=2.09,<3)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: traceloop-sdk (>=0.46.0)
+Requires-Dist: urllib3 (>=2.6.0)
 Description-Content-Type: text/markdown
 
 # Raindrop Python SDK

{raindrop_ai-0.0.34.dist-info → raindrop_ai-0.0.36.dist-info}/RECORD

@@ -1,10 +1,10 @@
 raindrop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-raindrop/analytics.py,sha256=uS4D8_r9VEzIVJgPDBpT7Nkr2i1PS9Yj_QCWgqGEBNM,28507
+raindrop/analytics.py,sha256=dL64bp5Sk1gq5PYcKAdQBNlmrn-5aW-LXTa0C3CApKA,30335
 raindrop/interaction.py,sha256=Kw8HNaDSHVIi1p-2HK5SsYGI4Xg1zQFml4dX2EhIT7w,3123
 raindrop/models.py,sha256=9lOOUQ2FF11RPkntuLZwN3e54pa9HtR8lGvCbzlWOPM,5198
 raindrop/redact.py,sha256=rMNUoI90KxOY3d_zcHAr0TFD2yQ_CDgpDz-1XJLVmHs,7658
-raindrop/version.py,sha256=46z7DjpRJ03wdmbNWoSr-zmeqNLwMTq3Ue5OAhyiKPU,19
+raindrop/version.py,sha256=l8qKbO0eI3Ja_0j_Y0EQzBqlq-Sw-KTFzu8GAbKB2VU,19
 raindrop/well-known-names.json,sha256=9giJF6u6W1R0APW-Pf1dvNUU32OXQEoQ9CBQXSnA3ks,144403
-raindrop_ai-0.0.34.dist-info/METADATA,sha256=24MFAmx6jhHt8W2MybBsEfbWa-wcTrFaU0ATAK6F3Xs,1269
-raindrop_ai-0.0.34.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-raindrop_ai-0.0.34.dist-info/RECORD,,
+raindrop_ai-0.0.36.dist-info/METADATA,sha256=DK-3tpiRg0EeXqXOOzr9TRext7DhnJ3J5kgnvZGQFag,1346
+raindrop_ai-0.0.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+raindrop_ai-0.0.36.dist-info/RECORD,,