refactor(logging): standardize logging approach across services and steps
This commit is contained in:
@@ -1,10 +1,8 @@
|
||||
"""xAI Files & Collections Service"""
|
||||
import os
|
||||
import aiohttp
|
||||
import logging
|
||||
from typing import Optional, List
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
from services.logging_utils import get_service_logger
|
||||
|
||||
XAI_FILES_URL = "https://api.x.ai"
|
||||
XAI_MANAGEMENT_URL = "https://management-api.x.ai"
|
||||
@@ -23,6 +21,7 @@ class XAIService:
|
||||
self.api_key = os.getenv('XAI_API_KEY', '')
|
||||
self.management_key = os.getenv('XAI_MANAGEMENT_KEY', '')
|
||||
self.ctx = ctx
|
||||
self.logger = get_service_logger('xai', ctx)
|
||||
self._session: Optional[aiohttp.ClientSession] = None
|
||||
|
||||
if not self.api_key:
|
||||
@@ -31,10 +30,9 @@ class XAIService:
|
||||
raise ValueError("XAI_MANAGEMENT_KEY not configured in environment")
|
||||
|
||||
def _log(self, msg: str, level: str = 'info') -> None:
|
||||
if self.ctx:
|
||||
getattr(self.ctx.logger, level, self.ctx.logger.info)(msg)
|
||||
else:
|
||||
getattr(logger, level, logger.info)(msg)
|
||||
"""Delegate logging to service logger"""
|
||||
log_func = getattr(self.logger, level, self.logger.info)
|
||||
log_func(msg)
|
||||
|
||||
async def _get_session(self) -> aiohttp.ClientSession:
|
||||
if self._session is None or self._session.closed:
|
||||
|
||||
Reference in New Issue
Block a user