(self, _session: Session)
| 382 | full_message=example_list) |
| 383 | |
| 384 | def choose_table_schema(self, _session: Session): |
| 385 | self.current_logs[OperationEnum.CHOOSE_TABLE] = start_log(session=_session, |
| 386 | operate=OperationEnum.CHOOSE_TABLE, |
| 387 | record_id=self.record.id, |
| 388 | local_operation=True) |
| 389 | self.chat_question.db_schema, tables = self.out_ds_instance.get_db_schema( |
| 390 | self.ds.id, self.chat_question.question) if self.out_ds_instance else get_table_schema( |
| 391 | session=_session, |
| 392 | current_user=self.current_user, |
| 393 | ds=self.ds, |
| 394 | question=self.chat_question.question) |
| 395 | |
| 396 | # Get sample data for all tables |
| 397 | if not self.out_ds_instance: |
| 398 | self.chat_question.sample_data = get_tables_sample_data( |
| 399 | session=_session, |
| 400 | current_user=self.current_user, |
| 401 | ds=self.ds, |
| 402 | table_list=tables) |
| 403 | |
| 404 | self.current_logs[OperationEnum.CHOOSE_TABLE] = end_log(session=_session, |
| 405 | log=self.current_logs[OperationEnum.CHOOSE_TABLE], |
| 406 | full_message=self.chat_question.db_schema) |
| 407 | |
| 408 | def generate_analysis(self, _session: Session): |
| 409 | fields = self.get_fields_from_chart(_session) |
no test coverage detected