- Notifications
You must be signed in to change notification settings - Fork 2.8k
/
Copy pathtest_bedrock_nova_json.py
34 lines (27 loc) · 1.04 KB
/
test_bedrock_nova_json.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
frombase_llm_unit_testsimportBaseLLMChatTest
importpytest
importsys
importos
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
importlitellm
classTestBedrockNovaJson(BaseLLMChatTest):
defget_base_completion_call_args(self) ->dict:
litellm._turn_on_debug()
return {
"model": "bedrock/converse/us.amazon.nova-micro-v1:0",
}
deftest_json_response_nested_pydantic_obj(self):
pass
deftest_json_response_nested_json_schema(self):
pass
deftest_tool_call_no_arguments(self, tool_call_no_arguments):
"""Test that tool calls with no arguments is translated correctly. Relevant issue: https://github.com/BerriAI/litellm/issues/6833"""
pass
@pytest.fixture(autouse=True)
defskip_non_json_tests(self, request):
ifnot"json"inrequest.function.__name__.lower():
pytest.skip(
f"Skipping non-JSON test: {request.function.__name__} does not contain 'json'"
)