diff --git a/gliner_inference_server/main.py b/gliner_inference_server/main.py index 0cfd97a..0de10e4 100644 --- a/gliner_inference_server/main.py +++ b/gliner_inference_server/main.py @@ -5,12 +5,15 @@ from typing import List, Optional, Union, Dict, Any, Tuple from contextlib import asynccontextmanager import os import torch +import logging from gliner import GLiNER from .models import * from .helpers import * # Global model instance model = None +logger = logging.getLogger(__name__) +logging.basicConfig(level=logging.INFO) @asynccontextmanager @@ -113,6 +116,7 @@ async def general_extraction(request: GeneralRequest): return {"outputs": results[0]} except Exception as e: + logger.error(f"Error in general: {str(e)}", exc_info=True) handle_error(e) @@ -285,6 +289,7 @@ async def relation_extraction(request: RelationExtractionRequest): return {"outputs": results[0]} except Exception as e: + logger.error(f"Error in relation extraction: {str(e)}", exc_info=True) handle_error(e) @@ -324,6 +329,7 @@ async def summarization(request: SummarizationRequest): return {"outputs": results[0]} except Exception as e: + logger.error(f"Error in summarization: {str(e)}", exc_info=True) handle_error(e)