Skip to content

Commit 46eb131

Browse files
authored
fix: image edit mode check (#161)
1 parent 8c7a94a commit 46eb131

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

core/middleware/distributor.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -280,6 +280,9 @@ func CheckRelayMode(requestMode mode.Mode, modelMode mode.Mode) bool {
280280
return modelMode == mode.ChatCompletions ||
281281
modelMode == mode.Completions ||
282282
modelMode == mode.Anthropic
283+
case mode.ImagesGenerations, mode.ImagesEdits:
284+
return modelMode == mode.ImagesGenerations ||
285+
modelMode == mode.ImagesEdits
283286
default:
284287
return requestMode == modelMode
285288
}

0 commit comments

Comments
 (0)