3a73309006
* build(model_prices_and_context_window.json): add bedrock llama4 models to model cost map * fix template conversion for Llama 4 models in Bedrock (#10557) * test: add testing to repro https://github.com/BerriAI/litellm/pull/10557 * test: add unit testing * test(test_main.py): refactor where test is kept --------- Co-authored-by: aswny <87371411+aswny@users.noreply.github.com>
23 lines
517 B
Python
23 lines
517 B
Python
from base_llm_unit_tests import BaseLLMChatTest
|
|
import pytest
|
|
import sys
|
|
import os
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system path
|
|
import litellm
|
|
|
|
|
|
class TestBedrockTestSuite(BaseLLMChatTest):
|
|
def test_tool_call_no_arguments(self, tool_call_no_arguments):
|
|
pass
|
|
|
|
def get_base_completion_call_args(self) -> dict:
|
|
litellm._turn_on_debug()
|
|
return {
|
|
"model": "bedrock/converse/us.meta.llama3-3-70b-instruct-v1:0",
|
|
}
|
|
|
|
|