Function generateSchemaFromAtoms

tsunami/engine/schema.go:250–292

generateSchemaFromAtoms generates a JSON schema from a map of atoms

(atoms map[string]genAtom, title, description string)

// generateSchemaFromAtoms generates a JSON schema from a map of atoms
func generateSchemaFromAtoms(atoms map[string]genAtom, title, description string) map[string]any {
	// Collect all struct definitions
	defs := make(map[reflect.Type]any)
	for _, atom := range atoms {
		atomType := atom.GetAtomType()
		if atomType != nil {
			collectStructDefs(atomType, defs)
		}
	}

	// Generate properties for each atom
	properties := make(map[string]any)
	for atomName, atom := range atoms {
		atomType := atom.GetAtomType()
		if atomType != nil {
			atomMeta := getAtomMeta(atom)
			properties[atomName] = generateShallowJSONSchema(atomType, atomMeta)
		}
	}

	// Build the final schema
	// schema line unnecessary for AI (and burns tokens)
	// also dropping title since it is mostly redundant
	schema := map[string]any{
		// "$schema":              "https://json-schema.org/draft/2020-12/schema",
		"type": "object",
		// "title":                title,
		"description":          description,
		"properties":           properties,
		"additionalProperties": false,
	}

	// Add definitions if any
	if len(defs) > 0 {
		definitions := make(map[string]any)
		for t, def := range defs {
			definitions[t.Name()] = def
		}
		schema["$defs"] = definitions
	}

	return schema
}

// GenerateConfigSchema generates a JSON schema for all config atoms
func GenerateConfigSchema(root *RootElem) map[string]any {