Execute a query with content_type == 'table'.
(self, query, export_tables_to_csv=False, results_only=True)
| 494 | return retrieval_dict |
| 495 | |
| 496 | def table_query(self, query, export_tables_to_csv=False, results_only=True): |
| 497 | |
| 498 | """ Execute a query with content_type == 'table'. """ |
| 499 | |
| 500 | filter_dict = {"content_type": "table"} |
| 501 | retrieval_dict = self.text_query_with_custom_filter(query, filter_dict,results_only=True) |
| 502 | |
| 503 | # output and write tables to csv files |
| 504 | if export_tables_to_csv: |
| 505 | for i, entry in enumerate(retrieval_dict["results"]): |
| 506 | f = self.export_one_table_to_csv(entry,output_fp=LLMWareConfig.get_query_path(), |
| 507 | output_fn="table_{}.csv".format(i)) |
| 508 | |
| 509 | logger.warning(f"update: csv created - {LLMWareConfig.get_query_path()}- {f}") |
| 510 | |
| 511 | return retrieval_dict |
| 512 | |
| 513 | def text_search_by_page (self, query, page_num=1, results_only=True): |
| 514 |
nothing calls this directly
no test coverage detected