From 78a3cbcc6b5e7e39fafda5d7b8df384a4a2c821d Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Mon, 28 Sep 2026 16:41:02 -0400 Subject: [PATCH] hotfix(fireworks): replace unavailable integration test model (#40890) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The [Fireworks release job](https://github.com/langchain-ai/langchain/actions/runs/36479033898/job/109120052547) failed 55 tests because `kimi-k2p6` returned `404 NOT_FOUND`; the `gpt-oss-120b` chat tests passed in that same job. - Use `accounts/fireworks/models/gpt-oss-120b` across the affected chat, completions, and standard integration tests, reusing the existing chat-model constant. - Preserve all assertions and coverage; leave production defaults and release workflows unchanged. - Fireworks [advertises GPT-OSS-120B as serverless](https://fireworks.ai/models/fireworks/gpt-oss-120b). Live completions and expanded chat coverage still need confirmation with CI credentials: no Fireworks API key is available locally. This is not evidence that Kimi was retired. Made by [Open SWE](https://github.com/langchain-ai/open-swe) · [view thread](https://openswe.langchain.dev/agents/b8a0feec-8262-501b-8d6f-204f467461e3) · openai:gpt-6-astra (medium) Co-authored-by: Mason Daugherty Co-authored-by: open-swe[bot] --- .../integration_tests/test_chat_models.py | 18 +++++------------- .../tests/integration_tests/test_llms.py | 2 +- .../tests/integration_tests/test_standard.py | 2 +- 3 files changed, 7 insertions(+), 15 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 eb6d93ef5a..cbbfc3a3b8 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py +++ b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py @@ -22,9 +22,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-k2p6", rate_limiter=rate_limiter - ) + llm = ChatFireworks(model=_MODEL, rate_limiter=rate_limiter) class MyTool(BaseModel): name: str @@ -62,9 +60,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-k2p6", rate_limiter=rate_limiter - ) + llm = ChatFireworks(model=_MODEL, rate_limiter=rate_limiter) full: BaseMessageChunk | None = None chunks_with_token_counts = 0 @@ -162,9 +158,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-k2p6", rate_limiter=rate_limiter - ) + llm = ChatFireworks(model=_MODEL, rate_limiter=rate_limiter) schema, validation_function = _get_joke_class(schema_type) # type: ignore[arg-type] chat = llm.with_structured_output(schema, method="json_schema") @@ -183,7 +177,7 @@ def test_structured_output_json_schema(schema_type: str) -> None: def test_reasoning_effort_parameter() -> None: """Test that the standard `reasoning_effort` parameter is accepted by the API.""" llm = ChatFireworks( - model="accounts/fireworks/models/kimi-k2p6", + model=_MODEL, reasoning_effort="high", rate_limiter=rate_limiter, ) @@ -199,9 +193,7 @@ def test_reasoning_effort_parameter() -> None: def test_reasoning_effort_call_time_kwarg() -> None: """Test that `reasoning_effort` is accepted as a call-time kwarg.""" - llm = ChatFireworks( - model="accounts/fireworks/models/kimi-k2p6", rate_limiter=rate_limiter - ) + llm = ChatFireworks(model=_MODEL, rate_limiter=rate_limiter) result = llm.invoke("Say hello in one sentence", reasoning_effort="high") diff --git a/libs/partners/fireworks/tests/integration_tests/test_llms.py b/libs/partners/fireworks/tests/integration_tests/test_llms.py index 11bc90e7f5..5e630ab865 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_llms.py +++ b/libs/partners/fireworks/tests/integration_tests/test_llms.py @@ -13,7 +13,7 @@ import pytest as pytest from langchain_fireworks import Fireworks -_MODEL = "accounts/fireworks/models/kimi-k2p6" +_MODEL = "accounts/fireworks/models/gpt-oss-120b" def test_fireworks_call() -> None: diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index f6ee0aaf8d..e7a99b7576 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/tests/integration_tests/test_standard.py @@ -19,7 +19,7 @@ class TestFireworksStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/kimi-k2p6", + "model": "accounts/fireworks/models/gpt-oss-120b", "temperature": 0, "rate_limiter": rate_limiter, }