VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add new struct filed dimensions for embedding API (#645)

Browse Source 
* add new struct filed dimensions for embedding API

* docs: remove long single-line comments

* change embedding request param Dimensions type to int
This commit is contained in:
chenhhA
2024-01-29 15:09:56 +08:00
committed by GitHub
parent 4c41f24a99
commit 06ff541559
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
embeddings.go
View File

@@ -157,6 +157,9 @@ type EmbeddingRequest struct {
Model EmbeddingModel `json:"model"` Model EmbeddingModel `json:"model"`
User string `json:"user"` User string `json:"user"`
EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"` EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"`
// Dimensions The number of dimensions the resulting output embeddings should have.
// Only supported in text-embedding-3 and later models.
Dimensions int `json:"dimensions,omitempty"`
} }
func (r EmbeddingRequest) Convert() EmbeddingRequest { func (r EmbeddingRequest) Convert() EmbeddingRequest {
@@ -181,6 +184,9 @@ type EmbeddingRequestStrings struct {
// Currently, only "float" and "base64" are supported, however, "base64" is not officially documented. // Currently, only "float" and "base64" are supported, however, "base64" is not officially documented.
// If not specified OpenAI will use "float". // If not specified OpenAI will use "float".
EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"` EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"`
// Dimensions The number of dimensions the resulting output embeddings should have.
// Only supported in text-embedding-3 and later models.
Dimensions int `json:"dimensions,omitempty"`
} }
func (r EmbeddingRequestStrings) Convert() EmbeddingRequest { func (r EmbeddingRequestStrings) Convert() EmbeddingRequest {
@@ -189,6 +195,7 @@ func (r EmbeddingRequestStrings) Convert() EmbeddingRequest {
Model: r.Model, Model: r.Model,
User: r.User, User: r.User,
EncodingFormat: r.EncodingFormat, EncodingFormat: r.EncodingFormat,
Dimensions: r.Dimensions,
} }
} }
@@ -209,6 +216,9 @@ type EmbeddingRequestTokens struct {
// Currently, only "float" and "base64" are supported, however, "base64" is not officially documented. // Currently, only "float" and "base64" are supported, however, "base64" is not officially documented.
// If not specified OpenAI will use "float". // If not specified OpenAI will use "float".
EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"` EncodingFormat EmbeddingEncodingFormat `json:"encoding_format,omitempty"`
// Dimensions The number of dimensions the resulting output embeddings should have.
// Only supported in text-embedding-3 and later models.
Dimensions int `json:"dimensions,omitempty"`
} }
func (r EmbeddingRequestTokens) Convert() EmbeddingRequest { func (r EmbeddingRequestTokens) Convert() EmbeddingRequest {
@@ -217,6 +227,7 @@ func (r EmbeddingRequestTokens) Convert() EmbeddingRequest {
Model: r.Model, Model: r.Model,
User: r.User, User: r.User,
EncodingFormat: r.EncodingFormat, EncodingFormat: r.EncodingFormat,
Dimensions: r.Dimensions,
} }
} }