rudderstack-ai 0.1.0__py3-none-any.whl

rudderstack/ai/__init__.py

@@ -0,0 +1,58 @@
+"""RudderStack AI events: product analytics for AI features, through your RudderStack client.
+
+Every answer your app gives produces one ``ai_response_received`` with totals, however many
+model calls, agents or tool steps it took, joined to the prompt and to later feedback by
+``interaction_id``. Events follow the AI events spec v0.2.
+
+Core API (this module): :func:`configure`, :func:`register_price`, :func:`ai_context`,
+:func:`ai_interaction`, :func:`start_interaction`, :func:`track_action`.
+
+Integrations (import explicitly; each needs its extra installed):
+
+* ``rudderstack.ai.openai``: :func:`~rudderstack.ai.openai.wrap_openai`
+* ``rudderstack.ai.anthropic``: :func:`~rudderstack.ai.anthropic.wrap_anthropic`
+* ``rudderstack.ai.langchain``: :class:`~rudderstack.ai.langchain.RudderStackCallbackHandler`
+* ``rudderstack.ai.claude_agent_sdk``: :func:`~rudderstack.ai.claude_agent_sdk.tracked_query`
+"""
+
+from ._core import (
+    SPEC_VERSION,
+    ActionType,
+    AIContext,
+    AnalyticsClient,
+    Interaction,
+    ResponseStatus,
+    Trigger,
+    ai_context,
+    ai_interaction,
+    configure,
+    current_context,
+    current_interaction,
+    start_interaction,
+    track_action,
+)
+from ._pricing import Price, clear_prices, register_price
+from ._proxy import unwrap
+from ._version import __version__
+
+__all__ = [
+    "SPEC_VERSION",
+    "AIContext",
+    "ActionType",
+    "AnalyticsClient",
+    "Interaction",
+    "Price",
+    "ResponseStatus",
+    "Trigger",
+    "__version__",
+    "ai_context",
+    "ai_interaction",
+    "clear_prices",
+    "configure",
+    "current_context",
+    "current_interaction",
+    "register_price",
+    "start_interaction",
+    "track_action",
+    "unwrap",
+]