Fix logger problem

This commit is contained in:
yangdx 2025-06-11 13:30:45 +08:00
parent 20c05a7e77
commit ad81e59d9a

View file

@ -1,7 +1,7 @@
from fastapi import APIRouter, HTTPException, Request from fastapi import APIRouter, HTTPException, Request
from pydantic import BaseModel from pydantic import BaseModel
from typing import List, Dict, Any, Optional from typing import List, Dict, Any, Optional
import logging from lightrag.utils import logger
import time import time
import json import json
import re import re
@ -278,7 +278,7 @@ class OllamaAPI:
else: else:
error_msg = f"Provider error: {error_msg}" 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 # Send error message to client
error_data = { error_data = {
@ -496,7 +496,7 @@ class OllamaAPI:
else: else:
error_msg = f"Provider error: {error_msg}" 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 # Send error message to client
error_data = { error_data = {