Function tokenClassifyResponseToEntities

core/backend/token_classify.go:132–150

tokenClassifyResponseToEntities converts the wire-format response into the value type consumed by callers. Extracted so the conversion can be unit-tested without a real backend (see token_classify_test.go).

(resp *pb.TokenClassifyResponse)

// the value type consumed by callers. Extracted so the conversion can be
// unit-tested without a real backend (see token_classify_test.go).
func tokenClassifyResponseToEntities(resp *pb.TokenClassifyResponse) []TokenEntity {
	if resp == nil {
		return nil
	}
	out := make([]TokenEntity, 0, len(resp.Entities))
	for _, e := range resp.Entities {
		if e == nil {
			continue
		}
		out = append(out, TokenEntity{
			Group: e.EntityGroup,
			Start: int(e.Start),
			End:   int(e.End),
			Score: e.Score,
			Text:  e.Text,
		})
	}
	return out
}