raindrop-ai 0.0.35__tar.gz → 0.0.36__tar.gz

{raindrop_ai-0.0.35 → raindrop_ai-0.0.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: raindrop-ai
-Version: 0.0.35
+Version: 0.0.36
 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.35 → raindrop_ai-0.0.36}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "raindrop-ai"
-version = "0.0.35"
+version = "0.0.36"
 description = "Raindrop AI (Python SDK)"
 authors = ["Raindrop AI <sdk@raindrop.ai>"]
 license = "MIT"
@@ -13,6 +13,7 @@ pydantic = ">=2.09,<3"
 requests = "^2.32.3"
 traceloop-sdk = ">=0.46.0"
 opentelemetry-sdk = ">=1.39.0"
+urllib3 = ">=2.6.0"
 
 
 [tool.poetry.group.dev.dependencies]

{raindrop_ai-0.0.35 → raindrop_ai-0.0.36}/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_ai-0.0.36/raindrop/version.py

@@ -0,0 +1 @@
+VERSION = "0.0.36"

raindrop_ai-0.0.35/raindrop/version.py

@@ -1 +0,0 @@
-VERSION = "0.0.35"