mirror of
https://github.com/mudler/LocalAI.git
synced 2025-03-13 07:54:34 +00:00
* squash past, centralize request middleware PR Signed-off-by: Dave Lee <dave@gray101.com> * migrate bruno request files to examples repo Signed-off-by: Dave Lee <dave@gray101.com> * fix Signed-off-by: Dave Lee <dave@gray101.com> * Update tests/e2e-aio/e2e_test.go Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com> --------- Signed-off-by: Dave Lee <dave@gray101.com> Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com> Co-authored-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
23 lines
688 B
Go
23 lines
688 B
Go
package schema
|
|
|
|
// This file and type represent a generic request to LocalAI - as opposed to requests to LocalAI-specific endpoints, which live in localai.go
|
|
type LocalAIRequest interface {
|
|
ModelName(*string) string
|
|
}
|
|
|
|
type BasicModelRequest struct {
|
|
Model string `json:"model" yaml:"model"`
|
|
// TODO: Should this also include the following fields from the OpenAI side of the world?
|
|
// If so, changes should be made to core/http/middleware/request.go to match
|
|
|
|
// Context context.Context `json:"-"`
|
|
// Cancel context.CancelFunc `json:"-"`
|
|
}
|
|
|
|
func (bmr *BasicModelRequest) ModelName(s *string) string {
|
|
if s != nil {
|
|
bmr.Model = *s
|
|
}
|
|
return bmr.Model
|
|
}
|