Update services/llama_generator.py

services/llama_generator.py

@@ -5,10 +5,15 @@ from typing import List, Dict, Any, Optional, Tuple
 from datetime import datetime
 import logging
 from config.config import settings
+
+import services.prompt_builder
+
 import asyncio
 from io import StringIO  
 import pandas as pd
 
+import os 
+
 # Initialize Langfuse
 os.environ["LANGFUSE_PUBLIC_KEY"] = "pk-lf-04d2302a-aa5c-4870-9703-58ab64c3bcae"
 os.environ["LANGFUSE_SECRET_KEY"] = "sk-lf-d34ea200-feec-428e-a621-784fce93a5af"