Soumik555 commited on
Commit
830b4bc
·
1 Parent(s): 67a965a

added gemini too

Browse files
Files changed (1) hide show
  1. controller.py +7 -0
controller.py CHANGED
@@ -302,6 +302,10 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
302
  decoded_url = unquote(csv_url)
303
  detailed_answer = request.get("detailed_answer")
304
  conversation_history = request.get("conversation_history", [])
 
 
 
 
305
 
306
  if if_initial_chat_question(query):
307
  answer = await asyncio.to_thread(
@@ -800,6 +804,9 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
800
  csv_url = unquote(request.get("csv_url", ""))
801
  detailed_answer = request.get("detailed_answer", False)
802
  conversation_history = request.get("conversation_history", [])
 
 
 
803
 
804
  loop = asyncio.get_running_loop()
805
  # First, try the langchain-based method if the question qualifies
 
302
  decoded_url = unquote(csv_url)
303
  detailed_answer = request.get("detailed_answer")
304
  conversation_history = request.get("conversation_history", [])
305
+ generate_report = request.get("generate_report")
306
+
307
+ if generate_report is True:
308
+ return {"answer": "Generate report is True"}
309
 
310
  if if_initial_chat_question(query):
311
  answer = await asyncio.to_thread(
 
804
  csv_url = unquote(request.get("csv_url", ""))
805
  detailed_answer = request.get("detailed_answer", False)
806
  conversation_history = request.get("conversation_history", [])
807
+ generate_report = request.get("generate_report", False)
808
+ if generate_report is True:
809
+ return {"orchestrator_response": "Generate report is True"}
810
 
811
  loop = asyncio.get_running_loop()
812
  # First, try the langchain-based method if the question qualifies