raindrop-ai 0.0.35__py3-none-any.whl → 0.0.37__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/interaction.py

@@ -1,4 +1,6 @@
 from __future__ import annotations
+import json
+import time
 from typing import (
     Any,
     Dict,
@@ -9,6 +11,7 @@ from typing import (
     Union,
     Iterator,
 )
+from datetime import datetime, timezone
 from uuid import uuid4
 from dataclasses import dataclass
 
@@ -107,6 +110,146 @@ class Interaction:
             convo_id=self._convo_id,
         )
 
+    def track_tool(
+        self,
+        *,
+        name: str,
+        input: Any | None = None,
+        output: Any | None = None,
+        duration_ms: float | int | None = None,
+        start_time: datetime | int | float | None = None,
+        error: BaseException | str | None = None,
+        properties: Dict[str, Any] | None = None,
+        version: int | None = None,
+    ) -> None:
+        """
+        Retroactively log a tool span tied to this interaction.
+        """
+        if not _core._tracing_enabled or not _core.TracerWrapper.verify_initialized():
+            return
+
+        # Duration normalization
+        dur_ms = float(duration_ms) if duration_ms is not None else 0.0
+        if dur_ms < 0:
+            dur_ms = 0.0
+        duration_ns = int(round(dur_ms * 1_000_000))
+
+        # start_time normalization (epoch nanoseconds)
+        start_ns: int | None = None
+        if isinstance(start_time, datetime):
+            dt = start_time
+            if dt.tzinfo is None:
+                dt = dt.replace(tzinfo=timezone.utc)
+            else:
+                dt = dt.astimezone(timezone.utc)
+
+            epoch = datetime(1970, 1, 1, tzinfo=timezone.utc)
+            delta = dt - epoch
+            total_us = (
+                delta.days * 86400 + delta.seconds
+            ) * 1_000_000 + delta.microseconds
+            start_ns = total_us * 1_000
+        elif isinstance(start_time, (int, float)):
+            v = float(start_time)
+            # Heuristic: values smaller than ~1973 in ms are likely epoch-seconds
+            start_ns = (
+                int(round(v * 1_000_000_000))
+                if abs(v) < 1e11
+                else int(round(v * 1_000_000))
+            )
+
+        if start_ns is None:
+            start_ns = time.time_ns() - duration_ns
+
+        end_ns = start_ns + duration_ns
+
+        tlp_kind = _core.TraceloopSpanKindValues.TOOL
+        span_name = f"{name}.{tlp_kind.value}"
+
+        tracer = _core.trace.get_tracer("traceloop.tracer")
+        span = tracer.start_span(span_name, start_time=start_ns)
+
+        try:
+            span.set_attribute(_core.SpanAttributes.TRACELOOP_SPAN_KIND, tlp_kind.value)
+            span.set_attribute(_core.SpanAttributes.TRACELOOP_ENTITY_NAME, name)
+            if version is not None:
+                span.set_attribute(
+                    _core.SpanAttributes.TRACELOOP_ENTITY_VERSION, version
+                )
+
+            association_props = {
+                "event_id": self._event_id,
+                "user_id": self._user_id,
+                "event": self._event,
+                "convo_id": self._convo_id,
+            }
+            for key, value in association_props.items():
+                if value is not None:
+                    span.set_attribute(f"traceloop.association.properties.{key}", value)
+
+            if properties:
+                for key, value in properties.items():
+                    if key in association_props:
+                        continue
+                    if value is None:
+                        continue
+                    if isinstance(value, (str, bool, int, float)):
+                        attr_val: Any = value
+                    else:
+                        try:
+                            attr_val = json.dumps(value, cls=_core.JSONEncoder)
+                        except Exception:
+                            attr_val = str(value)
+                    span.set_attribute(
+                        f"traceloop.association.properties.{key}", attr_val
+                    )
+
+            if duration_ms is not None:
+                span.set_attribute("traceloop.entity.duration_ms", dur_ms)
+
+            if _core._should_send_prompts():
+                if input is not None:
+                    try:
+                        json_input = json.dumps(
+                            {"args": [input]}, cls=_core.JSONEncoder
+                        )
+                        span.set_attribute(
+                            _core.SpanAttributes.TRACELOOP_ENTITY_INPUT,
+                            _core._truncate_json_if_needed(json_input),
+                        )
+                    except Exception as e:
+                        _core.logger.debug(
+                            f"[raindrop] Could not serialize input for span: {e}"
+                        )
+
+                if output is not None:
+                    try:
+                        json_output = json.dumps(output, cls=_core.JSONEncoder)
+                        span.set_attribute(
+                            _core.SpanAttributes.TRACELOOP_ENTITY_OUTPUT,
+                            _core._truncate_json_if_needed(json_output),
+                        )
+                    except Exception as e:
+                        _core.logger.debug(
+                            f"[raindrop] Could not serialize output for span: {e}"
+                        )
+
+            if error is not None:
+                exc = (
+                    error if isinstance(error, BaseException) else Exception(str(error))
+                )
+                span.set_status(_core.Status(_core.StatusCode.ERROR, str(exc)))
+                span.record_exception(exc)
+            else:
+                span.set_status(_core.Status(_core.StatusCode.OK))
+        finally:
+            span.end(end_time=end_ns)
+
+        if _core.debug_logs:
+            _core.logger.debug(
+                f'[raindrop] track_tool: logged tool span "{name}" (duration_ms={duration_ms})'
+            )
+
     # convenience
     @property
     def id(self) -> str:

