From c19066e78ebb73dd90c438434ca3280747b43e21 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 6 Aug 2024 13:55:22 -0700
Subject: [PATCH] use file_checksum

---
 litellm/utils.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/litellm/utils.py b/litellm/utils.py
index fbe6bd26e..c859a962d 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -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