json mode standard test (#25497)

Co-authored-by: Chester Curme <chester.curme@gmail.com>
This commit is contained in:
Bagatur
2024-12-17 10:47:34 -08:00
committed by GitHub
parent 24bf24270d
commit e4d3ccf62f
8 changed files with 127 additions and 2 deletions

View File

@@ -26,6 +26,10 @@ class BaseTestGroq(ChatModelIntegrationTests):
) -> None:
super().test_tool_message_histories_list_content(model, my_adder_tool)
@property
def supports_json_mode(self) -> bool:
return True
class TestGroqLlama(BaseTestGroq):
@property
@@ -41,6 +45,10 @@ class TestGroqLlama(BaseTestGroq):
"""Value to use for tool choice when used in tests."""
return "any"
@property
def supports_json_mode(self) -> bool:
return False # Not supported in streaming mode
@pytest.mark.xfail(
reason=("Fails with 'Failed to call a function. Please adjust your prompt.'")
)