Update src/generation/llm.py

src/generation/llm.py

@@ -105,6 +105,8 @@ class LLM:
             response = " ".join(response.split()).strip()
             print("Response generated successfully!")
             return response
+        except Exception as e:
+            raise RuntimeError(f"Failed to generate response: {str(e)}")
     
     def format_query_prompt(self, query: str, cuisines: List[str], dishes: List[str], price_ranges: List[str]) -> str:
         """