explicitly enable test

This commit is contained in:
Chester Curme 2025-05-21 09:27:16 -04:00
parent 7285b173b8
commit 272efba5b9
5 changed files with 21 additions and 0 deletions

View File

@ -41,6 +41,10 @@ class TestAnthropicStandard(ChatModelIntegrationTests):
def supports_anthropic_inputs(self) -> bool:
return True
@property
def enable_vcr_tests(self) -> bool:
return True
@property
def supported_usage_metadata_details(
self,

View File

@ -38,6 +38,10 @@ class TestAzureOpenAIStandard(ChatModelIntegrationTests):
def supports_json_mode(self) -> bool:
return True
@property
def enable_vcr_tests(self) -> bool:
return True
class TestAzureOpenAIStandardLegacy(ChatModelIntegrationTests):
"""Test a legacy model."""
@ -58,3 +62,7 @@ class TestAzureOpenAIStandardLegacy(ChatModelIntegrationTests):
@property
def structured_output_kwargs(self) -> dict:
return {"method": "function_calling"}
@property
def enable_vcr_tests(self) -> bool:
return True

View File

@ -57,6 +57,10 @@ class TestOpenAIStandard(ChatModelIntegrationTests):
]:
return {"invoke": ["reasoning_output", "cache_read_input"], "stream": []}
@property
def enable_vcr_tests(self) -> bool:
return True
def invoke_with_cache_read_input(self, *, stream: bool = False) -> AIMessage:
with open(REPO_ROOT_DIR / "README.md") as f:
readme = f.read()

View File

@ -2681,6 +2681,7 @@ class ChatModelIntegrationTests(ChatModelTests):
)
assert isinstance(response, AIMessage)
@pytest.mark.benchmark
def test_stream_time(
self, model: BaseChatModel, benchmark: BenchmarkFixture, vcr: vcr.VCR
) -> None:

View File

@ -201,6 +201,10 @@ class ChatModelTests(BaseStandardTests):
content."""
return False
@property
def enable_vcr_tests(self) -> bool:
return False
@property
def supported_usage_metadata_details(
self,