-
Notifications
You must be signed in to change notification settings - Fork 1
/
embeddings.go
38 lines (33 loc) · 1.54 KB
/
embeddings.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package openai_gosdk
const embeddingsURL = "https://api.openai.com/v1/embeddings"
type RequestEmbeddings struct {
// string Required
// ID of the model to use. You can use the List models API to see all of your available models, or see our Model overview for descriptions of them. https://platform.openai.com/docs/api-reference/models/list
Model *string `json:"model"`
// string or array Required
// Input text to get embeddings for, encoded as a string or array of tokens. To get embeddings for multiple inputs in a single request, pass an array of strings or array of token arrays. Each input must not exceed 8192 tokens in length.
Input StrongOrArray `json:"input"`
// string Optional
// A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. Learn more. https://platform.openai.com/docs/guides/safety-best-practices https://platform.openai.com/docs/api-reference/models/list
User string `json:"user,omitempty"`
}
type ResponseEmbeddings struct {
Object string `json:"object"`
Data []struct {
Object string `json:"object"`
Embedding []interface{} `json:"embedding"`
Index int `json:"index"`
} `json:"data"`
Model string `json:"model"`
Usage struct {
PromptTokens int `json:"prompt_tokens"`
TotalTokens int `json:"total_tokens"`
} `json:"usage"`
}
func NewEmbeddings(baseOpenAI BaseOpenAI) OpenAI[RequestEmbeddings, ResponseEmbeddings] {
return OpenAI[RequestEmbeddings, ResponseEmbeddings]{
BaseOpenAI: baseOpenAI,
TargetURL: embeddingsURL,
Method: POST,
}
}