Lets go

controller.py

@@ -214,11 +214,13 @@ def groq_chat(csv_url: str, question: str):
         except Exception as e:
             error_message = str(e)
             if "429" in error_message:
+                logger.warning("Rate limit exceeded. Switching to next API key.")
                 with current_groq_key_lock:
                     current_groq_key_index += 1
                     if current_groq_key_index >= len(groq_api_keys):
                         return {"error": "All API keys exhausted."}
             else:
+                logger.error("Error in groq_chat: %s", e)
                 return {"error": error_message}
 
 # Modified langchain_csv_chat with thread-safe key rotation
@@ -263,7 +265,14 @@ def langchain_csv_chat(csv_url: str, question: str, chart_required: bool):
             return result.get("output")
 
         except Exception as e:
-            logger.error(f"Error with key index {current_key}: {str(e)}")
+            error_message = str(e)
+            if "429" in error_message:
+                with current_langchain_chart_lock:
+                    current_langchain_chart_key_index = (current_langchain_chart_key_index + 1) % len(groq_api_keys)
+                    logger.warning(f"Rate limit exceeded. Switching to next API key: {groq_api_keys[current_langchain_chart_key_index]}")
+            else:
+                logger.error(f"Error with API key {api_key}: {error_message}")
+                return {"error": error_message}
 
     return {"error": "All API keys exhausted"}
 
@@ -737,8 +746,16 @@ def langchain_csv_chart(csv_url: str, question: str, chart_required: bool):
                 logger.info(f"Langchain chart error (key {current_key}): {output}")
 
         except Exception as e:
-            logger.error(f"Langchain chart error (key {current_key}): {str(e)}")
+            error_message = str(e)
+            if "429" in error_message:
+                with current_langchain_chart_lock:
+                    current_langchain_chart_key_index = (current_langchain_chart_key_index + 1) % len(groq_api_keys)
+                    logger.warning(f"Rate limit exceeded. Switching to next API key: {groq_api_keys[current_langchain_chart_key_index]}")
+            else:
+                logger.error(f"Error with API key {api_key}: {error_message}")
+                return {"error": error_message}
     
+    logger.error("All API keys exhausted for chart generation")
     return "Chart generation failed after all retries"