maas

package
v0.0.0-...-4e0481d Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 22, 2024 License: Apache-2.0 Imports: 25 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	ServiceName       = "ml_maas"
	APICert           = "cert"
	APIChat           = "chat"
	APIStreamChat     = "stream_chat"
	APITokenization   = "tokenization"
	APIClassification = "classification"
	APIEmbeddings     = "embeddings"
	APIImagesQuickGen = "images.quick_gen"
	APIImagesFlexGen  = "images.flex_gen"
	APIAudioSpeech    = "audio.speech"
	TOP               = "top"

	ServiceTimeout = time.Minute

	MaxBufferSize  = 4194304
	RespBufferSize = 32

	Terminator = "[DONE]"

	ChatRoleOfUser      = "user"
	ChatRoleOfAssistant = "assistant"
	ChatRoleOfSystem    = "system"
	ChatRoleOfFunction  = "function"
)

Variables

This section is empty.

Functions

func AesGcmDecryptBase64String 

func AesGcmDecryptBase64String(key, nonce []byte, ciphertext string) (string, error)

AesGcmDecryptBase64String decrypt message(base64.std.string) using AES-GCM

func AesGcmDecryptBytes 

func AesGcmDecryptBytes(key, nonce, cipherBytes []byte) ([]byte, error)

AesGcmDecryptBytes Decrypt message using AES-GCM

func AesGcmEncryptBase64String 

func AesGcmEncryptBase64String(key, nonce []byte, plaintext string) (string, error)

AesGcmEncryptBase64String Encrypt message from base64 string to string using AES-GCM

func AesGcmEncryptBytes 

func AesGcmEncryptBytes(key, nonce, plain []byte) ([]byte, error)

AesGcmEncryptBytes encrypt message using AES-GCM

func ECDHMarshalBinary 

func ECDHMarshalBinary(pub *ecdsa.PublicKey) []byte

func GenerateKey 

func GenerateKey(curve elliptic.Curve) (*ecdsa.PrivateKey, *ecdsa.PublicKey, error)

GenerateKey is the constructor for Private-Public key pair

func GetRetrySetting 

func GetRetrySetting(serviceRetrySettings, apiRetrySettings *base.RetrySettings) *base.RetrySettings

func GetTimeout 

func GetTimeout(serviceTimeout, apiTimeout time.Duration) time.Duration

func MakeRequest 

func MakeRequest(apiInfo *base.ApiInfo, endpointId string, serviceInfo *base.ServiceInfo, query url.Values, ct string) (*http.Request, error)

func MarshalBinary 

func MarshalBinary(pub *ecdsa.PublicKey) []byte

func MergeHeader 

func MergeHeader(header1, header2 http.Header) (header http.Header)

func MergeQuery 

func MergeQuery(query1, query2 url.Values) (query url.Values)

func ReadCertFromBytes 

func ReadCertFromBytes(pemBytes []byte) (*x509.Certificate, error)

func ReadCertFromString 

func ReadCertFromString(pemString string) (*x509.Certificate, error)

func ReadEcdsaPrivKeyFromBytes 

func ReadEcdsaPrivKeyFromBytes(pemBytes []byte) (*ecdsa.PrivateKey, error)

func ReadEcdsaPrivKeyFromString 

func ReadEcdsaPrivKeyFromString(key string) (*ecdsa.PrivateKey, error)

func UnmarshalBinary 

func UnmarshalBinary(curve elliptic.Curve, data []byte) (*big.Int, *big.Int)

Types

type KeyAgreementClient 

type KeyAgreementClient struct {
	// contains filtered or unexported fields
}

func NewP256KeyAgreementClient 

func NewP256KeyAgreementClient(pemString string) (*KeyAgreementClient, error)

NewP256KeyAgreementClient Load cert and extract public key

func (*KeyAgreementClient) CheckChain 

func (k *KeyAgreementClient) CheckChain(pemChainString string) error

CheckChain validate cert and cert chain

func (*KeyAgreementClient) GenerateECIESKeyPair 

func (k *KeyAgreementClient) GenerateECIESKeyPair() ([]byte, []byte, string, error)

GenerateECIESKeyPair generate ECIES key pair and return the tuple (key, nonce, session token, error)

type KeyAgreementVendor 

type KeyAgreementVendor struct {
	// contains filtered or unexported fields
}

func NewP256KeyAgreementVendor 

func NewP256KeyAgreementVendor(key string) (*KeyAgreementVendor, error)

NewP256KeyAgreementVendor Load private key

func (*KeyAgreementVendor) DecryptString 

func (k *KeyAgreementVendor) DecryptString(sessionToken string, ciphertext string) ([]byte, []byte, string, error)

