mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-05 08:23:17 +00:00
Merge pull request #15668 from BerriAI/litellm_fix_bedrock_gpt_oss
[Fix] GPT-OSS in Bedrock now supports streaming. Revert fake streaming
This commit is contained in:
@@ -1439,11 +1439,6 @@ class AmazonConverseConfig(BaseConfig):
|
||||
if stream is True:
|
||||
if model is not None:
|
||||
###################################################################
|
||||
# GPT-OSS models do not support streaming
|
||||
###################################################################
|
||||
if "gpt-oss" in model:
|
||||
return True
|
||||
###################################################################
|
||||
# AI21 models do not support streaming
|
||||
###################################################################
|
||||
if "ai21" in model:
|
||||
|
||||
@@ -13,7 +13,6 @@ from litellm.llms.bedrock.chat.converse_transformation import AmazonConverseConf
|
||||
|
||||
class TestBedrockGPTOSS(BaseLLMChatTest):
|
||||
def get_base_completion_call_args(self) -> dict:
|
||||
litellm._turn_on_debug()
|
||||
return {
|
||||
"model": "bedrock/converse/openai.gpt-oss-20b-1:0",
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user