simplai-sdk 0.1.0__py3-none-any.whl

simplai_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,42 @@
+billing/__init__.py,sha256=MyTlI603pQk0PZ6lQawQU71Wcz-CZpPW_5X3qef0evQ,80
+billing/api.py,sha256=eC2TEWFsQs_-J-M_1jz_Ks0r8hx1RJIYT5WtC_6vgPg,1534
+billing/client.py,sha256=IUGiu6VDC2c0tjyJ19BLcxTwqoeUidMZBXmTkioQBQY,393
+billing/schema.py,sha256=9oZUQOnAJCD4aLDATpP-YB982RPv70zxv8PeYLjuubU,515
+constants/__init__.py,sha256=ZMjr4ruppIFAZoF8ufmFD3bXKTt3TC8wGGIH3FOhBIY,3266
+core/__init__.py,sha256=tDi8hErxnY2cL_qMU1iGdInXRwbwS7V4iwc5KHWUlho,1125
+core/agents/__init__.py,sha256=slYJzoWIckLXraVmjyT6mnUcJ_CFYNxYfn9xBNr5Kjk,1006
+core/agents/models.py,sha256=oX0dkQXqB49304c4PrQ-oQp2_kw3ZNZW0XaHoKsvChs,2904
+core/agents/execution/__init__.py,sha256=3H5kf-EhHVqkj4mCGcOnuCzO10FlhQ7PSJNnCQxSFxU,1235
+core/agents/execution/api.py,sha256=y0zEVH0yV3FzCU2V3Ra6q62qy_jRV5C1IavPkw1jvKg,8946
+core/agents/execution/client.py,sha256=srLLq8-I6TDX72FORrGxTw1v1sM8zaLqojt9Ni3Nf2U,51572
+core/workflows/WORKFLOW_ARCHITECTURE.md,sha256=yq_K6r3fcAlE0RwHyrtNcQ-PZlGMtQQAZLtnavDzNPw,13987
+core/workflows/__init__.py,sha256=QXBj6GNykRIdzmmeIEmzOvnvdcgbVOxcjgLxBCy5MX8,668
+core/workflows/models.py,sha256=WMkgQHEvEkKTx5rGbgMWa_Ti9FHDWAKDcX_amEuJxK8,1248
+core/workflows/bulk/__init__.py,sha256=H3XFE08_5K8Dl-XizeOAcDg3Fu1EM5OQ_uOzwrCyddg,337
+core/workflows/bulk/api.py,sha256=gWiHMfTagbEF04t_L0_A7PtN2vse4kYa7L0g_OnysME,6146
+core/workflows/bulk/client.py,sha256=e7gY83nO4L1v5rvOj5zWM2Z6c41u96Yeh53_mRgsm10,3034
+core/workflows/bulk/schema.py,sha256=Snj-iCYTJcjdHoAXv4HWwDpza2KRGH2LgecHBVae8mI,2422
+core/workflows/scheduling/__init__.py,sha256=GKt7x5ww1Nf98bZOdZ8V4XRo97CFep5sGqK70bVXkpQ,179
+core/workflows/scheduling/api.py,sha256=M-EEPTdwiLCajW-lwWkdgn3xDHYpEFmPkB5XdQSvzto,5965
+core/workflows/scheduling/client.py,sha256=Blbn1sS7eA97MUD8gvufvQ9uo2y0g8si784Erbterss,4085
+core/workflows/scheduling/schema.py,sha256=7V4-uRxXX84d6eZUvDW3oSPg2er3YFDOlSWJRbK29e0,3355
+core/workflows/tool_execution/__init__.py,sha256=Z59MhfZbHYvBtxGcQupXT5Pt5As6tn1kKwf2d7N5iLw,292
+core/workflows/tool_execution/api.py,sha256=sgl5l87mDmaAZoye-ejR4usPb2_wI7LdM8OZVLqMIdg,5119
+core/workflows/tool_execution/client.py,sha256=lFXXsVgz_VDxWfBQ42ZNo8UcTOV7RumML6szW6s_ewQ,5256
+core/workflows/tool_execution/schema.py,sha256=To8uVGFNULcdupaCZbYnJTu4azAGTv0so75wuVdCKrI,1502
+exceptions/__init__.py,sha256=bUAIwVmyYtHhPvHOpO7LAYRJrzhY9N_KMB3-AfZCUlU,522
+simplai_sdk/__init__.py,sha256=vT8U1gQbTS4xXFyoVU767IMh9-5ZVqPaNjL5gF8YFKU,121
+simplai_sdk/simplai.py,sha256=U-jf1E4ygd6fyKNx6lMqy79_F2Fh15dAzW4E-pFOISY,8116
+simplai_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=byn2pn7QZM4hs8uCv8VSaiZ1XTk5KrpNSWZ7hYc2EpY,1064
+traces/__init__.py,sha256=Wn1H6KU_JXIGlj-O0NBkz9lykh3Axwj9UqJf7Sa5QMk,37
+traces/agents/__init__.py,sha256=wzwdX5Bxkol_ikS6x2M5wyM_zEAzlbM0NP6dJI9LmTU,1087
+traces/agents/api.py,sha256=Afv_blHayN1GwBxaSmcyTgugioyg62E9Fg67xsBNTeA,10730
+traces/agents/client.py,sha256=BxZUOFk7OMGelsK8bRts6my2n4UchCI2bUIEhUXiEXA,24975
+traces/agents/models.py,sha256=HVxHQJtixau-B0x0OwwUexnaJuOsKrAYkXRAtoPg0cs,8629
+traces/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+utils/config.py,sha256=uaEGZ3QfYOhKK34d5RBbuIVP18Lue2ZcLz-Hpuv_p_c,3354
+simplai_sdk-0.1.0.dist-info/METADATA,sha256=5RVEFOafHbdkvTT3zDWoZ-YoFLwt_Syuc-Oxb2jLnS8,23617
+simplai_sdk-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+simplai_sdk-0.1.0.dist-info/top_level.txt,sha256=CznxFBTncbDBZKLnqjDee5bfUXPKrrKXD4qVKq54SeA,59
+simplai_sdk-0.1.0.dist-info/RECORD,,

