fix: Restqdrant

class_mod/rest_qdrant.py

@@ -0,0 +1,34 @@
+import requests
+
+class RestQdrantClient:
+    def __init__(self, url, api_key=None, verify=True, timeout=5):
+        self.url = url.rstrip("/")
+        self.session = requests.Session()
+        self.session.verify = verify
+        self.session.headers.update({"Content-Type": "application/json"})
+        if api_key:
+            self.session.headers.update({"api-key": api_key})
+        self.timeout = timeout
+
+    def get_collections(self):
+        r = self.session.get(f"{self.url}/collections", timeout=self.timeout)
+        r.raise_for_status()
+        return r.json()
+
+    def upsert(self, collection_name, points):
+        r = self.session.put(
+            f"{self.url}/collections/{collection_name}/points",
+            json={"points": points},
+            timeout=self.timeout
+        )
+        r.raise_for_status()
+        return r.json()
+
+# Example usage:
+client = RestQdrantClient(
+    url="https://qdrant.taspolsd.dev",
+    api_key="YOUR_API_KEY",   # or None
+    verify=False              # only if needed
+)
+
+print(client.get_collections())

data_importer.py

@@ -3,6 +3,7 @@ from sentence_transformers import SentenceTransformer
 from qdrant_client import QdrantClient
 from qdrant_client.models import Distance, VectorParams, PointStruct
 from typing import List, Dict, Optional, Union
+from class_mod.rest_qdrant import RestQdrantClient
 import uuid
 
 class DataImporter:
@@ -20,7 +21,7 @@ class DataImporter:
     def _init_qdrant(self):
         """Initialize Qdrant client with error handling"""
         try:
-            self.client = QdrantClient(url=self.qdrant_url)
+            self.client = RestQdrantClient(url=self.qdrant_url)
             # Test connection
             self.client.get_collections()
             self.qdrant_available = True

utils/llm_caller.py

@@ -8,6 +8,7 @@ from qdrant_client import QdrantClient
 from openai import OpenAI
 from sentence_transformers import SentenceTransformer
 from interface import PlanResponse, TripPlan, PlanStep, TransportInfo, RetrievedItem, PlanRequest
+from class_mod.rest_qdrant import RestQdrantClient
 import json 
 
 load_dotenv()
@@ -15,14 +16,6 @@ SYSTEM_PROMPT = """You are a helpful travel assistant. Use the provided context
 If the context doesn't contain relevant information, say so politely and provide general advice if possible."""
 '''
 '''
-# @dataclass
-# class RetrievedItem:
-#     place_id: str
-#     place_name: str
-#     description: Optional[str]
-#     score: float
-#     metadata: Dict[str, Any]
-
 class LLMCaller:
     def __init__(self):
         # Environment variables
@@ -32,7 +25,7 @@ class LLMCaller:
                             )
         self.top_k = 3
         self.qdrant_host = os.getenv("QDRANT_HOST")
-        self.qdrant = QdrantClient(
+        self.qdrant = RestQdrantClient(
             url=self.qdrant_host,
         )
         self.system_prompt = SYSTEM_PROMPT