Error Handling

from subconscious import Subconscious
from subconscious.errors import (
    SubconsciousError,    # Base class
    AuthenticationError,  # 401
    ValidationError,      # 400
    NotFoundError,        # 404
)
import os

client = Subconscious(api_key=os.environ.get("SUBCONSCIOUS_API_KEY"))

try:
    run = client.run(
        engine="tim-gpt",
        input={"instructions": "Hello"},
        options={"await_completion": True},
    )
except AuthenticationError:
    print("Invalid API key")
except ValidationError as e:
    print(f"Bad request: {e.message}")
except SubconsciousError as e:
    print(f"Error {e.status}: {e.code} - {e.message}")

import time
import os
from subconscious import Subconscious
from subconscious.errors import SubconsciousError

client = Subconscious(api_key=os.environ.get("SUBCONSCIOUS_API_KEY"))

def run_with_retry(instructions, max_retries=3):
    for attempt in range(max_retries):
        try:
            return client.run(
                engine="tim-gpt",
                input={"instructions": instructions},
                options={"await_completion": True},
            )
        except SubconsciousError as e:
            if e.status < 500 or attempt == max_retries - 1:
                raise
            time.sleep(2 ** attempt)

run = run_with_retry("Summarize the news")

run = client.run(
    engine="tim-gpt",
    input={"instructions": "Your task"},
    options={"await_completion": True},
)

match run.status:
    case "succeeded":
        print(run.result.answer)
    case "failed":
        print(f"Failed: {run.error.message}")
    case "timed_out":
        print("Timed out")
    case "canceled":
        print("Canceled")