From 219ebc266d97c9ca39def5e72e16acce79f58b52 Mon Sep 17 00:00:00 2001 From: Zara Thomas Date: Wed, 19 Aug 2026 09:35:37 +0100 Subject: [PATCH] style: ruff format Co-authored-by: Cursor --- litellm/llms/fireworks_ai/chat/transformation.py | 4 +--- .../llms/fireworks_ai/chat/test_fireworks_schema_sanitize.py | 4 +--- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/litellm/llms/fireworks_ai/chat/transformation.py b/litellm/llms/fireworks_ai/chat/transformation.py index c4a9c85dbe2..2833c8f6f40 100644 --- a/litellm/llms/fireworks_ai/chat/transformation.py +++ b/litellm/llms/fireworks_ai/chat/transformation.py @@ -481,9 +481,7 @@ class FireworksAIConfig(FireworksAIMixin, OpenAIGPTConfig): if not isinstance(schema, dict): return for key in list(schema): - if key in FireworksAIConfig._FIREWORKS_TOOL_SCHEMA_STRIP_KEYS or ( - 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(): diff --git a/tests/test_litellm/llms/fireworks_ai/chat/test_fireworks_schema_sanitize.py b/tests/test_litellm/llms/fireworks_ai/chat/test_fireworks_schema_sanitize.py index 4f55143440e..fa4ca2b4cb7 100644 --- a/tests/test_litellm/llms/fireworks_ai/chat/test_fireworks_schema_sanitize.py +++ b/tests/test_litellm/llms/fireworks_ai/chat/test_fireworks_schema_sanitize.py @@ -3,9 +3,7 @@ import sys import pytest -sys.path.insert( - 0, os.path.abspath("../../../../..") -) # Adds the parent directory to the system path +sys.path.insert(0, os.path.abspath("../../../../..")) # Adds the parent directory to the system path from litellm.llms.fireworks_ai.chat.transformation import FireworksAIConfig