Providers

pip install tracia

OPENAI_API_KEY=sk-your-openai-key

result = client.run_local(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Hello!"},
    ],
    temperature=0.7,
    max_output_tokens=500,
)

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

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

ANTHROPIC_API_KEY=sk-ant-your-anthropic-key

result = client.run_local(
    model="claude-sonnet-4-20250514",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Explain quantum computing."},
    ],
    temperature=0.7,
    max_output_tokens=1000,
)

stream = client.run_local(
    model="claude-sonnet-4-20250514",
    messages=[{"role": "user", "content": "Write a story."}],
    stream=True,
)

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

GOOGLE_API_KEY=your-google-api-key

result = client.run_local(
    model="gemini-2.0-flash",
    messages=[
        {"role": "user", "content": "What are the benefits of Python?"},
    ],
    temperature=0.5,
    max_output_tokens=800,
)

stream = client.run_local(
    model="gemini-2.0-flash",
    messages=[{"role": "user", "content": "Explain machine learning."}],
    stream=True,
)

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

# Fine-tuned OpenAI model
result = client.run_local(
    model="ft:gpt-4o:my-org:custom-model:abc123",
    provider="openai",
    messages=[{"role": "user", "content": "Hello!"}],
)

import os

result = client.run_local(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}],
    provider_api_key=os.environ["OPENAI_API_KEY_SECONDARY"],
)

from tracia import Tracia, TraciaError, TraciaErrorCode

try:
    result = client.run_local(
        model="gpt-4o",
        messages=[{"role": "user", "content": "Hello!"}],
    )
except TraciaError as error:
    if error.code == TraciaErrorCode.MISSING_PROVIDER_API_KEY:
        print("Set OPENAI_API_KEY environment variable")
    elif error.code == TraciaErrorCode.PROVIDER_ERROR:
        print(f"Provider API error: {error.message}")
    elif error.code == TraciaErrorCode.UNSUPPORTED_MODEL:
        print("Specify provider explicitly for custom models")