diff --git a/docs/my-website/docs/caching/caching_api.md b/docs/my-website/docs/caching/caching_api.md
index 94db11075bd..969de441f30 100644
--- a/docs/my-website/docs/caching/caching_api.md
+++ b/docs/my-website/docs/caching/caching_api.md
@@ -30,7 +30,7 @@ response2 = completion(
 ```python
 import time
 import litellm
-from litellm import completion
+from litellm import completion, embedding
 from litellm.caching import Cache
 litellm.cache = Cache(type="hosted")