test orchestrator

controller.py

@@ -300,6 +300,7 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         query = request.get("query")
         csv_url = request.get("csv_url")
         decoded_url = unquote(csv_url)
+        detailed_answer = request.get("detailed_answer")
 
         if if_initial_chat_question(query):
             answer = await asyncio.to_thread(
@@ -309,12 +310,12 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
             return {"answer": jsonable_encoder(answer)}
         
         # Orchestrate the execution
-        orchestrator_answer = await asyncio.to_thread(
+        if detailed_answer is True:
+           orchestrator_answer = await asyncio.to_thread(
             csv_orchestrator_chat, decoded_url, query
-        )
-        
-        if orchestrator_answer is not None:
-            return {"answer": jsonable_encoder(orchestrator_answer)}
+           )
+           if orchestrator_answer is not None:
+              return {"answer": jsonable_encoder(orchestrator_answer)}
 
         # Process with groq_chat first
         groq_answer = await asyncio.to_thread(groq_chat, decoded_url, query)
@@ -796,6 +797,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
     try:
         query = request.get("query", "")
         csv_url = unquote(request.get("csv_url", ""))
+        detailed_answer = request.get("detailed_answer", False)
         
         loop = asyncio.get_running_loop()
         # First, try the langchain-based method if the question qualifies
@@ -813,12 +815,13 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
                # return FileResponse(langchain_result[0], media_type="image/png")
                
         # Use orchestrator to handle the user's chart query first
-        orchestrator_answer = await asyncio.to_thread(
-            csv_orchestrator_chat, csv_url, query
-        )
+        if detailed_answer is True:
+           orchestrator_answer = await asyncio.to_thread(
+             csv_orchestrator_chat, csv_url, query
+           )
         
-        if orchestrator_answer is not None:
-            return {"orchestrator_response": jsonable_encoder(orchestrator_answer)}
+           if orchestrator_answer is not None:
+             return {"orchestrator_response": jsonable_encoder(orchestrator_answer)}
         
         # Next, try the groq-based method
         groq_result = await loop.run_in_executor(