typesafego

import typesafe "github.com/adibhanna/typesafego"

package main

import (
	"context"
	"fmt"
	"log"

	typesafe "github.com/adibhanna/typesafego"
)

func main() {
	client, err := typesafe.NewClient() // reads TYPESAFE_API_KEY
	if err != nil {
		log.Fatal(err)
	}

	resp, err := client.SystemOne(context.Background(), typesafe.SystemOneRequest{
		State: "My payment failed twice and I still got charged. Fix this now!",
		Questions: typesafe.Questions{
			"is_urgent": typesafe.Noul("Does the customer convey urgency?"),
			"department": typesafe.Choice("Which team should handle this?", map[string]any{
				"billing":   "Payments, invoices, refunds",
				"technical": "Bugs, outages, integrations",
				"sales":     "Pricing, upgrades, new accounts",
			}),
			"frustration": typesafe.Score("How frustrated is the customer?",
				"Calm", "Frustrated", "Very angry"),
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	urgent, _ := resp.Noul("is_urgent")
	department, _ := resp.Choice("department")
	frustration, _ := resp.Score("frustration")

	fmt.Println(urgent.IsTrue(0.5), urgent.Noul)                 // true 0.93
	fmt.Println(department.Choice, department.Confidence)        // billing 0.81
	fmt.Println(frustration.Score, frustration.Label())          // 1.7 Very angry
	fmt.Println(resp.Model, resp.Usage.InputTokens)              // jev-1.13.0 312
}

typesafe.Noul("Is the customer asking for a refund?")

// Optionally spell out what a yes and a no mean.
typesafe.Noul("Is the customer asking for a refund?").WithCriteria(
	"The customer explicitly asks for money back or a chargeback.",
	"Anything else, including billing questions without a refund request.",
)

// Options with descriptions. A nil description sends null.
typesafe.Choice("Which team should handle this?", map[string]any{
	"billing":   "Payments, invoices, refunds",
	"technical": map[string]any{"handles": []string{"bugs", "outages"}},
	"sales":     nil,
})

// Options without descriptions.
typesafe.ChoiceOf("What is the sentiment?", "positive", "neutral", "negative")

typesafe.Score("How frustrated is the customer?",
	"Calm",         // level 0
	"Frustrated",   // level 1
	"Very angry",   // level 2
)

typesafe.RawQuestion{"type": "noul", "instructions": "Is this spam?"}

department, err := resp.Choice("department")
if err != nil {
	return err // typesafe: answer not found: "department"
}

department, _ := resp.Choice("department")
if department.Confidence < 0.7 {
	return routeToHuman(ticket, department.Ranked())
}
return routeTo(department.Choice)

priority := 0.4*frustration.Normalized() + 0.35*urgent.Noul + 0.25*churn.Noul

policy := typesafe.DefaultRetryPolicy()
policy.MaxRetries = 4

client, err := typesafe.NewClient(
	typesafe.WithAPIKey(os.Getenv("MY_TYPESAFE_KEY")),
	typesafe.WithModel(typesafe.ModelJevPreview),
	typesafe.WithTimeout(20*time.Second),
	typesafe.WithRetryPolicy(policy),
	typesafe.WithLogger(slog.Default()),
)

resp, err := client.SystemOne(ctx, req, typesafe.WithModel("jev-1.13.0"), typesafe.WithMaxRetries(0))

var apiErr *typesafe.APIError
switch {
case errors.As(err, &apiErr):
	// Non-2xx response after retries. StatusCode, Message, Body, Header,
	// and RequestID are available; RetryAfter() reads the retry headers.
	log.Printf("%d %s (request id %s)", apiErr.StatusCode, apiErr.Message, apiErr.RequestID)
case errors.Is(err, typesafe.ErrInvalidRequest):
	// Rejected locally before sending: nil state, no questions, a choice
	// without options, a score with fewer than two levels, ...
case errors.Is(err, typesafe.ErrInvalidResponse):
	// The body could not be decoded.
default:
	// Transport failure or context cancellation; errors.Is works on the
	// underlying error, e.g. errors.Is(err, context.DeadlineExceeded).
}

server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")
	io.WriteString(w, `{"model":"jev-1.13.0","answers":{"is_urgent":{"type":"noul","noul":0.9}},"usage":{"input_tokens":10,"output_tokens":1}}`)
}))
defer server.Close()

client, err := typesafe.NewClient(
	typesafe.WithAPIKey("test"),
	typesafe.WithBaseURL(server.URL),
	typesafe.WithMaxRetries(0),
)

answer := &typesafe.ScoreAnswer{Score: 1.6, Legend: []any{"Calm", "Frustrated", "Very angry"}}