phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

use file size _ name to get file check sum

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-06 15:18:50 -07:00
parent c19066e78e
commit aa06df4101
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -555,8 +555,11 @@ def function_setup(
):
_file_name: BinaryIO = args[1] if len(args) > 1 else kwargs["file"]
file_name = getattr(_file_name, "name", "audio_file")
# Add file checksum to metadata
file_checksum = hashlib.sha256(_file_name.read()).hexdigest()
file_descriptor = _file_name.fileno()
file_stat = os.fstat(file_descriptor)
file_size = str(file_stat.st_size)
file_checksum = _file_name.name + file_size
if "metadata" in kwargs:
kwargs["metadata"]["file_checksum"] = file_checksum
else: