cli.go

package clip_as_service_rs_go_cli

import (
	"context"
	pb "github.com/Rorical/clip-as-service-rs-go-cli/encoder"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

type ClipRsClient struct {
	client pb.EncoderClient
}

func NewClipRsClient(uri string) (*ClipRsClient, error) {
	conn, err := grpc.Dial(uri, grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		return nil, err
	}
	cli := pb.NewEncoderClient(conn)
	return &ClipRsClient{
		client: cli,
	}, nil
}

func (cli *ClipRsClient) EncodeText(ctx context.Context, texts []string) ([][]float32, error) {
	request := &pb.EncodeTextRequest{Texts: texts}
	response, err := cli.client.EncodeText(ctx, request, grpc.MaxCallSendMsgSize(10000000))
	if err != nil {
		return nil, err
	}
	batch := make([][]float32, len(response.GetEmbedding()))
	for i, b := range response.GetEmbedding() {
		batch[i] = b.Point
	}
	return batch, nil
}

func (cli *ClipRsClient) EncodeImage(ctx context.Context, images [][]byte) ([][]float32, error) {
	request := &pb.EncodeImageRequest{Images: images}
	response, err := cli.client.EncodeImage(ctx, request, grpc.MaxCallSendMsgSize(10000000))
	if err != nil {
		return nil, err
	}
	batch := make([][]float32, len(response.GetEmbedding()))
	for i, b := range response.GetEmbedding() {
		batch[i] = b.Point
	}
	return batch, nil
}