update loggers (#810)

ag2ai · Feb 6, 2025 · d35bea2 · d35bea2
1 parent da5ff48
commit d35bea2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/autogen/logger/file_logger.py b/autogen/logger/file_logger.py
@@ -94,7 +94,13 @@ def log_chat_completion(
     ) -> None:
         """Log a chat completion."""
         thread_id = threading.get_ident()
-        source_name = source if isinstance(source, str) else source.name
+        source_name = (
+            source
+            if isinstance(source, str)
+            else source.name
+            if hasattr(source, "name") and source.name is not None
+            else ""
+        )
         try:
             log_data = json.dumps(
                 {

diff --git a/autogen/logger/sqlite_logger.py b/autogen/logger/sqlite_logger.py
@@ -275,7 +275,13 @@ def log_chat_completion(
         else:
             response_messages = json.dumps(to_dict(response), indent=4)
 
-        source_name = source if isinstance(source, str) else source.name
+        source_name = (
+            source
+            if isinstance(source, str)
+            else source.name
+            if hasattr(source, "name") and source.name is not None
+            else ""
+        )
 
         query = """
             INSERT INTO chat_completions (