“vinit5112” commited on
Commit
52b027f
·
1 Parent(s): 45e068b

async conflict

Browse files
Files changed (2) hide show
  1. backend/Qdrant.py +2 -2
  2. backend/vector_store.py +6 -6
backend/Qdrant.py CHANGED
@@ -43,7 +43,7 @@ class QdrantManager:
43
  vector_size = 384
44
 
45
  # Create collection with vector configuration
46
- await self.qdrant_client.create_collection(
47
  collection_name=collection_name,
48
  vectors_config=models.VectorParams(
49
  size=vector_size,
@@ -62,7 +62,7 @@ class QdrantManager:
62
  }
63
 
64
  for field_name, schema_type in payload_indices.items():
65
- await self.qdrant_client.create_payload_index(
66
  collection_name=collection_name,
67
  field_name=field_name,
68
  field_schema=schema_type
 
43
  vector_size = 384
44
 
45
  # Create collection with vector configuration
46
+ self.qdrant_client.create_collection(
47
  collection_name=collection_name,
48
  vectors_config=models.VectorParams(
49
  size=vector_size,
 
62
  }
63
 
64
  for field_name, schema_type in payload_indices.items():
65
+ self.qdrant_client.create_payload_index(
66
  collection_name=collection_name,
67
  field_name=field_name,
68
  field_schema=schema_type
backend/vector_store.py CHANGED
@@ -64,7 +64,7 @@ class VectorStore:
64
  """
65
  try:
66
  # Try to get collection info - this is more reliable than just listing collections
67
- collection_info = await self.client.get_collection(self.collection_name)
68
  print(f"Collection '{self.collection_name}' exists and is accessible")
69
  return True
70
  except Exception as e:
@@ -85,7 +85,7 @@ class VectorStore:
85
  vector_size = 384
86
 
87
  # Create collection with vector configuration
88
- await self.client.create_collection(
89
  collection_name=self.collection_name,
90
  vectors_config=models.VectorParams(
91
  size=vector_size,
@@ -108,7 +108,7 @@ class VectorStore:
108
 
109
  for field_name, schema_type in payload_indices.items():
110
  try:
111
- await self.client.create_payload_index(
112
  collection_name=self.collection_name,
113
  field_name=field_name,
114
  field_schema=schema_type
@@ -201,7 +201,7 @@ class VectorStore:
201
  )
202
 
203
  # Store in Qdrant
204
- result = await self.client.upsert(
205
  collection_name=self.collection_name,
206
  points=[point]
207
  )
@@ -244,7 +244,7 @@ class VectorStore:
244
  query_embedding = self.embedding_model.encode([query])[0]
245
 
246
  # Search in Qdrant
247
- results = await self.client.search(
248
  collection_name=self.collection_name,
249
  query_vector=query_embedding.tolist(),
250
  limit=limit
@@ -269,7 +269,7 @@ class VectorStore:
269
  async def get_collection_info(self) -> Dict:
270
  """Get information about the collection"""
271
  try:
272
- collection_info = await self.client.get_collection(self.collection_name)
273
  return {
274
  "name": collection_info.config.name,
275
  "vector_size": collection_info.config.params.vectors.size,
 
64
  """
65
  try:
66
  # Try to get collection info - this is more reliable than just listing collections
67
+ collection_info = self.client.get_collection(self.collection_name)
68
  print(f"Collection '{self.collection_name}' exists and is accessible")
69
  return True
70
  except Exception as e:
 
85
  vector_size = 384
86
 
87
  # Create collection with vector configuration
88
+ self.client.create_collection(
89
  collection_name=self.collection_name,
90
  vectors_config=models.VectorParams(
91
  size=vector_size,
 
108
 
109
  for field_name, schema_type in payload_indices.items():
110
  try:
111
+ self.client.create_payload_index(
112
  collection_name=self.collection_name,
113
  field_name=field_name,
114
  field_schema=schema_type
 
201
  )
202
 
203
  # Store in Qdrant
204
+ result = self.client.upsert(
205
  collection_name=self.collection_name,
206
  points=[point]
207
  )
 
244
  query_embedding = self.embedding_model.encode([query])[0]
245
 
246
  # Search in Qdrant
247
+ results = self.client.search(
248
  collection_name=self.collection_name,
249
  query_vector=query_embedding.tolist(),
250
  limit=limit
 
269
  async def get_collection_info(self) -> Dict:
270
  """Get information about the collection"""
271
  try:
272
+ collection_info = self.client.get_collection(self.collection_name)
273
  return {
274
  "name": collection_info.config.name,
275
  "vector_size": collection_info.config.params.vectors.size,