simplai_sdk-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

simplai_sdk-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 SimplAI
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

simplai_sdk-0.1.0.dist-info/top_level.txt

@@ -0,0 +1,7 @@
+billing
+constants
+core
+exceptions
+simplai_sdk
+traces
+utils

traces/__init__.py

@@ -0,0 +1 @@
+"""Traces module for SimplAI SDK."""

traces/agents/__init__.py

@@ -0,0 +1,55 @@
+"""Simplai Tracing APIs for agents.
+
+This module provides APIs for fetching and analyzing traces from agent conversations.
+"""
+
+from .api import (
+    fetch_traces,
+    get_trace_details,
+    fetch_trace_nodes,
+    get_aggregate_output,
+    get_trace_tree,
+    get_trace_node_details,
+    download_traces,
+    get_agent_metrics,
+    get_tool_metrics,
+)
+from .client import TraceClient
+from .models import (
+    EvalMetrics,
+    FetchTraceFilters,
+    MetricsRequestDto,
+    PageableResponse,
+    RagTrace,
+    TraceData,
+    TraceError,
+    TraceFilters,
+    TraceNode,
+    ChildInfo,
+)
+
+__all__ = [
+    # API functions
+    "fetch_traces",
+    "get_trace_details",
+    "fetch_trace_nodes",
+    "get_aggregate_output",
+    "get_trace_tree",
+    "get_trace_node_details",
+    "download_traces",
+    "get_agent_metrics",
+    "get_tool_metrics",
+    # Client
+    "TraceClient",
+    # Models
+    "EvalMetrics",
+    "FetchTraceFilters",
+    "MetricsRequestDto",
+    "PageableResponse",
+    "RagTrace",
+    "TraceData",
+    "TraceError",
+    "TraceFilters",
+    "TraceNode",
+    "ChildInfo",
+]

traces/agents/api.py

