Skip to content

Commit

Permalink
Update logger_factory.py
Browse files Browse the repository at this point in the history
Add 'cosmos' to logger_type
  • Loading branch information
rioriost authored Feb 29, 2024
1 parent c8cfb86 commit 7b22578
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions autogen/logger/logger_factory.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from typing import Any, Dict, Optional
from autogen.logger.base_logger import BaseLogger
from autogen.logger.sqlite_logger import SqliteLogger
from autogen.logger.cosmos_logger import CosmosLogger

__all__ = ("LoggerFactory",)

Expand All @@ -13,5 +14,7 @@ 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 CosmosLogger(config)
else:
raise ValueError(f"[logger_factory] Unknown logger type: {logger_type}")

0 comments on commit 7b22578

Please sign in to comment.