Logger fixes
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user