Peek at the data in the table
(self, collection: str = None, n_results: int = 20)
| 71 | return df_results |
| 72 | |
| 73 | def peek_table(self, collection: str = None, n_results: int = 20) -> pd.DataFrame: |
| 74 | """ |
| 75 | Peek at the data in the table |
| 76 | """ |
| 77 | if not collection: |
| 78 | collection = self.collection_name |
| 79 | results = self.peek(collection=collection, n_results=n_results) |
| 80 | df_results = pd.DataFrame([json.loads(item['response']) for item in results['metadatas']]) |
| 81 | return df_results |
| 82 | |
| 83 | class ToolReranker(Reranker): |
| 84 | def rerank(self, query_text: str, query_df: pd.DataFrame) -> str: |