Update src/lighteval/models/litellm_model.py

Co-authored-by: Albert Villanova del Moral <[email protected]>

src/lighteval/models/litellm_model.py

@@ -52,6 +52,7 @@
     import litellm
     from litellm import encode
     from litellm.caching.caching import Cache
+    from litellm.utils import ModelResponse
 
     logging.getLogger("LiteLLM").setLevel(logging.WARNING)
     logging.getLogger("LiteLLM").handlers.clear()