Fix generate_csv flag

Dataherald · Nov 7, 2023 · 147f207 · 147f207
1 parent 029ad1d
commit 147f207
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dataherald/api/fastapi.py b/dataherald/api/fastapi.py
@@ -174,7 +174,6 @@ def answer_question(
             )
         if generated_answer.csv_file_path:
             generated_answer.sql_query_result = None
-        generated_answer.confidence_score = confidence_score
         generated_answer.exec_time = time.time() - start_generated_answer
         response_repository = ResponseRepository(self.storage)
         return response_repository.insert(generated_answer)
@@ -505,7 +504,9 @@ def create_response(
                 start_generated_answer = time.time()
 
                 generates_nl_answer = GeneratesNlAnswer(self.system, self.storage)
-                response = generates_nl_answer.execute(response, sql_response_only)
+                response = generates_nl_answer.execute(
+                    response, sql_response_only, generate_csv
+                )
         except openai.error.AuthenticationError as e:
             raise HTTPException(status_code=400, detail=str(e)) from e
         except ValueError as e: