feat: add detailed OpenAI API error handling

src/generator/generator.py

@@ -201,16 +201,11 @@ class LyricGenerator:
             chat_history = []
 
         try:
-            print("Starting lyrics generation...")
-            print(f"OpenAI API Key present: {bool(Settings.OPENAI_API_KEY)}")
-            
             # Get source documents with scores first
-            print("Searching for similar documents...")
             docs_and_scores = self.vector_store.similarity_search_with_score(
                 prompt,
                 k=20
             )
-            print(f"Found {len(docs_and_scores)} similar documents")
 
             # Sort by similarity (convert distance to similarity)
             docs_and_scores.sort(key=lambda x: x[1], reverse=False)
@@ -226,22 +221,38 @@ class LyricGenerator:
                     'content': doc.page_content[:200] + "..."  # First 200 chars
                 })
 
-            print("Generating response using QA chain...")
             try:
+                # Test OpenAI connection first
+                print("Testing OpenAI connection...")
+                test_response = self.embeddings.embed_query("test")
+                print("OpenAI connection successful")
+                
                 # Generate response using invoke
+                print("Generating lyrics with QA chain...")
                 response = self.qa_chain.invoke({
                     "question": prompt,
                     "chat_history": chat_history
                 })
-                print("Successfully generated response")
+                print("Successfully generated lyrics")
+
             except Exception as e:
-                print(f"Error in QA chain: {str(e)}")
-                if "Connection" in str(e):
+                error_msg = str(e)
+                print(f"OpenAI API error: {error_msg}")
+                if "auth" in error_msg.lower():
+                    raise RuntimeError(
+                        "OpenAI API authentication failed. Please check your API key."
+                    )
+                elif "rate" in error_msg.lower():
+                    raise RuntimeError(
+                        "OpenAI API rate limit exceeded. Please try again in a moment."
+                    )
+                elif "connect" in error_msg.lower():
                     raise RuntimeError(
-                        "OpenAI API connection error. Please check your internet "
-                        "connection and API key."
+                        "Connection to OpenAI failed. Please check your internet "
+                        "connection and try again."
                     )
-                raise
+                else:
+                    raise RuntimeError(f"OpenAI API error: {error_msg}")
 
             # Add detailed context to response
             response["source_documents_with_scores"] = docs_and_scores