Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update main.go

Browse Source 
format code
This commit is contained in:
NongMO 2024-04-26 10:38:56 +08:00 committed by GitHub
parent 2b82cd25e4
commit 639af52c23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
relay/adaptor/ollama/main.go
View File

@ -33,14 +33,14 @@ func ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {
Stream: request.Stream, Stream: request.Stream,
} }
for _, message := range request.Messages { for _, message := range request.Messages {
openaiContent := message.ParseContent() openaiContent := message.ParseContent()
var imageUrls []string var imageUrls []string
var contentText string var contentText string
for _, part := range openaiContent { for _, part := range openaiContent {
logger.SysLog(part.Type) logger.SysLog(part.Type)
switch part.Type { switch part.Type {
case model.ContentTypeText: case model.ContentTypeText:
contentText = part.Text contentText = part.Text
case model.ContentTypeImageURL: case model.ContentTypeImageURL:
_ , data, _ := image.GetImageFromUrl(part.ImageURL.Url) _ , data, _ := image.GetImageFromUrl(part.ImageURL.Url)
imageUrls = append(imageUrls, data) imageUrls = append(imageUrls, data)
@ -50,7 +50,7 @@ func ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {
ollamaRequest.Messages = append(ollamaRequest.Messages, Message{ ollamaRequest.Messages = append(ollamaRequest.Messages, Message{
Role: message.Role, Role: message.Role,
Content: contentText, Content: contentText,
Images: imageUrls, Images: imageUrls,
}) })
} }
return &ollamaRequest return &ollamaRequest