Help us shape the future of Go! Take the 2026 Go Developer Survey

generation

package
v0.1.2 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 3, 2026 License: MIT Imports: 12 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var (
	CommonInputFields  = []string{"query", "input", "question", "message", "prompt", "text"}
	CommonOutputFields = []string{"answer", "output", "response", "result"}
)

Common field names for heuristic extraction.

Functions

func ExportToFile

func ExportToFile(dataset []map[string]any, outputPath string)

ExportToFile exports a generated dataset to a file (JSON or CSV).

func ExportToLangSmith

func ExportToLangSmith(ctx context.Context, sdk *langsmith.Client, dataset []map[string]any, datasetName string, datasetType string)

ExportToLangSmith uploads a generated dataset to LangSmith.

func ExtractDocuments

func ExtractDocuments(outputs map[string]any) []string

ExtractDocuments extracts document text from retriever outputs.

func ExtractFinalOutput

func ExtractFinalOutput(runs []RunData, outputFields []string) string

ExtractFinalOutput searches all runs newest-first for any run with matching outputs.

func ExtractFromMessages

func ExtractFromMessages(messages []any, role string) string

ExtractFromMessages extracts content from a messages array by role.

func ExtractToolSequence

func ExtractToolSequence(runs []RunData, depth *int) []string

ExtractToolSequence extracts the sequence of tool names from runs.

func ExtractTraceInputs

func ExtractTraceInputs(root RunData, inputFields []string, asDict bool) any

ExtractTraceInputs extracts the primary input from a trace's root run.

func ExtractTraceOutput

func ExtractTraceOutput(root RunData, outputFields []string, messagesOnly bool) string

ExtractTraceOutput extracts the primary output from a trace's root run.

func ExtractValue

func ExtractValue(data map[string]any, fields []string, commonFields []string, messageRole string, fallbackToRaw bool) any

ExtractValue extracts a value from a dict using a priority chain.

func FindRetrievalData

func FindRetrievalData(runs []RunData) map[string]any

FindRetrievalData finds retrieval data from retriever runs.

func GenerateDataset

func GenerateDataset(traces []Trace, datasetType string, runName string, depth *int, inputFields []string, outputFields []string, messagesOnly bool, samplePerTrace *int) []map[string]any

GenerateDataset generates a dataset from traces.

func GetNodeIO

func GetNodeIO(runs []RunData, runName string) []map[string]any

GetNodeIO gets input/output for runs matching a name.

Types

type RunData

type RunData struct {
	RunID       string         `json:"run_id"`
	TraceID     string         `json:"trace_id"`
	Name        string         `json:"name"`
	RunType     string         `json:"run_type"`
	ParentRunID string         `json:"parent_run_id"`
	StartTime   string         `json:"start_time"`
	EndTime     string         `json:"end_time"`
	Status      string         `json:"status"`
	Inputs      map[string]any `json:"inputs"`
	Outputs     map[string]any `json:"outputs"`
	Error       string         `json:"error"`
	Tags        []string       `json:"tags"`
}

RunData represents a single run loaded from a trace file.

type Trace

type Trace struct {
	TraceID string
	Root    RunData
	Runs    []RunData
}

Trace represents a loaded trace: (trace_id, root_run, all_runs).

func LoadTracesFromDir

func LoadTracesFromDir(inputDir string, sortOrder string) ([]Trace, error)

LoadTracesFromDir loads traces from a directory of JSONL files.

func LoadTracesFromFile

func LoadTracesFromFile(inputFile string, sortOrder string) ([]Trace, error)

LoadTracesFromFile loads traces from a single file.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL