trying groq kimi

orchestrator_functions.py

@@ -342,8 +342,10 @@ def langchain_csv_chart(csv_url: str, question: str, chart_required: bool):
     logger.info("All API keys exhausted for chart generation")
     return None
 
-####################################### Orchestrator Function for Chat #######################################
 
+
+
+####################################### Orchestrator Function for Chat #######################################
 async def csv_chat(csv_url: str, query: str) -> dict:
     """
     Generate a response based on the provided CSV URL and query.
@@ -398,15 +400,32 @@ def is_valid_response(response) -> bool:
     return True
 
 
+
+
 ####################################### Orchestrator Function for Chart #######################################
 
+
 async def csv_chart(csv_url: str, query: str, chat_id: str) -> Dict[str, str]:
     """
     Generate a chart based on the provided CSV URL and query.
-    Prioritizes LangChain-Groq first, then falls back to LangChain-Gemini.
+    Prioritizes LangChain-Gemini first, then falls back to LangChain-Groq.
+    
+    Parameters:
+    - csv_url (str): URL of the CSV file
+    - query (str): Query for generating the chart
+    - chat_id (str): Chat session ID for file storage
+    
+    Returns:
+    - dict: Either {"image_url": "url"} on success or {"error": "message"} on failure
+    
+    Example:
+    - csv_url: "https://example.com/data.csv"
+    - query: "Show sales trends as line chart"
+    Returns:
+    - dict: {"image_url": "https://storage.example.com/chart_uuid.png"}
     """
     error_messages = []
-
+    
     async def upload_and_return(image_path: str) -> Dict[str, str]:
         """Handle image upload and return public URL"""
         unique_name = f'{uuid.uuid4()}.png'
@@ -418,50 +437,54 @@ async def csv_chart(csv_url: str, query: str, chat_id: str) -> Dict[str, str]:
             logger.warning(f"Could not delete temp file {image_path}: {str(e)}")
         return {"image_url": public_url}
 
-    # --- 1. First Attempt: LangChain Groq ---
-    try:
-        lc_groq_paths = await asyncio.to_thread(
-            langchain_csv_chart, csv_url, query, True
-        )
-        logger.info(f"LangChain-Groq chart result: {lc_groq_paths}")
-
-        if lc_groq_paths:
-            if isinstance(lc_groq_paths, list) and lc_groq_paths:
-                first_path = lc_groq_paths[0]
-                if os.path.exists(first_path):
-                    return await upload_and_return(first_path)
-
-            if isinstance(lc_groq_paths, str) and os.path.exists(lc_groq_paths):
-                return await upload_and_return(lc_groq_paths)
-
-        error_messages.append("LangChain-Groq returned invalid result")
-    except Exception as lc_groq_error:
-        error_messages.append(f"LangChain-Groq error: {str(lc_groq_error)}")
-        logger.error(f"Groq chart failed: {str(lc_groq_error)}")
-
-    # --- 2. Fallback Attempt: LangChain Gemini ---
+    # --- 1. First Attempt: LangChain Gemini ---
     try:
         gemini_result = await asyncio.to_thread(
             langchain_gemini_csv_handler, csv_url, query, True
         )
         logger.info(f"LangChain-Gemini chart result: {gemini_result}")
-
+        
         if gemini_result:
+            # Handle string or list response
             if isinstance(gemini_result, str):
                 clean_path = gemini_result.strip()
                 if os.path.exists(clean_path):
                     return await upload_and_return(clean_path)
-
+            
             if isinstance(gemini_result, list) and gemini_result:
                 first_path = gemini_result[0]
                 if os.path.exists(first_path):
                     return await upload_and_return(first_path)
-
+        
         error_messages.append("LangChain-Gemini returned invalid result")
     except Exception as gemini_error:
         error_messages.append(f"LangChain-Gemini error: {str(gemini_error)}")
         logger.error(f"Gemini chart failed: {str(gemini_error)}")
 
+    # --- 2. Fallback Attempt: LangChain Groq ---
+    try:
+        lc_groq_paths = await asyncio.to_thread(
+            langchain_csv_chart, csv_url, query, True
+        )
+        logger.info(f"LangChain-Groq chart result: {lc_groq_paths}")
+        
+        if lc_groq_paths:
+            if isinstance(lc_groq_paths, list) and lc_groq_paths:
+                first_path = lc_groq_paths[0]
+                if os.path.exists(first_path):
+                    return await upload_and_return(first_path)
+            
+            if isinstance(lc_groq_paths, str) and os.path.exists(lc_groq_paths):
+                return await upload_and_return(lc_groq_paths)
+        
+        error_messages.append("LangChain-Groq returned invalid result")
+    except Exception as lc_groq_error:
+        error_messages.append(f"LangChain-Groq error: {str(lc_groq_error)}")
+        logger.error(f"Groq chart failed: {str(lc_groq_error)}")
+
     # --- Final Error Handling ---
     logger.error(f"All chart generation failed. Errors: {'; '.join(error_messages)}")
     return {"error": "Could not generate chart from the provided data"}
+
+
+