forked from franciscoescher/goopenai
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinterface.go
55 lines (41 loc) · 3.41 KB
/
interface.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package goopenai
import "context"
// ClientInterface is the interface for the OpenAI API client.
// Documentation at https://platform.openai.com/docs/api-reference/introduction
type ClientInterface interface {
// ListModels Lists the currently available models, and provides basic information about each one such as the owner and availability.
ListModels(ctx context.Context) (response ListModelsResponse, err error)
// RetrieveModel Retrieves a model instance, providing basic information about the model such as the owner and permissioning.
RetrieveModel(ctx context.Context, id string) (response RetrieveModelResponse, err error)
// CreateCompletions Creates a completion for the provided prompt and parameters
CreateCompletions(ctx context.Context, r CreateCompletionsRequest) (response CreateCompletionsResponse, err error)
// CreateChats Creates a completion for the chat message
CreateChats(ctx context.Context, r CreateChatsRequest) (response CreateChatsResponse, err error)
// CreateEdits Creates a new edit for the provided input, instruction, and parameters.
CreateEdits(ctx context.Context, r CreateEditsRequest) (response CreateEditsResponse, err error)
// CreateImages Creates an image given a prompt.
CreateImages(ctx context.Context, r CreateImagesRequest) (response CreateImagesResponse, err error)
// CreateImagesEdits Creates an edited or extended image given an original image and a prompt.
CreateImagesEdits(ctx context.Context, r CreateImagesEditsRequest) (response CreateImagesEditsResponse, err error)
// CreateImagesVariations Creates a variation of a given image.
CreateImagesVariations(ctx context.Context, r CreateImagesVariationsRequest) (response CreateImagesVariationsResponse, err error)
// CreateEmbeddings Creates an embedding vector representing the input text.
CreateEmbeddings(ctx context.Context, r CreateEmbeddingsRequest) (response CreateEmbeddingsResponse, err error)
// CreateTranscriptions Transcribes audio into the input language.
CreateTranscriptions(ctx context.Context, r CreateTranscriptionsRequest) (response CreateTranscriptionsResponse, err error)
// CreateTranslations Translates audio into into English.
CreateTranslations(ctx context.Context, r CreateTranslationsRequest) (response CreateTranslationsResponse, err error)
// CreateModerations Classifies if text violates OpenAI's Content Policy
CreateModerations(ctx context.Context, r CreateModerationsRequest) (response CreateModerationsResponse, err error)
// Not yet implemented:
// ListFiles(ctx context.Context) (response ListFilesResponse, err error)
// UploadFile(ctx context.Context, r UploadFileRequest) (response UploadFileResponse, err error)
// DeleteFile(ctx context.Context, id string) (response DeleteFileResponse, err error)
// RetrieveFile(ctx context.Context, id string) (response RetrieveFileResponse, err error)
// RetrieveFileContent(ctx context.Context, id string) (response RetrieveFileContentResponse, err error)
// CreateFineTune(ctx context.Context, r CreateFineTuneRequest) (response CreateFineTuneResponse, err error)
// ListFineTunes(ctx context.Context) (response ListFineTunesResponse, err error)
// RetrieveFineTunes(ctx context.Context, id string) (response RetrieveFineTunesResponse, err error)
// ListFineTuneEvents(ctx context.Context, id string) (response ListFineTuneEventsResponse, err error)
// DeleteFineTuneModel(ctx context.Context, id string) (response DeleteFineTuneModelResponse, err error)
}