fix: update Init method to correctly set RequestMode based on upstream model name prefixes
This commit is contained in:
@@ -38,10 +38,10 @@ func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInf
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *Adaptor) Init(info *relaycommon.RelayInfo) {
|
func (a *Adaptor) Init(info *relaycommon.RelayInfo) {
|
||||||
if strings.HasPrefix(info.UpstreamModelName, "claude-3") {
|
if strings.HasPrefix(info.UpstreamModelName, "claude-2") || strings.HasPrefix(info.UpstreamModelName, "claude-instant") {
|
||||||
a.RequestMode = RequestModeMessage
|
|
||||||
} else {
|
|
||||||
a.RequestMode = RequestModeCompletion
|
a.RequestMode = RequestModeCompletion
|
||||||
|
} else {
|
||||||
|
a.RequestMode = RequestModeMessage
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user