Organize logging (#138)

This commit is contained in:
Jack Gerrits
2024-06-27 14:36:05 -04:00
committed by GitHub
parent a13c971b16
commit ec17dd16ed
6 changed files with 100 additions and 12 deletions

View File

@@ -1,13 +1,18 @@
from ._events import DeliveryStage, LLMCallEvent, MessageEvent, MessageKind
from ._llm_usage import LLMUsageTracker
ROOT_LOGGER_NAME = "agnext"
"""str: Logger name used for structured event logging"""
EVENT_LOGGER_NAME = "agnext.events"
"""str: Logger name used for structured event logging"""
TRACE_LOGGER_NAME = "agnext.trace"
"""str: Logger name used for developer intended trace logging. The content and format of this log should not be depended upon."""
__all__ = [
"LLMCallEvent",
"ROOT_LOGGER_NAME",
"EVENT_LOGGER_NAME",
"TRACE_LOGGER_NAME",
"LLMUsageTracker",
"MessageEvent",
"MessageKind",
"DeliveryStage",
]

View File

@@ -1,6 +1,6 @@
import logging
from ._events import LLMCallEvent
from .events import LLMCallEvent
class LLMUsageTracker(logging.Handler):

View File

@@ -32,7 +32,8 @@ from openai.types.chat import (
from openai.types.shared_params import FunctionDefinition, FunctionParameters
from typing_extensions import Unpack
from ...application.logging import EVENT_LOGGER_NAME, LLMCallEvent
from ...application.logging import EVENT_LOGGER_NAME
from ...application.logging.events import LLMCallEvent
from .. import (
FunctionCall,
Image,