Exceptions

package com.example.tnsai.docs;

import com.tnsai.agents.Agent;
import com.tnsai.exceptions.ActionExecutionException;
import com.tnsai.exceptions.LLMException;
import com.tnsai.exceptions.RateLimitException;
import com.tnsai.exceptions.TnsAIException;

import java.util.Map;

public final class ExceptionsExample {
    private ExceptionsExample() {}

    // An action call: choose a recovery from the category.
    static Object forecast(Agent agent) {
        try {
            return agent.executeAction("forecast", Map.of("city", "Paris"));
        } catch (ActionExecutionException e) {
            System.err.println(e.getDetailedMessage());
            return switch (e.getCategory()) {
                case NETWORK, SERVER_ERROR -> "retry in " + e.getSuggestedRetryDelayMs() + " ms";
                case PARAMETER, VALIDATION -> "fix the arguments for " + e.getActionName();
                default -> throw e;
            };
        }
    }

    // A chat turn: catch the most specific type first.
    static String ask(Agent agent, String question) {
        try {
            return agent.chat(question);
        } catch (RateLimitException e) {
            return "rate limited, retry in " + e.getSuggestedRetryDelayMs() + " ms";
        } catch (LLMException e) {
            if (e.getErrorType() == LLMException.ErrorType.CONTEXT_TOO_LONG) {
                return "shorten the question";
            }
            throw e;
        } catch (TnsAIException e) {
            System.err.println(e.getErrorCode() + " retryable=" + e.isRetryable());
            throw e;
        }
    }
}