yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-12 23:41:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove logging from autogen agentchat (#4510)

Browse Source
This commit is contained in:
Eric Zhu 2024-12-03 14:45:10 -08:00 committed by GitHub
parent 50e84b945e
commit b62f8f63dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 18 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/packages/autogen-agentchat/src/autogen_agentchat/logging/__init__.py
View File

@ -1,4 +0,0 @@
from ._console_log_handler import ConsoleLogHandler
from ._file_log_handler import FileLogHandler
__all__ = ["ConsoleLogHandler", "FileLogHandler"]

20
python/packages/autogen-agentchat/src/autogen_agentchat/logging/_console_log_handler.py
View File

@ -1,20 +0,0 @@
import json
import logging
import sys
from datetime import datetime
from pydantic import BaseModel
class ConsoleLogHandler(logging.Handler):
def emit(self, record: logging.LogRecord) -> None:
ts = datetime.fromtimestamp(record.created).isoformat()
if isinstance(record.msg, BaseModel):
record.msg = json.dumps(
{
"timestamp": ts,
"message": record.msg.model_dump_json(indent=2),
"type": record.msg.__class__.__name__,
},
)
sys.stdout.write(f"{record.msg}\n")

2
python/packages/autogen-agentchat/tests/test_assistant_agent.py
View File

@ -7,7 +7,6 @@ import pytest
from autogen_agentchat import EVENT_LOGGER_NAME
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.base import Handoff, TaskResult
from autogen_agentchat.logging import FileLogHandler
from autogen_agentchat.messages import (
HandoffMessage,
MultiModalMessage,
@ -24,6 +23,7 @@ from openai.types.chat.chat_completion_chunk import ChatCompletionChunk
from openai.types.chat.chat_completion_message import ChatCompletionMessage
from openai.types.chat.chat_completion_message_tool_call import ChatCompletionMessageToolCall, Function
from openai.types.completion_usage import CompletionUsage
from utils import FileLogHandler
logger = logging.getLogger(EVENT_LOGGER_NAME)
logger.setLevel(logging.DEBUG)

2
python/packages/autogen-agentchat/tests/test_group_chat.py
View File

@ -12,7 +12,6 @@ from autogen_agentchat.agents import (
CodeExecutorAgent,
)
from autogen_agentchat.base import Handoff, Response, TaskResult
from autogen_agentchat.logging import FileLogHandler
from autogen_agentchat.messages import (
AgentMessage,
ChatMessage,
@ -41,6 +40,7 @@ from openai.types.chat.chat_completion_chunk import ChatCompletionChunk
from openai.types.chat.chat_completion_message import ChatCompletionMessage
from openai.types.chat.chat_completion_message_tool_call import ChatCompletionMessageToolCall, Function
from openai.types.completion_usage import CompletionUsage
from utils import FileLogHandler
logger = logging.getLogger(EVENT_LOGGER_NAME)
logger.setLevel(logging.DEBUG)

2
python/packages/autogen-agentchat/tests/test_magentic_one_group_chat.py
View File

@ -9,7 +9,6 @@ from autogen_agentchat.agents import (
BaseChatAgent,
)
from autogen_agentchat.base import Response
from autogen_agentchat.logging import FileLogHandler
from autogen_agentchat.messages import (
ChatMessage,
TextMessage,
@ -19,6 +18,7 @@ from autogen_agentchat.teams import (
)
from autogen_core.base import CancellationToken
from autogen_ext.models import ReplayChatCompletionClient
from utils import FileLogHandler
logger = logging.getLogger(EVENT_LOGGER_NAME)
logger.setLevel(logging.DEBUG)

15
python/packages/autogen-agentchat/src/autogen_agentchat/logging/_file_log_handler.py → python/packages/autogen-agentchat/tests/utils.py
View File

@ -1,5 +1,6 @@
import json
import logging
import sys
from datetime import datetime
from pydantic import BaseModel
@ -22,3 +23,17 @@ class FileLogHandler(logging.Handler):
},
)
self.file_handler.emit(record)
class ConsoleLogHandler(logging.Handler):
def emit(self, record: logging.LogRecord) -> None:
ts = datetime.fromtimestamp(record.created).isoformat()
if isinstance(record.msg, BaseModel):
record.msg = json.dumps(
{
"timestamp": ts,
"message": record.msg.model_dump_json(indent=2),
"type": record.msg.__class__.__name__,
},
)
sys.stdout.write(f"{record.msg}\n")