From 2715a7499a642c99901e6cfe5e9a6ea3aecedb0f Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Thu, 23 Apr 2026 16:08:55 -0400 Subject: [PATCH] fix(fireworks): swap undeployed Kimi K2 slug in integration tests (#36975) Replace `accounts/fireworks/models/kimi-k2-instruct-0905` with `accounts/fireworks/models/kimi-k2p6` across the Fireworks integration tests. Fireworks appears to have pulled the 0905 slug from serverless (returns 404 `NOT_FOUND` despite still appearing "Ready" in their UI); `kimi-k2p6` is the current deployed successor and supports the same capabilities used by these tests (tool calls, streaming, structured output). --- .../fireworks/tests/integration_tests/test_chat_models.py | 6 +++--- .../fireworks/tests/integration_tests/test_standard.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py index a41f439b0c6..a36fa4f6f6f 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py +++ b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py @@ -21,7 +21,7 @@ _MODEL = "accounts/fireworks/models/gpt-oss-120b" @pytest.mark.parametrize("strict", [None, True, False]) def test_tool_choice_bool(strict: bool | None) -> None: # noqa: FBT001 """Test that tool choice is respected with different strict values.""" - llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2-instruct-0905") + llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2p6") class MyTool(BaseModel): name: str @@ -59,7 +59,7 @@ def test_tool_choice_bool(strict: bool | None) -> None: # noqa: FBT001 async def test_astream() -> None: """Test streaming tokens from ChatFireworks.""" - llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2-instruct-0905") + llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2p6") full: BaseMessageChunk | None = None chunks_with_token_counts = 0 @@ -157,7 +157,7 @@ def _get_joke_class( @pytest.mark.parametrize("schema_type", ["pydantic", "typeddict", "json_schema"]) def test_structured_output_json_schema(schema_type: str) -> None: - llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2-instruct-0905") + llm = ChatFireworks(model="accounts/fireworks/models/kimi-k2p6") schema, validation_function = _get_joke_class(schema_type) # type: ignore[arg-type] chat = llm.with_structured_output(schema, method="json_schema") diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index d60a4be13db..7f1b1de5299 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/tests/integration_tests/test_standard.py @@ -18,7 +18,7 @@ class TestFireworksStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/kimi-k2-instruct-0905", + "model": "accounts/fireworks/models/kimi-k2p6", "temperature": 0, }