raindrop/version.py

@@ -1 +1 @@
-VERSION = "0.0.35"
+VERSION = "0.0.37"

{raindrop_ai-0.0.35.dist-info → raindrop_ai-0.0.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: raindrop-ai
-Version: 0.0.35
+Version: 0.0.37
 Summary: Raindrop AI (Python SDK)
 License: MIT
 Author: Raindrop AI
@@ -15,6 +15,7 @@ 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.37.dist-info/RECORD

@@ -0,0 +1,10 @@
+raindrop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+raindrop/analytics.py,sha256=dL64bp5Sk1gq5PYcKAdQBNlmrn-5aW-LXTa0C3CApKA,30335
+raindrop/interaction.py,sha256=qMTGN-nMDeYmujLhBqIUPq4pKDQx6HX0fqVwB2aq0eA,8639
+raindrop/models.py,sha256=9lOOUQ2FF11RPkntuLZwN3e54pa9HtR8lGvCbzlWOPM,5198
+raindrop/redact.py,sha256=rMNUoI90KxOY3d_zcHAr0TFD2yQ_CDgpDz-1XJLVmHs,7658
+raindrop/version.py,sha256=bPu3tNWPJ9tg3HaQ8cbDkapv1vcKPLFyWLaaRG7eW2Q,19
+raindrop/well-known-names.json,sha256=9giJF6u6W1R0APW-Pf1dvNUU32OXQEoQ9CBQXSnA3ks,144403
+raindrop_ai-0.0.37.dist-info/METADATA,sha256=BO2HCUOURX8Rlhyu4L_XCon7XwS9ln1EP3qEfk6topY,1346
+raindrop_ai-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+raindrop_ai-0.0.37.dist-info/RECORD,,

raindrop_ai-0.0.35.dist-info/RECORD

@@ -1,10 +0,0 @@
-raindrop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-raindrop/analytics.py,sha256=uS4D8_r9VEzIVJgPDBpT7Nkr2i1PS9Yj_QCWgqGEBNM,28507
-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=XbHQAtEbxGWuqWT-2KR0HWozWyaEyD9WkdIKQJfT_OA,19
-raindrop/well-known-names.json,sha256=9giJF6u6W1R0APW-Pf1dvNUU32OXQEoQ9CBQXSnA3ks,144403
-raindrop_ai-0.0.35.dist-info/METADATA,sha256=0mJdipZfyB_rnDfBZjsnINUjiKqiJBSyUEMokqy6Bpg,1313
-raindrop_ai-0.0.35.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-raindrop_ai-0.0.35.dist-info/RECORD,,