From ad81e59d9a9c4c14c92a0d91b60324c3814b81c6 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 11 Jun 2025 13:30:45 +0800 Subject: [PATCH] Fix logger problem --- lightrag/api/routers/ollama_api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lightrag/api/routers/ollama_api.py b/lightrag/api/routers/ollama_api.py index 93394fbb..8d0895bb 100644 --- a/lightrag/api/routers/ollama_api.py +++ b/lightrag/api/routers/ollama_api.py @@ -1,7 +1,7 @@ from fastapi import APIRouter, HTTPException, Request from pydantic import BaseModel from typing import List, Dict, Any, Optional -import logging +from lightrag.utils import logger import time import json import re @@ -278,7 +278,7 @@ class OllamaAPI: else: error_msg = f"Provider error: {error_msg}" - logging.error(f"Stream error: {error_msg}") + logger.error(f"Stream error: {error_msg}") # Send error message to client error_data = { @@ -496,7 +496,7 @@ class OllamaAPI: else: error_msg = f"Provider error: {error_msg}" - logging.error(f"Stream error: {error_msg}") + logger.error(f"Stream error: {error_msg}") # Send error message to client error_data = {