From 0c26404cb3ce8915d1baae9812ed993c2462c10f Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 10 Apr 2024 16:44:40 -0700
Subject: [PATCH] fix - allow base64 cache hits embedding responses

---
 litellm/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index eee7f8202e9..7ce98174cf9 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -236,7 +236,7 @@ class HiddenParams(OpenAIObject):
 
     class Config:
         extra = "allow"
-        protected_namespaces = ()      
+        protected_namespaces = ()
 
     def get(self, key, default=None):
         # Custom .get() method to access attributes with a default value if the attribute doesn't exist
@@ -606,7 +606,7 @@ class ModelResponse(OpenAIObject):
 
 
 class Embedding(OpenAIObject):
-    embedding: list = []
+    embedding: Union[list, str] = []
     index: int
     object: str