Update notes/llm_notes_generator.py

notes/llm_notes_generator.py

@@ -1,13 +1,3 @@
-class FlexibleFinancialNoteGenerator:
-	def __init__(self):
-		pass
-
-	def generate_note(self, note_number, trial_balance_path=None):
-		return True
-
-	def generate_all_notes(self, trial_balance_path=None):
-		return {"dummy": True}
-
 import json
 import os
 import logging
@@ -19,7 +9,7 @@ import re
 import sys
 from typing import Dict, List, Any, Optional, Tuple
 import pandas as pd
-from pydantic import BaseModel, ValidationError
+from pydantic import BaseModel, Field, ValidationError
 from pydantic_settings import BaseSettings
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from utils.utils import convert_note_json_to_lakhs
@@ -30,10 +20,13 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 class Settings(BaseSettings):
-	openrouter_api_key: str = os.getenv('OPENROUTER_API_KEY', '')
 	api_url: str = "https://openrouter.ai/api/v1/chat/completions"
 	output_dir: str = "data/generated_notes"
 	trial_balance_json: str = "data/output1/parsed_trial_balance.json"
+	
+	model_config = {
+		"extra": "ignore"
+	}
 
 settings = Settings()
 
@@ -55,15 +48,11 @@ class GeneratedNote(BaseModel):
 
 class FlexibleFinancialNoteGenerator:
     def __init__(self, user_api_key: Optional[str] = None):
-        if user_api_key:
-            self.openrouter_api_key = user_api_key
-            logger.info("Using user-provided API key")
-        else:
-            self.openrouter_api_key = settings.openrouter_api_key
-            if not self.openrouter_api_key:
-                logger.error("OpenRouter API key not found. Please provide a valid OPENROUTER_API_KEY.")
-                raise ValueError("OpenRouter API key not found. Please provide a valid OPENROUTER_API_KEY.")
-            logger.info("Using API key from .env file")
+        if not user_api_key:
+            logger.error("OpenRouter API key is required. Please provide OPENROUTER_API_KEY.")
+            raise ValueError("OpenRouter API key is required. Please provide OPENROUTER_API_KEY.")
+        self.openrouter_api_key = user_api_key
+        logger.info("Using provided OpenRouter API key")
         
         self.api_url = settings.api_url
         self.headers = {
@@ -772,7 +761,13 @@ GENERATE VALID JSON NOW (NO ADDITIONAL TEXT):"""
 
 def main() -> None:
     try:
-        generator = FlexibleFinancialNoteGenerator()
+        # Get API key from environment
+        user_api_key = os.environ.get("OPENROUTER_API_KEY")
+        if not user_api_key:
+            logger.error("OpenRouter API key is required. Please provide OPENROUTER_API_KEY.")
+            sys.exit(1)
+        
+        generator = FlexibleFinancialNoteGenerator(user_api_key=user_api_key)
         if not generator.note_templates:
             logger.error("No note templates loaded. Check notes_template.py")
             return