This commit is contained in:
Wu Shuwen 2026-09-23 14:47:01 +00:00 • committed by GitHub
commit 2ec41a309d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions

View file

@ -4,7 +4,8 @@ from litellm.llms.base_llm.chat.transformation import BaseLLMException
class InfinityError(BaseLLMException):
def __init__(self, status_code: int, message: str, headers: dict | httpx.Headers = {}):
def __init__(self, status_code: int, message: str, headers: dict | httpx.Headers | None = None):
resolved_headers = {} if headers is None else headers
self.status_code = status_code
self.message = message
self.request = httpx.Request(method="POST", url="https://github.com/michaelfeil/infinity")
@ -14,5 +15,5 @@ class InfinityError(BaseLLMException):
message=message,
request=self.request,
response=self.response,
headers=headers,
headers=resolved_headers,
) # Call the base class constructor with the parameters it needs

View file

@ -616,12 +616,15 @@ def _get_equivalent_key(key: str, available_keys: set) -> str | None:
return None
def check_if_part_exists_in_parts(parts: list[PartType], part: PartType, excluded_keys: list[str] = []) -> bool:
def check_if_part_exists_in_parts(
parts: list[PartType], part: PartType, excluded_keys: list[str] | None = None
) -> bool:
"""
Check if a part exists in a list of parts
Handles both camelCase and snake_case key variations (e.g., function_call vs functionCall)
"""
keys_to_compare: Final = set(part.keys()) - set(excluded_keys)
resolved_excluded_keys = [] if excluded_keys is None else excluded_keys
keys_to_compare: Final = set(part.keys()) - set(resolved_excluded_keys)
for p in parts:
p_keys = set(p.keys())
# Check if all keys in part have equivalent values in p