mirror of
https://github.com/mudler/LocalAI.git
synced 2025-03-13 15:56:39 +00:00
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
|
||
|
|
}
|