diff --git a/ai_processor/article_processor.py b/ai_processor/article_processor.py index 3521b08..2977300 100644 --- a/ai_processor/article_processor.py +++ b/ai_processor/article_processor.py @@ -10,10 +10,10 @@ import time from datetime import datetime from typing import List, Tuple -from .fact_extractor import FactExtractor -from .cache_manager import CacheManager -from .metrics_collector import metrics_collector -from .config import BATCH_SIZE, CACHE_FILE +from fact_extractor import FactExtractor +from cache_manager import CacheManager +from metrics_collector import metrics_collector +from config import BATCH_SIZE, CACHE_FILE logger = logging.getLogger(__name__) diff --git a/ai_processor/fact_extractor.py b/ai_processor/fact_extractor.py index 8485f31..ee5f6f8 100644 --- a/ai_processor/fact_extractor.py +++ b/ai_processor/fact_extractor.py @@ -8,8 +8,8 @@ import logging import requests from typing import Dict, Any, Optional -from .config import AI_SERVER_URL, AI_SERVICE_API_KEY, FACT_EXTRACTION_MODEL -from .metrics_collector import metrics_collector +from config import AI_SERVER_URL, AI_SERVICE_API_KEY, FACT_EXTRACTION_MODEL +from metrics_collector import metrics_collector logger = logging.getLogger(__name__) diff --git a/ai_processor/main.py b/ai_processor/main.py index d6e6cd4..7ab8ea7 100644 --- a/ai_processor/main.py +++ b/ai_processor/main.py @@ -8,26 +8,32 @@ import sys import os from datetime import datetime -# Add the current directory to Python path -sys.path.append(os.path.dirname(os.path.abspath(__file__))) - -from article_processor import ArticleProcessor -from metrics_collector import metrics_collector -from cache_manager import CacheManager -from config import CACHE_FILE, LOG_FILE - # Setup logging logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', handlers=[ - logging.FileHandler(LOG_FILE), + logging.FileHandler('ai_processor.log'), logging.StreamHandler() ] ) logger = logging.getLogger(__name__) +def setup_logging(): + """Setup logging configuration.""" + # Ensure log directory exists + log_dir = os.path.dirname('ai_processor.log') + if log_dir: + os.makedirs(log_dir, exist_ok=True) + +from article_processor import ArticleProcessor +from metrics_collector import metrics_collector +from cache_manager import CacheManager +from config import CACHE_FILE, LOG_FILE + +logger = logging.getLogger(__name__) + def setup_logging(): """Setup logging configuration.""" # Ensure log directory exists