From e2da29c54d143f1cc43cf39c55fb5c79f831687b Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 6 Jun 2025 09:15:08 -0700
Subject: [PATCH] test: update test

---
 tests/llm_translation/test_bedrock_completion.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/llm_translation/test_bedrock_completion.py b/tests/llm_translation/test_bedrock_completion.py
index 54cb1fd4ec9..b147132cd07 100644
--- a/tests/llm_translation/test_bedrock_completion.py
+++ b/tests/llm_translation/test_bedrock_completion.py
@@ -3147,6 +3147,8 @@ def test_bedrock_meta_llama_function_calling():
     Tests that:
     - meta llama models support function calling
     """
+    from litellm.utils import return_raw_request
+    from litellm.types.utils import CallTypes
     tools = [
             {
                 "type": "function",
@@ -3182,7 +3184,10 @@ def test_bedrock_meta_llama_function_calling():
         "model": "bedrock/us.meta.llama4-scout-17b-instruct-v1:0",
     }
 
-    response = completion(**request_args)
+    response = return_raw_request(
+        endpoint=CallTypes.completion,
+        kwargs=request_args,
+    )
 
     print(response)