feat: support ollama (close #112)

This commit is contained in:
CaIon
2024-03-12 02:36:39 +08:00
parent c3a01decd8
commit d53d3386e9
8 changed files with 117 additions and 4 deletions

View File

@@ -0,0 +1,31 @@
package ollama
import "one-api/dto"
func requestOpenAI2Ollama(request dto.GeneralOpenAIRequest) *OllamaRequest {
messages := make([]dto.Message, 0, len(request.Messages))
for _, message := range request.Messages {
messages = append(messages, dto.Message{
Role: message.Role,
Content: message.Content,
})
}
str, ok := request.Stop.(string)
var Stop []string
if ok {
Stop = []string{str}
} else {
Stop, _ = request.Stop.([]string)
}
return &OllamaRequest{
Model: request.Model,
Messages: messages,
Stream: request.Stream,
Options: &OllamaOptions{
Temperature: request.Temperature,
Seed: request.Seed,
Topp: request.TopP,
Stop: Stop,
},
}
}