Skip to content

Commit

Permalink
Merge pull request #847 from narengogi/fix/bedrock/remove-unsupported…
Browse files Browse the repository at this point in the history
…-params-cohere

Fix error with cohere streaming on bedrock
  • Loading branch information
VisargD authored Dec 26, 2024
2 parents 3a176ac + d4cdade commit 467c321
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 8 deletions.
5 changes: 0 additions & 5 deletions src/providers/bedrock/chatComplete.ts
Original file line number Diff line number Diff line change
Expand Up @@ -553,11 +553,6 @@ export const BedrockConverseCohereChatCompleteConfig: ProviderConfig = {
transform: (params: BedrockConverseCohereChatCompletionsParams) =>
transformCohereAdditionalModelRequestFields(params),
},
stream: {
param: 'additionalModelRequestFields',
transform: (params: BedrockConverseCohereChatCompletionsParams) =>
transformCohereAdditionalModelRequestFields(params),
},
};

export const BedrockConverseAI21ChatCompleteConfig: ProviderConfig = {
Expand Down
3 changes: 0 additions & 3 deletions src/providers/bedrock/utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -120,9 +120,6 @@ export const transformCohereAdditionalModelRequestFields = (
if (params['logit_bias']) {
additionalModelRequestFields['logitBias'] = params['logit_bias'];
}
if (params['stream']) {
additionalModelRequestFields['stream'] = params['stream'];
}
return additionalModelRequestFields;
};

Expand Down

0 comments on commit 467c321

Please sign in to comment.