File size: 851 Bytes
116524e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""Kayba tracing — instrument your agents and send traces to Kayba.



Usage::



    from kayba_tracing import configure, trace, start_span



    configure(api_key="kb-...")



    @trace

    def my_agent(query: str) -> str:

        with start_span("retrieval") as span:

            span.set_inputs({"query": query})

            results = search(query)

            span.set_outputs(results)

        return synthesize(results)



Install::



    pip install kayba-tracing

"""

from kayba_tracing._wrapper import (
    configure,
    disable,
    enable,
    get_folder,
    get_trace,
    search_traces,
    set_folder,
    start_span,
    trace,
)

__all__ = [
    "configure",
    "disable",
    "enable",
    "get_folder",
    "get_trace",
    "search_traces",
    "set_folder",
    "start_span",
    "trace",
]