mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-14 11:08:37 +00:00
Krish Dholakia
d6bdd0473a
* test(base_llm_unit_tests.py): return '<thinking>' tag in response content * fix(converse_transformation.py): extract `<thinking>` block from nova tool use response Fixes https://github.com/BerriAI/litellm/issues/9063 * fix(factory.py): handle non-signature reasoning blocks to bedrock pass as text input - bedrock raises ""User messages cannot contain reasoning content. Please remove the r easoning content and try again." otherwise * fix(main.py): Add drop params support for gpt Fixes https://github.com/BerriAI/litellm/issues/10501 * fix(converse_transformation.py): fix linting error * fix(utils.py): fix linting error * test: cleanup test * test: skip test until we have bedrock prompt caching permission
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
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 TestBedrockNovaJson(BaseLLMChatTest):
|
|
def get_base_completion_call_args(self) -> dict:
|
|
litellm._turn_on_debug()
|
|
return {
|
|
"model": "bedrock/converse/us.amazon.nova-micro-v1:0",
|
|
}
|
|
|
|
def test_json_response_nested_pydantic_obj(self):
|
|
pass
|
|
|
|
def test_json_response_nested_json_schema(self):
|
|
pass
|
|
|
|
def test_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
|
|
|
|
def test_prompt_caching(self):
|
|
"""
|
|
Remove override once we have access to Bedrock prompt caching
|
|
"""
|
|
pass
|
|
|
|
# @pytest.fixture(autouse=True)
|
|
# def skip_non_json_tests(self, request):
|
|
# if not "json" in request.function.__name__.lower():
|
|
# pytest.skip(
|
|
# f"Skipping non-JSON test: {request.function.__name__} does not contain 'json'"
|
|
# )
|