Go SDK examples

git clone https://github.com/trulayer/demo-go.git
cd demo-go
cp .env.example .env      # add TRULAYER_API_KEY and any provider keys
go run ./examples/basic_trace

TRULAYER_DRY_RUN=true go run ./examples/basic_trace

package main

import (
    "context"
    "log"
    "os"

    "github.com/trulayer/client-go/trulayer"
)

func main() {
    ctx := context.Background()

    tl := trulayer.NewClient(os.Getenv("TRULAYER_API_KEY"))
    defer func() { _ = tl.Shutdown(ctx) }()

    trace, ctx := tl.NewTrace(ctx, "answer-question")
    trace.SetInput("What is the capital of France?")

    span, ctx := trace.NewSpan(ctx, "llm-call", trulayer.SpanTypeLLM,
        trulayer.WithSpanModel("gpt-4o-mini"),
        trulayer.WithSpanInput("What is the capital of France?"),
    )
    // call your LLM here ...
    answer := "Paris."
    span.SetOutput(answer)
    span.SetTokens(12, 4)
    span.End(ctx)

    trace.SetOutput(answer)
    trace.End(ctx)

    if err := tl.Flush(ctx); err != nil {
        log.Printf("flush: %v", err)
    }
}

trace, ctx := tl.NewTrace(ctx, "rag-pipeline")
trace.SetInput(userQuestion)

// Step 1 — retrieval
rs, rctx := trace.NewSpan(ctx, "vector-retrieval", trulayer.SpanTypeRetrieval)
docs := vectorSearch(rctx, userQuestion)
rs.SetOutput("retrieved 3 documents")
rs.SetMetadata(map[string]interface{}{"doc_count": len(docs)})
rs.End(rctx)

// Step 2 — LLM generation
ls, lctx := trace.NewSpan(ctx, "llm-generate", trulayer.SpanTypeLLM,
    trulayer.WithSpanModel("gpt-4o"),
)
answer := callLLM(lctx, userQuestion, docs)
ls.SetOutput(answer)
ls.SetTokens(340, 72)
ls.End(lctx)

trace.SetOutput(answer)
trace.End(ctx)

trace, ctx := tl.NewTrace(ctx, "answer-question")
// ... run the pipeline ...
trace.End(ctx)
traceID := trace.ID() // save this alongside your response

// Later, when you receive user feedback:
err := tl.SubmitFeedback(ctx, traceID, trulayer.FeedbackData{
    Label:   "good",
    Comment: "Correct and concise.",
})
if err != nil {
    log.Printf("feedback: %v", err)
}