eudr_retriever

Sleeping

ppsingh commited on Aug 19

Commit

7c5ecb4

verified ·

1 Parent(s): 5254f40

Update app/retriever.py

Files changed (1) hide show

app/retriever.py CHANGED Viewed

@@ -224,7 +224,7 @@ def get_context(
         }
         model = SentenceTransformer(config.get("embeddings", "MODEL_NAME"))
         query_vector = model.encode(query).tolist()
-        retrieved_docs = client.search(
                       collection_name="EUDR",
                       query_vector=query_vector,
                       limit=top_k,

         }
         model = SentenceTransformer(config.get("embeddings", "MODEL_NAME"))
         query_vector = model.encode(query).tolist()
+        retrieved_docs = vectorstore.search(
                       collection_name="EUDR",
                       query_vector=query_vector,
                       limit=top_k,