test new groq agent

controller.py

@@ -34,6 +34,7 @@ from orchestrator_agent import csv_orchestrator_chat_gemini
 from python_code_executor_service import CsvChatResult, PythonExecutor
 from supabase_service import upload_file_to_supabase
 from cerebras_csv_agent import query_csv_agent_cerebras
+from groq_csv_agent import query_csv_agent_groq
 from util_service import _prompt_generator, process_answer
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -441,17 +442,30 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
             logger.info("Processing detailed answer with orchestrator...")
             return await handle_detailed_answer(decoded_url, query, conversation_history, chat_id)
 
-        # Process with standard CSV agent (Cerebras)
-        logger.info("Processing with standard CSV agent (Cerebras)...")
+        # # Process with standard CSV agent (Cerebras)
+        # logger.info("Processing with standard CSV agent (Cerebras)...")
+        # try:
+        #     result = await query_csv_agent_cerebras(decoded_url, query, chat_id)
+        #     logger.info(f"Standard CSV agent (Cerebras) result: {str(result)[:200]}...")
+        #     if result is not None and result != "":
+        #         return {"answer": result}
+        #     else:
+        #         logger.warning("Standard CSV agent (Cerebras) returned empty or None result")
+        # except Exception as e:
+        #     logger.error(f"Standard CSV agent (Cerebras) failed: {str(e)}")
+        
+        
+        # Process with standard CSV agent (Groq)
+        logger.info("Processing with standard CSV agent (Groq)...")
         try:
-            result = await query_csv_agent_cerebras(decoded_url, query, chat_id)
-            logger.info(f"Standard CSV agent (Cerebras) result: {str(result)[:200]}...")
+            result = await query_csv_agent_groq(decoded_url, query, chat_id)
+            logger.info(f"Standard CSV agent (Groq) result: {str(result)[:200]}...")
             if result is not None and result != "":
                 return {"answer": result}
             else:
-                logger.warning("Standard CSV agent (Cerebras) returned empty or None result")
+                logger.warning("Standard CSV agent (Groq) returned empty or None result")
         except Exception as e:
-            logger.error(f"Standard CSV agent (Cerebras) failed: {str(e)}")
+            logger.error(f"Standard CSV agent (Groq) failed: {str(e)}")
 
         # Fallback to langchain
         logger.info("Falling back to langchain CSV chat...")
@@ -726,14 +740,20 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         #     return {"image_url": image_public_url}
             # return FileResponse(groq_result, media_type="image/png")
         
-        logger.info("Trying cerebras ai llama...")
-        result = await query_csv_agent_cerebras(csv_url, query, chat_id)
-        logger.info("cerebras ai result ==>", result)
+        # logger.info("Trying cerebras ai...")
+        # result = await query_csv_agent_cerebras(csv_url, query, chat_id)
+        # logger.info("cerebras ai result ==>", result)
+        # if result is not None and result != "":
+        #  return {"orchestrator_response": jsonable_encoder(result)}
+        
+        logger.info("Trying groq ai...")
+        result = await query_csv_agent_groq(csv_url, query, chat_id)
+        logger.info("groq ai result ==>", result)
         if result is not None and result != "":
          return {"orchestrator_response": jsonable_encoder(result)}
           
         # Fallback: try langchain-based again
-        logger.error("Cerebras ai llama response failed, trying langchain groq....")
+        logger.error("Groq ai response failed, trying langchain groq....")
         langchain_paths = await loop.run_in_executor(
             process_executor, langchain_csv_chart, csv_url, query, True
         )
@@ -745,7 +765,6 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
            logger.info("Image uploaded to Supabase and Image URL is... ", {image_public_url})
            os.remove(langchain_paths[0])
            return {"image_url": image_public_url}
-           return FileResponse(langchain_paths[0], media_type="image/png")
         else:
            logger.error("All chart generation methods failed")
            return {"answer": "error"}