Sessions

from tracia import Tracia

client = Tracia(api_key="tr_your_api_key")

# Start a new session
session = client.create_session()

# Or continue an existing trace (e.g., across HTTP requests)
session = client.create_session(
    trace_id="tr_existing_trace_id",
    parent_span_id="sp_last_span_id",  # optional: chain from a specific span
)

session = client.create_session()

# First call - creates the session (trace)
result1 = session.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "What is the weather?"}],
)
# result1.span_id = "sp_abc123"
# session.trace_id = "tr_xyz789" (session ID)

# Second call - automatically linked to first
result2 = session.run_local(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "What is the weather?"},
        result1.message,
        {"role": "user", "content": "What about tomorrow?"},
    ],
)
# Automatically sends: trace_id="tr_xyz789", parent_span_id="sp_abc123"

# Third call - chain continues
result3 = session.run_local(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "What is the weather?"},
        result1.message,
        {"role": "user", "content": "What about tomorrow?"},
        result2.message,
        {"role": "user", "content": "Thanks!"},
    ],
)
# Automatically sends: trace_id="tr_xyz789", parent_span_id=result2.span_id

# Sync
result = session.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
)

# Async
result = await session.arun_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
)

trace_id = session.trace_id
# Returns: "tr_abc123" or None if no calls made yet

last_id = session.last_span_id
# Returns: "sp_xyz789" or None if no calls made yet

session.reset()
# Next call will start a new trace session

# Django view - continue a trace from a previous request
def chat(request):
    data = json.loads(request.body)
    message = data["message"]
    trace_id = data.get("trace_id")
    last_span_id = data.get("last_span_id")

    session = client.create_session(
        trace_id=trace_id,              # Continue the same trace
        parent_span_id=last_span_id,    # Chain from the last span
    )

    result = session.run_local(
        model="gpt-4o",
        messages=[{"role": "user", "content": message}],
    )

    return JsonResponse({
        "text": result.text,
        "trace_id": session.trace_id,
        "last_span_id": result.span_id,
    })

# First call
result1 = client.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
)

# Second call - manually link
result2 = client.run_local(
    model="gpt-4o",
    messages=[...],
    trace_id=result1.span_id,       # Use first span as session ID
    parent_span_id=result1.span_id,  # Parent is the first span
)

# Third call - continue chain
result3 = client.run_local(
    model="gpt-4o",
    messages=[...],
    trace_id=result1.span_id,       # Same session
    parent_span_id=result2.span_id,  # Parent is the second span
)

session = client.create_session()

stream = session.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Write a poem."}],
    stream=True,
)

for chunk in stream:
    print(chunk, end="")

result = stream.result.result()  # Future[StreamResult] → StreamResult
# session.trace_id is now set