diff --git a/litellm/llms/fireworks_ai/chat/transformation.py b/litellm/llms/fireworks_ai/chat/transformation.py index 5de471eba59..3cbe8a8d9dc 100644 --- a/litellm/llms/fireworks_ai/chat/transformation.py +++ b/litellm/llms/fireworks_ai/chat/transformation.py @@ -480,9 +480,9 @@ class FireworksAIConfig(FireworksAIMixin, OpenAIGPTConfig): if not isinstance(schema, dict): return for key in list(schema): - if key in FireworksAIConfig._FIREWORKS_TOOL_SCHEMA_STRIP_KEYS: - schema.pop(key, None) - elif key == "default" and schema[key] is None: + if key in FireworksAIConfig._FIREWORKS_TOOL_SCHEMA_STRIP_KEYS or ( + key == "default" and schema[key] is None + ): schema.pop(key, None) elif key == "properties" and isinstance(schema[key], dict): for prop in schema[key].values():