fix(ai-proxy): bedrock support additional request fields (#2480)

This commit is contained in:
HaoJie Liu
2025-06-26 11:29:32 +08:00
committed by GitHub
parent 527e922d50
commit 864bf5af39
4 changed files with 31 additions and 11 deletions

View File

@@ -733,11 +733,16 @@ func (b *bedrockProvider) buildBedrockTextGenerationRequest(origRequest *chatCom
Temperature: origRequest.Temperature,
TopP: origRequest.TopP,
},
AdditionalModelRequestFields: map[string]interface{}{},
AdditionalModelRequestFields: make(map[string]interface{}),
PerformanceConfig: PerformanceConfiguration{
Latency: "standard",
},
}
for key, value := range b.config.bedrockAdditionalFields {
request.AdditionalModelRequestFields[key] = value
}
requestBytes, err := json.Marshal(request)
b.setAuthHeaders(requestBytes, headers)
return requestBytes, err