From 56fba95b4a0c0d91a7067d2e7398a35dbb8b018e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 15 Feb 2024 17:24:27 -0800 Subject: [PATCH] (fix) importing enterprise features --- litellm/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 63e5e3595..c05e2524a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -85,8 +85,10 @@ enterprise_path = abspath(join(project_path, "enterprise")) sys.path.append(enterprise_path) verbose_logger.debug(f"sys.path: {sys.path}") -from enterprise.callbacks.generic_api_callback import GenericAPILogger - +try: + from enterprise.callbacks.generic_api_callback import GenericAPILogger +except Exception as e: + verbose_logger.debug(f"Exception import enterprise features {str(e)}") from typing import cast, List, Dict, Union, Optional, Literal, Any from .caching import Cache