update model

Author: SunnySoldier357

tests/pytest/test_pytest_klavis_mcp.py

@@ -22,7 +22,7 @@ class ResponseFormat(BaseModel):
 @evaluation_test(
     input_dataset=["tests/pytest/datasets/klavis_mcp_test.jsonl"],
     rollout_processor=AgentRolloutProcessor(),
-    completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/kimi-k2-thinking"}],
+    completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/kimi-k2p5"}],
     mode="pointwise",
     mcp_config_path="tests/pytest/mcp_configurations/klavis_strata_mcp.json",
 )
@@ -34,7 +34,7 @@ async def test_pytest_klavis_mcp(row: EvaluationRow) -> EvaluationRow:
         api_key=os.environ["FIREWORKS_API_KEY"], base_url="https://api.fireworks.ai/inference/v1"
     ) as client:
         response = await client.chat.completions.create(
-            model="accounts/fireworks/models/kimi-k2-thinking",
+            model="accounts/fireworks/models/kimi-k2p5",
             messages=[
                 {
                     "role": "system",

tests/pytest/test_pytest_klavis_sandbox.py

@@ -58,7 +58,7 @@ def klavis_gmail_sandbox_dataset_adapter(rows: list[dict]) -> list[EvaluationRow
     rollout_processor=KlavisSandboxRolloutProcessor(
         server_name="gmail",
     ),
-    completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/kimi-k2-thinking"}],
+    completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/kimi-k2p5"}],
     mode="pointwise",
     dataset_adapter=klavis_gmail_sandbox_dataset_adapter,
 )
@@ -110,7 +110,7 @@ async def test_pytest_gmail_sandbox(row: EvaluationRow) -> EvaluationRow:
 
         try:
             response = await client.chat.completions.create(
-                model="accounts/fireworks/models/kimi-k2-thinking",
+                model="accounts/fireworks/models/kimi-k2p5",
                 messages=[
                     {
                         "role": "system",