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

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.36"
+VERSION = "0.0.37"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: raindrop-ai
-Version: 0.0.36
+Version: 0.0.37
 Summary: Raindrop AI (Python SDK)
 License: MIT
 Author: Raindrop AI

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

@@ -1,10 +1,10 @@
 raindrop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 raindrop/analytics.py,sha256=dL64bp5Sk1gq5PYcKAdQBNlmrn-5aW-LXTa0C3CApKA,30335
-raindrop/interaction.py,sha256=Kw8HNaDSHVIi1p-2HK5SsYGI4Xg1zQFml4dX2EhIT7w,3123
+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=l8qKbO0eI3Ja_0j_Y0EQzBqlq-Sw-KTFzu8GAbKB2VU,19
+raindrop/version.py,sha256=bPu3tNWPJ9tg3HaQ8cbDkapv1vcKPLFyWLaaRG7eW2Q,19
 raindrop/well-known-names.json,sha256=9giJF6u6W1R0APW-Pf1dvNUU32OXQEoQ9CBQXSnA3ks,144403
-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,,
+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,,