Qdrant

databases/qdrant/manifests/01-regional-pd/ha-app.yaml

@@ -24,7 +24,7 @@ spec:
     resourceKind: StatefulSet
     selector:
       matchLabels:
-        app.kubernetes.io/instance=qdrant-database
+        app.kubernetes.io/instance: "qdrant-database"
   policy:
     storageSettings:
       requireRegionalStorage: true

databases/qdrant/manifests/04-qdrant-client/app.py

@@ -0,0 +1,64 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from qdrant_client import QdrantClient
+from qdrant_client.http import models
+import os
+import sys
+import csv
+
+def main(query_string):
+
+    qdrant = QdrantClient(
+        url="http://qdrant-database:6333", api_key=os.getenv("APIKEY"))
+
+    # Create a collection
+    books = [*csv.DictReader(open('/usr/local/dataset/dataset.csv'))]
+
+    documents: list[dict[str, any]] = []
+    metadata: list[dict[str, any]] = []
+    ids: list[int] = []
+
+    for idx, doc in enumerate(books):
+        ids.append(idx)
+        documents.append(doc["description"])
+        metadata.append(
+            {
+                "title": doc["title"],
+                "author": doc["author"],
+                "publishDate": doc["publishDate"],
+            }
+        )
+
+    # Add my_books to the collection 
+    qdrant.add(collection_name="my_books", documents=documents, metadata=metadata, ids=ids, parallel=2)
+
+    # Query the collection
+    results = qdrant.query(
+        collection_name="my_books",
+        query_text=query_string,
+        limit=2,
+    )
+    for result in results:
+        print("Title:", result.metadata["title"], "\nAuthor:", result.metadata["author"])
+        print("Description:", result.metadata["document"], "Published:", result.metadata["publishDate"], "\nScore:", result.score)
+        print("-----")
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        query_string = " ".join(sys.argv[1:])
+        print("Querying qdrant for: ", query_string)
+        main(query_string)
+    else:
+        print("Please provide a query string as an argument.")

databases/qdrant/manifests/04-qdrant-client/client-pod.yaml

@@ -41,8 +41,13 @@ spec:
     volumeMounts:
     - name: books-dataset
       mountPath: /usr/local/dataset
+    - name: script-volume
+      mountPath: /usr/local/script
   volumes:
   - name: books-dataset
     configMap:
       name: books-dataset
+  - name: script-volume
+    configMap:
+      name: demo-app
  # [END gke_databases_qdrant_04_client_pod]