added gemini too
Browse files- 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
|