From b0a48d516c53c4d44eb096fe61a77853404fceed Mon Sep 17 00:00:00 2001
From: mateo <mateo@berri.ai>
Date: Thu, 30 Jul 2026 01:59:25 +0000
Subject: [PATCH] test(fireworks_ai): align Kimi output-limit expectations with
 cost map fix

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 tests/test_litellm/test_utils.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index b22e69f0942..3b2bc7e647d 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -4432,7 +4432,7 @@ _FIREWORKS_MODELS = [
         4e-06,
         1.9e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4442,7 +4442,7 @@ _FIREWORKS_MODELS = [
         8e-06,
         3.8e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4452,7 +4452,7 @@ _FIREWORKS_MODELS = [
         4e-06,
         1.6e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4462,7 +4462,7 @@ _FIREWORKS_MODELS = [
         8e-06,
         3e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),