package agentteambridge

import (
	"encoding/json"
	"io"

	"github.com/tidbcloud/AgentTeam/internal/apr/types"
)

func marshalEmbeddingRequest(request *types.EmbeddingRequest) ([]byte, error) {
	return json.Marshal(request)
}

func decodeEmbeddingResponse(reader io.Reader) (*types.EmbeddingResponse, error) {
	var response types.EmbeddingResponse
	if err := json.NewDecoder(reader).Decode(&response); err != nil {
		return nil, err
	}
	return &response, nil
}