translation.go

package hfapigo

import (
	"encoding/json"
	"errors"
)

const (
	RecommendedRussianToEnglishModel = "Helsinki-NLP/opus-mt-ru-en"
)

// Request structure for the Translation endpoint
type TranslationRequest struct {
	// (Required) a string to be translated in the original languages
	Inputs []string `json:"inputs,omitempty"`

	Options Options `json:"options,omitempty"`
}

// Response structure from the Translation endpoint
type TranslationResponse struct {
	// The translated Input string
	TranslationText string `json:"translation_text,omitempty"`
}

func SendTranslationRequest(model string, request *TranslationRequest) ([]*TranslationResponse, error) {
	if request == nil {
		return nil, errors.New("nil TranslationRequest")
	}

	jsonBuf, err := json.Marshal(request)
	if err != nil {
		return nil, err
	}

	respBody, err := MakeHFAPIRequest(jsonBuf, model)
	if err != nil {
		return nil, err
	}

	tresps := make([]*TranslationResponse, len(request.Inputs))
	err = json.Unmarshal(respBody, &tresps)
	if err != nil {
		return nil, err
	}

	return tresps, nil
}