From 9495778c2d396113339fd0e4681d70ec22350545 Mon Sep 17 00:00:00 2001 From: anouarbm Date: Mon, 3 Nov 2025 05:27:41 +0100 Subject: [PATCH] refactor: reorder Langfuse import logic for improved clarity Moved logger import before Langfuse block to fix NameError. --- lightrag/llm/openai.py | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) diff --git a/lightrag/llm/openai.py b/lightrag/llm/openai.py index 07942101..1eaf3bd5 100644 --- a/lightrag/llm/openai.py +++ b/lightrag/llm/openai.py @@ -10,17 +10,6 @@ import pipmaster as pm if not pm.is_installed("openai"): pm.install("openai") -# Try to import Langfuse for LLM observability (optional) -# Falls back to standard OpenAI client if not available -try: - from langfuse.openai import AsyncOpenAI - LANGFUSE_ENABLED = True - logger.info("Langfuse observability enabled for OpenAI client") -except ImportError: - from openai import AsyncOpenAI - LANGFUSE_ENABLED = False - logger.debug("Langfuse not available, using standard OpenAI client") - from openai import ( APIConnectionError, RateLimitError, @@ -37,6 +26,19 @@ from lightrag.utils import ( safe_unicode_decode, logger, ) + +# Try to import Langfuse for LLM observability (optional) +# Falls back to standard OpenAI client if not available +try: + from langfuse.openai import AsyncOpenAI + + LANGFUSE_ENABLED = True + logger.info("Langfuse observability enabled for OpenAI client") +except ImportError: + from openai import AsyncOpenAI + + LANGFUSE_ENABLED = False + logger.debug("Langfuse not available, using standard OpenAI client") from lightrag.types import GPTKeywordExtractionFormat from lightrag.api import __api_version__