fix: Restqdrant

class_mod/rest_qdrant.py

@@ -15,6 +15,45 @@ class RestQdrantClient:
         r.raise_for_status()
         return r.json()
 
+    def get_collection(self, collection_name):
+        r = self.session.get(f"{self.url}/collections/{collection_name}", timeout=self.timeout)
+        r.raise_for_status()
+        return r.json()
+    def search(self, collection_name, query_vector, limit=10, with_payload=True):
+        payload = {
+            "vector": query_vector,
+            "limit": limit,
+            "with_payload": with_payload
+        }
+        r = self.session.post(
+            f"{self.url}/collections/{collection_name}/points/search",
+            json=payload,
+            timeout=self.timeout
+        )
+        r.raise_for_status()
+        return r.json()
+
+    def delete_collection(self, collection_name):
+        r = self.session.delete(f"{self.url}/collections/{collection_name}", timeout=self.timeout)
+        if r.status_code not in [200, 404]:  # 404 means collection didn't exist
+            r.raise_for_status()
+        return r.json() if r.text else {}
+
+    def create_collection(self, collection_name, vector_size, distance="Cosine"):
+        payload = {
+            "vectors": {
+                "size": vector_size,
+                "distance": distance.upper()  # "COSINE", "EUCLIDEAN", "DOT"
+            }
+        }
+        r = self.session.put(f"{self.url}/collections/{collection_name}", json=payload, timeout=self.timeout)
+        r.raise_for_status()
+        return r.json()
+    def recreate_collection(self, collection_name, vector_size, distance="Cosine"):
+        # Delete if exists
+        self.delete_collection(collection_name)
+        # Create new collection
+        return self.create_collection(collection_name, vector_size, distance)
     def upsert(self, collection_name, points):
         r = self.session.put(
             f"{self.url}/collections/{collection_name}/points",