forked from phoenix/litellm-mirror
16 lines
600 B
Python
16 lines
600 B
Python
from litellm.types.utils import GenericStreamingChunk as GChunk
|
|
|
|
|
|
def generic_chunk_has_all_required_fields(chunk: dict) -> bool:
|
|
"""
|
|
Checks if the provided chunk dictionary contains all required fields for GenericStreamingChunk.
|
|
|
|
:param chunk: The dictionary to check.
|
|
:return: True if all required fields are present, False otherwise.
|
|
"""
|
|
_all_fields = GChunk.__annotations__
|
|
|
|
# this is an optional field in GenericStreamingChunk, it's not required to be present
|
|
_all_fields.pop("provider_specific_fields", None)
|
|
|
|
return all(key in chunk for key in _all_fields)
|