Logger fixes

This commit is contained in:
2025-02-26 20:09:06 +00:00
parent ee9c82f096
commit db9c538d8a
4 changed files with 13 additions and 12 deletions

View File

@@ -6,18 +6,18 @@ API requests, tool conversion, and response processing for OpenAI.
"""
import json
import logging
import traceback
from typing import Any
from airflow.utils.log.logging_mixin import LoggingMixin
from openai import OpenAI
from airflow_wingman.providers.base import BaseLLMProvider
from airflow_wingman.tools import execute_airflow_tool
from airflow_wingman.tools.conversion import convert_to_openai_tools
# Create a logger instance
logger = LoggingMixin().log
# Create a properly namespaced logger for the Airflow plugin
logger = logging.getLogger("airflow.plugins.wingman")
class OpenAIProvider(BaseLLMProvider):