diff options
Diffstat (limited to 'notebooks')
| -rw-r--r-- | notebooks/testing.ipynb | 54 |
1 files changed, 52 insertions, 2 deletions
diff --git a/notebooks/testing.ipynb b/notebooks/testing.ipynb index 806e3fb..d255438 100644 --- a/notebooks/testing.ipynb +++ b/notebooks/testing.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 9, + "execution_count": 27, "id": "c1f56ae3-a056-4b31-bcab-27c2c97c00f1", "metadata": {}, "outputs": [], @@ -17,10 +17,60 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 28, "id": "240df289-d9d6-424b-a5b9-e09dcfefd57e", "metadata": {}, "outputs": [], + "source": [ + "from rag.retriever.retriever import Retriever" + ] + }, + { + "cell_type": "code", + "execution_count": 29, + "id": "01305de3-0d6d-46e7-a0f0-2e2a2f72d563", + "metadata": {}, + "outputs": [ + { + "name": "stderr", + "output_type": "stream", + "text": [ + "\u001b[32m2024-04-13 22:30:45.267\u001b[0m | \u001b[34m\u001b[1mDEBUG \u001b[0m | \u001b[36mrag.retriever.document\u001b[0m:\u001b[36m__configure\u001b[0m:\u001b[36m28\u001b[0m - \u001b[34m\u001b[1mCreating documents table if it does not exist...\u001b[0m\n", + "\u001b[32m2024-04-13 22:30:45.278\u001b[0m | \u001b[34m\u001b[1mDEBUG \u001b[0m | \u001b[36mrag.retriever.vector\u001b[0m:\u001b[36m__configure\u001b[0m:\u001b[36m43\u001b[0m - \u001b[34m\u001b[1mCollection knowledge-base already exists...\u001b[0m\n" + ] + } + ], + "source": [ + "client = Retriever().vec_db" + ] + }, + { + "cell_type": "code", + "execution_count": 32, + "id": "4fcb1862-19a3-482b-b397-ce7be074c187", + "metadata": {}, + "outputs": [ + { + "data": { + "text/plain": [ + "True" + ] + }, + "execution_count": 32, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "client.client.delete_collection(\"knowledge-base\")" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "c86d538c-d613-4e4d-8dd2-3abcea2cfeef", + "metadata": {}, + "outputs": [], "source": [] } ], |