github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-04-23 20:23:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(fireworks): swap undeployed Kimi K2 slug in integration tests (#36975)

Browse Source 
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).
This commit is contained in:
Mason Daugherty
2026-04-23 16:08:55 -04:00
committed by GitHub
parent 2d3b49162c
commit 2715a7499a
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/fireworks/tests/integration_tests/test_chat_models.py
View File

@@ -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")

2
libs/partners/fireworks/tests/integration_tests/test_standard.py
View File

@@ -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,
}