from shim import ShimClient

with ShimClient(api_key="sk_live_...") as shim:
    result = shim.repair(raw_output='{"name": "John", "age": 30')

    if result.success:
        print(result.repaired)  # {'name': 'John', 'age': 30}

from shim import ShimClient

shim = ShimClient(
    api_key="sk_live_...",   # Required
    base_url="https://api.shim.so"  # Optional, default
)

# Always close when done, or use as context manager
shim.close()

from shim import AsyncShimClient

shim = AsyncShimClient(
    api_key="sk_live_...",
    base_url="https://api.shim.so"
)

await shim.close()

from shim import ShimClient

with ShimClient(api_key="sk_live_...") as shim:
    result = shim.repair(raw_output='{"name": "John", "age": "30"')

    print(result.success)              # True
    print(result.repaired)             # {'name': 'John', 'age': '30'}
    print(result.metadata.confidence)  # 'high'

from shim import AsyncShimClient

async with AsyncShimClient(api_key="sk_live_...") as shim:
    result = await shim.repair(raw_output='{"name": "John", "age": "30"')
    print(result.repaired)

from shim import ShimClient

schema = {
    "type": "object",
    "properties": {
        "name": {"type": "string"},
        "age":  {"type": "number"}
    },
    "required": ["name", "age"]
}

with ShimClient(api_key="sk_live_...") as shim:
    result = shim.repair(
        raw_output='{"name": "John", "age": "30"',
        schema=schema,
        mode="strict"
    )

    if result.success:
        print(result.repaired)             # {'name': 'John', 'age': 30}
        print(result.metadata.confidence)  # 'medium' — schema repair present

session = shim.stream.start(
    schema={"type": "object", "properties": {"name": {"type": "string"}}},
    mode="strict"
)

print(session.session_id)  # 'sess_abc123'
print(session.expires_at)  # unix timestamp, 60s from now

state = shim.stream.push(session.session_id, '{"name": "Jo')

print(state.structurally_complete)  # False
print(state.buffered)               # '{"name": "Jo'
print(state.safe_to_emit)           # False

result = shim.stream.finalize(session.session_id)
print(result.repaired)  # {'name': 'John'}

from shim import ShimClient

with ShimClient(api_key="sk_live_...") as shim:
    session = shim.stream.start()

    # Push chunks as they arrive from LLM
    shim.stream.push(session.session_id, '{"name": "Jo')
    shim.stream.push(session.session_id, 'hn", "age": 30')

    # Finalize when stream ends
    result = shim.stream.finalize(session.session_id)
    print(result.repaired)  # {'name': 'John', 'age': 30}

from shim import AsyncShimClient

async with AsyncShimClient(api_key="sk_live_...") as shim:
    session = await shim.stream.start()

    await shim.stream.push(session.session_id, '{"name": "Jo')
    await shim.stream.push(session.session_id, 'hn", "age": 30')

    result = await shim.stream.finalize(session.session_id)
    print(result.repaired)  # {'name': 'John', 'age': 30}

from shim import ShimClient, RepairResponse, StreamingState

with ShimClient(api_key="sk_live_...") as shim:
    result: RepairResponse = shim.repair(raw_output="{}")

    # Full autocomplete on all fields
    print(result.metadata.confidence)       # 'high' | 'medium' | 'low' | 'n/a'
    print(result.metadata.syntax_repairs)   # list[RepairDetail]
    print(result.metadata.warnings)         # list[Warning]
    print(result.metadata.errors)           # list[RepairError]

import httpx
from shim import ShimClient

with ShimClient(api_key="sk_live_...") as shim:
    try:
        result = shim.repair(raw_output="not json at all")

        if not result.success:
            for error in result.metadata.errors:
                print(f"{error.code}: {error.message}")
                print(f"  recoverable: {error.recoverable}")
    except httpx.HTTPStatusError as e:
        print(f"Transport error: {e.response.status_code}")