mirror of
https://github.com/mudler/LocalAI.git
synced 2025-05-20 10:35:01 +00:00
docs(swagger): enhance coverage of APIs (#2753)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
parent
401ee553f4
commit
f120a0c9f9
3 changed files with 29 additions and 12 deletions
|
@ -123,7 +123,10 @@ func getFileFromRequest(c *fiber.Ctx) (*File, error) {
|
||||||
return nil, fmt.Errorf("unable to find file id %s", id)
|
return nil, fmt.Errorf("unable to find file id %s", id)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFilesEndpoint https://platform.openai.com/docs/api-reference/files/retrieve
|
// GetFilesEndpoint is the OpenAI API endpoint to get files https://platform.openai.com/docs/api-reference/files/retrieve
|
||||||
|
// @Summary Returns information about a specific file.
|
||||||
|
// @Success 200 {object} File "Response"
|
||||||
|
// @Router /v1/files/{file_id} [get]
|
||||||
func GetFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
func GetFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
||||||
return func(c *fiber.Ctx) error {
|
return func(c *fiber.Ctx) error {
|
||||||
file, err := getFileFromRequest(c)
|
file, err := getFileFromRequest(c)
|
||||||
|
@ -135,13 +138,17 @@ func GetFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.Applicat
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// DeleteFilesEndpoint https://platform.openai.com/docs/api-reference/files/delete
|
type DeleteStatus struct {
|
||||||
|
Id string
|
||||||
|
Object string
|
||||||
|
Deleted bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteFilesEndpoint is the OpenAI API endpoint to delete files https://platform.openai.com/docs/api-reference/files/delete
|
||||||
|
// @Summary Delete a file.
|
||||||
|
// @Success 200 {object} DeleteStatus "Response"
|
||||||
|
// @Router /v1/files/{file_id} [delete]
|
||||||
func DeleteFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
func DeleteFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
||||||
type DeleteStatus struct {
|
|
||||||
Id string
|
|
||||||
Object string
|
|
||||||
Deleted bool
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(c *fiber.Ctx) error {
|
return func(c *fiber.Ctx) error {
|
||||||
file, err := getFileFromRequest(c)
|
file, err := getFileFromRequest(c)
|
||||||
|
@ -174,7 +181,11 @@ func DeleteFilesEndpoint(cm *config.BackendConfigLoader, appConfig *config.Appli
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFilesContentsEndpoint https://platform.openai.com/docs/api-reference/files/retrieve-contents
|
// GetFilesContentsEndpoint is the OpenAI API endpoint to get files content https://platform.openai.com/docs/api-reference/files/retrieve-contents
|
||||||
|
// @Summary Returns information about a specific file.
|
||||||
|
// @Success 200 {string} binary "file"
|
||||||
|
// @Router /v1/files/{file_id}/content [get]
|
||||||
|
// GetFilesContentsEndpoint
|
||||||
func GetFilesContentsEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
func GetFilesContentsEndpoint(cm *config.BackendConfigLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
|
||||||
return func(c *fiber.Ctx) error {
|
return func(c *fiber.Ctx) error {
|
||||||
file, err := getFileFromRequest(c)
|
file, err := getFileFromRequest(c)
|
||||||
|
|
|
@ -6,6 +6,10 @@ import (
|
||||||
"github.com/mudler/LocalAI/core/services"
|
"github.com/mudler/LocalAI/core/services"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// ListModelsEndpoint is the OpenAI Models API endpoint https://platform.openai.com/docs/api-reference/models
|
||||||
|
// @Summary List and describe the various models available in the API.
|
||||||
|
// @Success 200 {object} schema.ModelsDataResponse "Response"
|
||||||
|
// @Router /v1/models [get]
|
||||||
func ListModelsEndpoint(lms *services.ListModelsService) func(ctx *fiber.Ctx) error {
|
func ListModelsEndpoint(lms *services.ListModelsService) func(ctx *fiber.Ctx) error {
|
||||||
return func(c *fiber.Ctx) error {
|
return func(c *fiber.Ctx) error {
|
||||||
// If blank, no filter is applied.
|
// If blank, no filter is applied.
|
||||||
|
@ -18,10 +22,7 @@ func ListModelsEndpoint(lms *services.ListModelsService) func(ctx *fiber.Ctx) er
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return c.JSON(struct {
|
return c.JSON(schema.ModelsDataResponse{
|
||||||
Object string `json:"object"`
|
|
||||||
Data []schema.OpenAIModel `json:"data"`
|
|
||||||
}{
|
|
||||||
Object: "list",
|
Object: "list",
|
||||||
Data: dataModels,
|
Data: dataModels,
|
||||||
})
|
})
|
||||||
|
|
|
@ -155,3 +155,8 @@ type OpenAIRequest struct {
|
||||||
// AutoGPTQ
|
// AutoGPTQ
|
||||||
ModelBaseName string `json:"model_base_name" yaml:"model_base_name"`
|
ModelBaseName string `json:"model_base_name" yaml:"model_base_name"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ModelsDataResponse struct {
|
||||||
|
Object string `json:"object"`
|
||||||
|
Data []OpenAIModel `json:"data"`
|
||||||
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue