phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use file_checksum

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-06 13:55:22 -07:00
parent 5b7d1b0ae4
commit 75cbdda714
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -13,6 +13,7 @@ import base64
import binascii
import copy
import datetime
import hashlib
import inspect
import itertools
import json
@ -554,6 +555,12 @@ 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()
if "metadata" in kwargs:
kwargs["metadata"]["file_checksum"] = file_checksum
else:
kwargs["metadata"] = {"file_checksum": file_checksum}
messages = file_name
elif (
call_type == CallTypes.aspeech.value or call_type == CallTypes.speech.value