DecryptString decide ciphered text with ECIES DH protocol and return the tuple of (key, nonce, decrypted text)

func (*KeyAgreementVendor) ExtractECIESKeyPair 

func (k *KeyAgreementVendor) ExtractECIESKeyPair(sessionToken string) ([]byte, []byte, error)

ExtractECIESKeyPair extract ECIES key pair from session token and return the tuple of (key, nonce)

type MaaS 

type MaaS struct {
	*base.Client
}

MaaS ... use base client

func NewInstance 

func NewInstance(host, region string) *MaaS

NewInstance ...

func (*MaaS) Chat 

func (cli *MaaS) Chat(req *api.ChatReq) (*api.ChatResp, int, error)

POST method Chat ...

func (*MaaS) ChatImpl 

func (cli *MaaS) ChatImpl(ctx context.Context, body []byte) (*api.ChatResp, int, error)

func (*MaaS) ChatWithCtx 

func (cli *MaaS) ChatWithCtx(ctx context.Context, req *api.ChatReq) (*api.ChatResp, int, error)

POST method ChatWithCtx ...

func (*MaaS) Classification 

func (cli *MaaS) Classification(req *api.ClassificationReq) (*api.ClassificationResp, int, error)

POST method Classification

func (*MaaS) ClassificationWithCtx 

func (cli *MaaS) ClassificationWithCtx(ctx context.Context, req *api.ClassificationReq) (*api.ClassificationResp, int, error)

func (*MaaS) Embeddings 

func (cli *MaaS) Embeddings(req *api.EmbeddingsReq) (*api.EmbeddingsResp, int, error)

POST method Embeddings

func (*MaaS) EmbeddingsWithCtx 

func (cli *MaaS) EmbeddingsWithCtx(ctx context.Context, req *api.EmbeddingsReq) (*api.EmbeddingsResp, int, error)

func (*MaaS) SecretChat 

func (cli *MaaS) SecretChat(req *api.ChatReq) (*api.ChatResp, int, error)

POST method SecretChat is like `Chat`, except its messages are encrypted to ensure that messages are not intercepted by receivers other than the model.

func (*MaaS) SecretChatWithCtx 

func (cli *MaaS) SecretChatWithCtx(ctx context.Context, req *api.ChatReq) (*api.ChatResp, int, error)

POST method SecretChatWithCtx is like `ChatWithCtx`, except its messages are encrypted to ensure that messages are not intercepted by receivers other than the model.

func (*MaaS) SecretStreamChat 

func (cli *MaaS) SecretStreamChat(req *api.ChatReq) (ch <-chan *api.ChatResp, err error)

POST method SecretStreamChat is like `StreamChat`, except its messages are encrypted to ensure that messages are not intercepted by receivers other than the model.

func (*MaaS) SecretStreamChatWithCtx 

func (cli *MaaS) SecretStreamChatWithCtx(ctx context.Context, req *api.ChatReq) (ch <-chan *api.ChatResp, err error)

POST method SecretStreamChatWithCtx is like `StreamChatWithCtx`, except its messages are encrypted to ensure that messages are not intercepted by receivers other than the model.

func (*MaaS) StreamChat 

func (cli *MaaS) StreamChat(req *api.ChatReq) (ch <-chan *api.ChatResp, err error)

POST method StreamChat make stream chat request

  1. if any error returned, a channel=`nil` is returned;
  2. if no error returned, the channel are closed after all responses processed.

func (*MaaS) StreamChatImpl 

func (cli *MaaS) StreamChatImpl(ctx context.Context, body []byte) (<-chan *api.ChatResp, error)

func (*MaaS) StreamChatWithCtx 

func (cli *MaaS) StreamChatWithCtx(ctx context.Context, req *api.ChatReq) (ch <-chan *api.ChatResp, err error)

POST method StreamChat make stream chat request

  1. if any error returned, a channel=`nil` is returned;
  2. if no error returned, the channel are closed after all responses processed.

func (*MaaS) Tokenization 

func (cli *MaaS) Tokenization(req *api.TokenizeReq) (*api.TokenizeResp, int, error)

POST method Tokenization

func (*MaaS) TokenizationWithCtx 

func (cli *MaaS) TokenizationWithCtx(ctx context.Context, req *api.TokenizeReq) (*api.TokenizeResp, int, error)

type ResponseContent 

type ResponseContent interface {
	io.ReadCloser
	GetRequestId() string
}

func NewBinaryResponseContent 

func NewBinaryResponseContent(i io.ReadCloser, requestId string, cancel context.CancelFunc) ResponseContent

Source Files

View all Source files

Directories

Path Synopsis
models
api
api/v2

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.