From b04b144c02180b466536029d6370f6507ba69a0b Mon Sep 17 00:00:00 2001
From: Dario Lopez Padial <dario.lopez@affirm.com>
Date: Fri, 22 Mar 2024 19:44:05 +0100
Subject: [PATCH] Update deployment guide to work locally

---
 config/qdrant_local.yaml | 3 +++
 doc/deployment_guide.md  | 8 +++-----
 requirements.txt         | 2 +-
 src/initialize.py        | 4 ++--
 4 files changed, 9 insertions(+), 8 deletions(-)
 create mode 100644 config/qdrant_local.yaml

diff --git a/config/qdrant_local.yaml b/config/qdrant_local.yaml
new file mode 100644
index 0000000..91ef499
--- /dev/null
+++ b/config/qdrant_local.yaml
@@ -0,0 +1,3 @@
+service:
+  api_key: 823e071f67c198cc05c73f8bd4580865e6a8819a1f3fe57d2cd49b5c892a5233
+  read_only_api_key: d1aab4f05ae4fd7f4e4b8d9e5924469494ebb7897aed46cf2b0df0915410e0b0
diff --git a/doc/deployment_guide.md b/doc/deployment_guide.md
index cff24cd..6e0b5f4 100644
--- a/doc/deployment_guide.md
+++ b/doc/deployment_guide.md
@@ -15,9 +15,7 @@ docker pull qdrant/qdrant
 ### Run the service:
 
 ```
-docker run -p 6333:6333 -p 6334:6334 \
-    -v $(pwd)/qdrant_storage:/qdrant/storage:z \
-    qdrant/qdrant
+docker run -p 6333:6333 -p 6334:6334 -v $(pwd)/justicio/config/qdrant_local.yaml:/qdrant/config/production.yaml -v $(pwd)/qdrant_storage:/qdrant/storage:z qdrant/qdrant
 ```
 
 * REST API: localhost:6333
@@ -50,8 +48,8 @@ export SENDGRID_API_KEY=<your_sendgrid_api_key>
 export OPENAI_API_KEY=<your_open_api_key>
 export TOKENIZERS_PARALLELISM=false
 export TAVILY_API_KEY=""
-export QDRANT_API_KEY=""
-export QDRANT_API_URL="http://0.0.0.0:6333"
+export QDRANT_API_KEY="823e071f67c198cc05c73f8bd4580865e6a8819a1f3fe57d2cd49b5c892a5233"
+export QDRANT_API_URL="http://localhost:6333"
 ```
 
 ### Add some vector to the vector database
diff --git a/requirements.txt b/requirements.txt
index dab0ff7..dbdfe5f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,7 @@ schedule==1.2.1
 langchain==0.0.305
 # langchainplus-sdk==0.0.20
 langsmith==0.0.41
-qdrant-client==1.5.4
+qdrant-client==1.8.0
 supabase==1.0.2
 pinecone-client==2.2.2
 sentence_transformers==2.2.2
diff --git a/src/initialize.py b/src/initialize.py
index 42dd93b..5f1f865 100644
--- a/src/initialize.py
+++ b/src/initialize.py
@@ -156,10 +156,10 @@ def _init_openai_client():
     return client
 
 
-def _exists_collection(client, collection_name):
+def _exists_collection(qdrant_client, collection_name):
     logger = lg.getLogger(_exists_collection.__name__)
     try:
-        client.get_collection(collection_name=collection_name)
+        qdrant_client.get_collection(collection_name=collection_name)
         return True
     except:
         logger.warn("Collection [%s] doesn't exist", collection_name)