Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions asknews_sdk/api/chat.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,8 @@
"gemini-3-pro",
"claude-sonnet-4-6",
"claude-opus-4-6",
"claude-fable-5",
"claude-fable-5-1",
"claude-opus-4-5-20251101",
"claude-sonnet-4-6",
"claude-sonnet-5",
Expand Down
4 changes: 4 additions & 0 deletions asknews_sdk/dto/alert.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,8 @@
"claude-sonnet-4-6",
"claude-sonnet-5",
"claude-opus-4-6",
"claude-fable-5",
"claude-fable-5-1",
"claude-opus-4-5-20251101",
"gemini-2.5-flash",
"o3",
Expand Down Expand Up @@ -64,6 +66,8 @@
"claude-sonnet-4-5-20250929",
"claude-opus-4-5-20251101",
"claude-opus-4-6",
"claude-fable-5",
"claude-fable-5-1",
"claude-sonnet-4-6",
"claude-sonnet-5",
"gpt-5.6-sol",
Expand Down
8 changes: 8 additions & 0 deletions tests/api/test_chat_model_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@


GPT_6_ASTRA = "gpt-6-astra"
CLAUDE_FABLE_MODELS = ("claude-fable-5", "claude-fable-5-1")


def test_gpt_6_astra_is_a_direct_deepnews_model_only():
Expand Down Expand Up @@ -38,3 +39,10 @@ def test_gpt_6_astra_is_absent_from_alert_model_contracts():
for name, model in getmembers(alert_dto, isclass):
if model.__module__ == alert_dto.__name__ and issubclass(model, BaseModel):
assert GPT_6_ASTRA not in json.dumps(model.model_json_schema()), name


@pytest.mark.parametrize("model", CLAUDE_FABLE_MODELS)
def test_claude_fable_models_match_deepnews_and_alert_report_contracts(model):
assert model in get_args(DeepNewsModel)
assert model in get_args(alert_dto.DeepNewsModel)
assert model in get_args(alert_dto.AlertReportModel)
Loading