@@ -0,0 +1,350 @@
+from __future__ import annotations
+
+from datetime import datetime
+from typing import Any, Dict, List, Optional, Union
+
+from utils.config import get_api_key, get_base_url
+
+from .client import TraceClient
+from constants import DEFAULT_BASE_URL
+from .models import (
+    FetchTraceFilters,
+    MetricsRequestDto,
+    PageableResponse,
+    RagTrace,
+    TraceError,
+    TraceFilters,
+    TraceNode,
+)
+
+
+def fetch_traces(
+    *,
+    api_key: Optional[str] = None,
+    filters: Optional[FetchTraceFilters] = None,
+    page: int = 0,
+    size: int = 20,
+    sort: str = "id",
+    direction: str = "DESC",
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> PageableResponse:
+    """Fetch paginated request-level traces (RAG traces).
+
+    Args:
+        api_key: PIM-SID API key.
+        filters: Optional filters for querying traces.
+        page: Page number (0-indexed).
+        size: Page size.
+        sort: Sort field.
+        direction: Sort direction (ASC or DESC).
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        PageableResponse containing paginated RagTrace objects.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.fetch_traces(filters=filters, page=page, size=size, sort=sort, direction=direction)
+
+
+def get_trace_details(
+    *,
+    api_key: Optional[str] = None,
+    trace_id: Optional[str] = None,
+    request_id: Optional[str] = None,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> RagTrace:
+    """Get details of a specific request-level trace.
+
+    Args:
+        api_key: PIM-SID API key.
+        trace_id: Trace ID (optional, but either trace_id or request_id required).
+        request_id: Request ID (optional, but either trace_id or request_id required).
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        RagTrace object with full details.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.get_trace_details(trace_id=trace_id, request_id=request_id)
+
+
+def fetch_trace_nodes(
+    *,
+    api_key: Optional[str] = None,
+    filters: Optional[TraceFilters] = None,
+    page: int = 0,
+    size: int = 20,
+    sort: str = "id",
+    direction: str = "DESC",
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> PageableResponse:
+    """Fetch paginated tree-based trace nodes (root traces).
+
+    Args:
+        api_key: PIM-SID API key.
+        filters: Optional filters for querying traces.
+        page: Page number (0-indexed).
+        size: Page size.
+        sort: Sort field.
+        direction: Sort direction (ASC or DESC).
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        PageableResponse containing paginated TraceNode objects.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.fetch_trace_nodes(filters=filters, page=page, size=size, sort=sort, direction=direction)
+
+
+def get_aggregate_output(
+    *,
+    api_key: Optional[str] = None,
+    filters: Optional[TraceFilters] = None,
+    page: int = 0,
+    size: int = 20,
+    sort: str = "id",
+    direction: str = "DESC",
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> Dict[str, Any]:
+    """Get aggregate output traces (summary view).
+
+    Args:
+        api_key: PIM-SID API key.
+        filters: Optional filters for querying traces.
+        page: Page number (0-indexed).
+        size: Page size.
+        sort: Sort field.
+        direction: Sort direction (ASC or DESC).
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        Dictionary containing aggregate trace data.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.get_aggregate_output(filters=filters, page=page, size=size, sort=sort, direction=direction)
+
+
+def get_trace_tree(
+    *,
+    api_key: Optional[str] = None,
+    tree_id: str,
+    node_id: str,
+    max_depth: Optional[int] = None,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+    seller_id: Optional[str] = None,
+    client_id: Optional[str] = None,
+    seller_profile_id: Optional[str] = None,
+    raw_response: bool = False,
+) -> Union[TraceNode, Dict[str, Any]]:
+    """Get a trace sub-tree starting from a specific node.
+
+    Args:
+        api_key: PIM-SID API key.
+        tree_id: Tree ID of the trace.
+        node_id: Node ID to start the tree from.
+        max_depth: Maximum depth to traverse (optional).
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+        seller_id: Optional seller ID.
+        client_id: Optional client ID.
+        seller_profile_id: Optional seller profile ID.
+        raw_response: If True, return raw JSON response instead of parsed TraceNode.
+
+    Returns:
+        TraceNode representing the root of the sub-tree, or raw JSON dict if raw_response=True.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+        seller_id=seller_id,
+        client_id=client_id,
+        seller_profile_id=seller_profile_id,
+    )
+    return client.get_trace_tree(tree_id=tree_id, node_id=node_id, max_depth=max_depth, raw_response=raw_response)
+
+
+def get_trace_node_details(
+    *,
+    api_key: Optional[str] = None,
+    tree_id: str,
+    node_id: str,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> TraceNode:
+    """Get details of a specific trace node.
+
+    Args:
+        api_key: PIM-SID API key.
+        tree_id: Tree ID of the trace.
+        node_id: Node ID to get details for.
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        TraceNode object with full details.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.get_trace_node_details(tree_id=tree_id, node_id=node_id)
+
+
+def download_traces(
+    *,
+    api_key: Optional[str] = None,
+    filters: Optional[TraceFilters] = None,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> bytes:
+    """Download traces as CSV.
+
+    Args:
+        api_key: PIM-SID API key.
+        filters: Optional filters for querying traces.
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        Bytes containing CSV file content.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.download_traces(filters=filters)
+
+
+def get_agent_metrics(
+    *,
+    api_key: Optional[str] = None,
+    metrics_request: Optional[MetricsRequestDto] = None,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> Dict[str, Any]:
+    """Get agent-level metrics from traces.
+
+    Args:
+        api_key: PIM-SID API key.
+        metrics_request: Optional metrics request with filters.
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        Dictionary containing aggregated agent metrics.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.get_agent_metrics(metrics_request=metrics_request)
+
+
+def get_tool_metrics(
+    *,
+    api_key: Optional[str] = None,
+    metrics_request: Optional[MetricsRequestDto] = None,
+    base_url: Optional[str] = None,
+    user_id: Optional[str] = None,
+    tenant_id: str = "1",
+    project_id: Optional[int] = None,
+) -> Dict[str, Any]:
+    """Get tool-level metrics from traces.
+
+    Args:
+        api_key: PIM-SID API key.
+        metrics_request: Optional metrics request with filters.
+        base_url: Override the default base URL if needed.
+        user_id: Optional user ID for trace operations.
+        tenant_id: Tenant ID (defaults to "1").
+        project_id: Optional project ID.
+
+    Returns:
+        Dictionary containing aggregated tool metrics.
+    """
+    client = TraceClient(
+        api_key=get_api_key(api_key),
+        base_url=get_base_url(base_url),
+        user_id=user_id,
+        tenant_id=tenant_id,
+        project_id=project_id,
+    )
+    return client.get_tool_metrics(metrics_request=metrics_request)
+