tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-14 15:05:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_bedrock_nova_grounding_web_search_options_non_streaming

Browse Source
This commit is contained in:
Ishaan Jaffer
2026-01-31 11:23:48 -08:00
parent 9fe0819e77
commit 8b575f4656
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/llm_translation/test_bedrock_completion.py
+3 -6
View File
@@ -3984,12 +3984,11 @@ def test_bedrock_nova_grounding_web_search_options_non_streaming():
with patch.object(client, "post") as mock_post:
try:
completion(
model="us.amazon.nova-pro-v1:0", # No bedrock/ prefix when using api_base
model="bedrock/us.amazon.nova-pro-v1:0",
messages=messages,
web_search_options={}, # Enables Nova grounding
max_tokens=500,
client=client,
api_base="https://bedrock-runtime.us-east-1.amazonaws.com",
)
except Exception:
pass # Expected - we're just checking the request structure
@@ -4059,13 +4058,12 @@ def test_bedrock_nova_grounding_with_function_tools():
with patch.object(client, "post") as mock_post:
try:
completion(
model="us.amazon.nova-pro-v1:0", # No bedrock/ prefix when using api_base
model="bedrock/us.amazon.nova-pro-v1:0",
messages=messages,
tools=tools,
web_search_options={}, # Also enable web grounding
max_tokens=500,
client=client,
api_base="https://bedrock-runtime.us-east-1.amazonaws.com",
)
except Exception:
pass # Expected - we're just checking the request structure
@@ -4121,12 +4119,11 @@ async def test_bedrock_nova_grounding_async():
with patch.object(client, "post", new=AsyncMock()) as mock_post:
try:
await litellm.acompletion(
model="us.amazon.nova-pro-v1:0", # No bedrock/ prefix when using api_base
model="bedrock/us.amazon.nova-pro-v1:0",
messages=messages,
web_search_options={},
max_tokens=500,
client=client,
api_base="https://bedrock-runtime.us-east-1.amazonaws.com",
)
except Exception:
pass # Expected - we're just checking the request structure