package l_bert

import (
	"encoding/json"

	"fmt"
	"gitea.cdlsxd.cn/self-tools/l_request"
)

type Bert struct {
	host string
	auth string
}

func NewClient(args ...Option) *Bert {
	s := &Bert{
		host: "https://117.175.169.61:5003",
	}
	for _, opt := range args {
		opt(s)
	}
	return s
}

func (s *Bert) BatchCreate(in *Predict) (*PredictRes, error) {
	path := "/predict"

	requestJson, err := StructToMap(in)
	if err != nil {
		return nil, err
	}
	req := l_request.Request{
		Method: "POST",
		Url:    s.host + path,
		Json:   requestJson,
	}
	resp, err := req.Send()

	if err != nil {
		return nil, fmt.Errorf("请求失败，err: %v", err)
	}
	var resData PredictRes
	if err = json.Unmarshal(resp.Content, &resData); err != nil {
		return nil, fmt.Errorf("解析响应失败，err: %v", err)
	}
	if resData.Status != "success" {
		return nil, err
	}
	return &resData, nil
}