GirishaBuilds01 commited on
Commit
3c2b266
·
verified ·
1 Parent(s): b43e7eb

Update vector_store.py

Browse files
Files changed (1) hide show
  1. vector_store.py +1 -12
vector_store.py CHANGED
@@ -9,34 +9,23 @@ def index_doc(text, payload):
9
  vector = embed(text)
10
 
11
  client.upsert(
12
-
13
  collection_name=COLLECTION,
14
-
15
  points=[{
16
-
17
  "id": str(uuid.uuid4()),
18
-
19
  "vector": vector,
20
-
21
  "payload": payload | {"text": text}
22
-
23
  }]
24
-
25
  )
26
 
27
 
28
- def search(query, k=5):
29
 
30
  vector = embed(query)
31
 
32
  results = client.search(
33
-
34
  collection_name=COLLECTION,
35
-
36
  query_vector=vector,
37
-
38
  limit=k
39
-
40
  )
41
 
42
  return [r.payload for r in results]
 
9
  vector = embed(text)
10
 
11
  client.upsert(
 
12
  collection_name=COLLECTION,
 
13
  points=[{
 
14
  "id": str(uuid.uuid4()),
 
15
  "vector": vector,
 
16
  "payload": payload | {"text": text}
 
17
  }]
 
18
  )
19
 
20
 
21
+ def search(query, k=3):
22
 
23
  vector = embed(query)
24
 
25
  results = client.search(
 
26
  collection_name=COLLECTION,
 
27
  query_vector=vector,
 
28
  limit=k
 
29
  )
30
 
31
  return [r.payload for r in results]