revert: remove unnecessary connection handling changes

config/settings.py

@@ -13,11 +13,6 @@ class Settings:
     
     # API Keys
     OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-    if not OPENAI_API_KEY:
-        raise ValueError(
-            "OpenAI API key not found. Please set OPENAI_API_KEY "
-            "environment variable."
-        )
     HF_TOKEN = os.getenv("HF_TOKEN")
     
     # Directory Paths

requirements.txt

@@ -6,7 +6,6 @@ chromadb>=0.4.22
 streamlit==1.41.0
 python-dotenv==1.0.1
 huggingface-hub>=0.19.4
-datasets>=2.16.0
 pytest>=7.4.0
 black>=23.0.0
 flake8>=6.0.0

src/generator/generator.py

@@ -12,10 +12,7 @@ class LyricGenerator:
     def __init__(self):
         """Initialize the generator with embeddings"""
         self.embeddings_dir = Settings.get_embeddings_path()
-        self.embeddings = OpenAIEmbeddings(
-            request_timeout=60,  # Increase timeout for embeddings
-            max_retries=3  # Add retries for robustness
-        )
+        self.embeddings = OpenAIEmbeddings()
         self.vector_store = None
         self.qa_chain = None
         
@@ -163,16 +160,14 @@ class LyricGenerator:
             template=system_template
         )
 
-        # Initialize language model with proxy settings
+        # Initialize language model
         llm = ChatOpenAI(
             temperature=0.9,
             model_name="gpt-4",
             max_tokens=1000,
             top_p=0.95,
             presence_penalty=0.0,
-            frequency_penalty=0.1,
-            http_client=None,  # Let OpenAI handle proxy settings
-            request_timeout=60  # Increase timeout
+            frequency_penalty=0.1
         )
 
         # Create QA chain
@@ -223,41 +218,15 @@ class LyricGenerator:
                     'artist': doc.metadata['artist'],
                     'song': doc.metadata['song_title'],
                     'similarity': similarity,
-                    'content': doc.page_content[:200] + "..."  # First 200 chars
-                })
-
-            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
+                    # First 200 chars
+                    'content': doc.page_content[:200] + "..."
                 })
-                print("Successfully generated lyrics")
 
-            except Exception as 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(
-                        "Connection to OpenAI failed. Please check your internet "
-                        "connection and try again."
-                    )
-                else:
-                    raise RuntimeError(f"OpenAI API error: {error_msg}")
+            # Generate response using invoke
+            response = self.qa_chain.invoke({
+                "question": prompt,
+                "chat_history": chat_history
+            })
 
             # Add detailed context to response
             response["source_documents_with_scores"] = docs_and_scores
@@ -266,5 +235,4 @@ class LyricGenerator:
             return response
 
         except Exception as e:
-            print(f"Error in generate_lyrics: {str(e)}")
             raise RuntimeError(f"Failed to generate lyrics: {str(e)}")