dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-29 22:20:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

documented remaining endpoints

Browse source
This commit is contained in:
Jacob Payne 2023-06-05 15:25:21 -07:00
parent cfba345021
commit 92a761c0a5
2 changed files with 28 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/api.go
View file

@ -3,7 +3,7 @@ package api
import ( import (
"errors" "errors"
_ "github.com/go-skynet/LocalAI/docs/swagger" _ "github.com/go-skynet/LocalAI/docs"
model "github.com/go-skynet/LocalAI/pkg/model" model "github.com/go-skynet/LocalAI/pkg/model"
"github.com/gofiber/fiber/v2" "github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/cors" "github.com/gofiber/fiber/v2/middleware/cors"

27
api/openai.go
View file

@ -440,6 +440,22 @@ func chatEndpoint(cm ConfigMerger, debug bool, loader *model.ModelLoader, thread
} }
} }
// OpenAI CHAT API endpoint for editing model parameters
// editEndpoint mimics the /edits endpoint for local models
// @Summary Modify model parameters
// @Description Alter model parameters for text completion models.
// @Tags Chat
// @Accept json
// @Produce json
// @Param model body string true "Define which model to modify."
// @Param instruction body string true "Define the initial prompt for the model."
// @Param input body string true "Initial input prompt for model."
// @Param stop body string true "Define stop words for the model to reply after."
// @Success 200 {object} OpenAIResponse
// @Failure 400 {object} APIError
// @Failure 401 {object} APIError
// @Failure 500 {object} APIError
// @Router /edits [post]
func editEndpoint(cm ConfigMerger, debug bool, loader *model.ModelLoader, threads, ctx int, f16 bool) func(c *fiber.Ctx) error { func editEndpoint(cm ConfigMerger, debug bool, loader *model.ModelLoader, threads, ctx int, f16 bool) func(c *fiber.Ctx) error {
return func(c *fiber.Ctx) error { return func(c *fiber.Ctx) error {
config, input, err := readConfig(cm, c, loader, debug, threads, ctx, f16) config, input, err := readConfig(cm, c, loader, debug, threads, ctx, f16)
@ -487,6 +503,17 @@ func editEndpoint(cm ConfigMerger, debug bool, loader *model.ModelLoader, thread
} }
} }
// OpenAI CHAT API endpoint for listing loaded models
// listModels mimics the /models endpoint for listing loaded local models
// @Summary Get a list of all currently loaded models
// @Description List all currently loaded models.
// @Tags Chat
// @Produce json
// @Success 200 {object} OpenAIResponse
// @Failure 400 {object} APIError
// @Failure 401 {object} APIError
// @Failure 500 {object} APIError
// @Router /models [get]
func listModels(loader *model.ModelLoader, cm ConfigMerger) func(ctx *fiber.Ctx) error { func listModels(loader *model.ModelLoader, cm ConfigMerger) func(ctx *fiber.Ctx) error {
return func(c *fiber.Ctx) error { return func(c *fiber.Ctx) error {
models, err := loader.ListModels() models, err := loader.ListModels()