Update logger_factory.py

autogen/logger/logger_factory.py

@@ -6,6 +6,11 @@
 
 __all__ = ("LoggerFactory",)
 
+try:
+    from autogen.logger.cosmos_db_logger import CosmosDBLogger
+    cosmos_imported = True
+except ImportError:
+    cosmos_imported = False
 
 class LoggerFactory:
     @staticmethod
@@ -16,6 +21,9 @@ def get_logger(logger_type: str = "sqlite", config: Optional[Dict[str, Any]] = N
         if logger_type == "sqlite":
             return SqliteLogger(config)
         elif logger_type == "cosmos":
-            return CosmosDBLogger(config)
+            if cosmos_imported:
+                return CosmosDBLogger(config)
+            else:
+                raise ImportError("CosmosDBLogger could not be imported. Please ensure the cosmos package is installed.")
         else:
             raise ValueError(f"[logger_factory] Unknown logger type: {logger_type}")