Error Handling

from tracia import Tracia, TraciaError, TraciaErrorCode

client = Tracia(api_key="tr_your_api_key")

try:
    result = client.prompts.run("welcome-email", {"name": "Alice"})
    print(result.text)
except TraciaError as error:
    print(f"Error [{error.code}]: {error.message}")

from tracia import Tracia, TraciaError, TraciaErrorCode

client = Tracia(api_key="tr_your_api_key")

try:
    result = client.prompts.run("welcome-email", {"name": "Alice"})
    print(result.text)
except TraciaError as error:
    if error.code == TraciaErrorCode.UNAUTHORIZED:
        print("Invalid API key")
    elif error.code == TraciaErrorCode.NOT_FOUND:
        print("Prompt does not exist")
    elif error.code == TraciaErrorCode.CONFLICT:
        print("Resource already exists")
    elif error.code == TraciaErrorCode.MISSING_VARIABLES:
        print("Missing required template variables")
    elif error.code == TraciaErrorCode.PROVIDER_ERROR:
        print(f"LLM provider error: {error.message}")
    elif error.code == TraciaErrorCode.NETWORK_ERROR:
        print(f"Network error: {error.message}")
    elif error.code == TraciaErrorCode.TIMEOUT:
        print("Request timed out")
    elif error.code == TraciaErrorCode.MISSING_PROVIDER_API_KEY:
        print("Provider API key not set")
    elif error.code == TraciaErrorCode.UNSUPPORTED_MODEL:
        print("Model not recognized")
    else:
        print(f"Unknown error: {error.message}")

# Prompt expects {{name}} and {{product}}
result = client.prompts.run("welcome-email", {
    "name": "Alice"
    # Missing "product" variable
})
# Raises TraciaError with code MISSING_VARIABLES

result = client.prompts.run("welcome-email", {"name": "Alice"})
# Raises TraciaError with code PROVIDER_ERROR
# Message: "No OpenAI API key configured. Add one in Settings > Providers."

prompt = client.prompts.get("nonexistent-prompt")
# Raises TraciaError with code NOT_FOUND

result = client.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}]
)
# Raises TraciaError with code MISSING_PROVIDER_API_KEY
# Message: "Missing API key for openai. Set the OPENAI_API_KEY environment variable
#           or provide provider_api_key in options."

# Option 1: Environment variable
# OPENAI_API_KEY=sk-your-key

# Option 2: Explicit key
result = client.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
    provider_api_key="sk-your-key",
)

result = client.run_local(
    model="my-fine-tuned-model",
    messages=[{"role": "user", "content": "Hello!"}]
)
# Raises TraciaError with code UNSUPPORTED_MODEL

result = client.run_local(
    model="my-fine-tuned-model",
    provider="openai",
    messages=[{"role": "user", "content": "Hello!"}]
)

import time
from tracia import Tracia, TraciaError, TraciaErrorCode

client = Tracia(api_key="tr_your_api_key")

NON_RETRYABLE = {
    TraciaErrorCode.UNAUTHORIZED,
    TraciaErrorCode.NOT_FOUND,
    TraciaErrorCode.MISSING_VARIABLES,
    TraciaErrorCode.INVALID_REQUEST,
}


def run_with_retry(slug: str, variables: dict[str, str], max_retries: int = 3):
    for attempt in range(1, max_retries + 1):
        try:
            return client.prompts.run(slug, variables)
        except TraciaError as error:
            if error.code in NON_RETRYABLE:
                raise
            if attempt < max_retries:
                delay = 2**attempt
                time.sleep(delay)
                continue
            raise