Spaces:
Running
Running
Solving bug with rag searching
Browse files
app.py
CHANGED
@@ -23,8 +23,9 @@ def embed_query(text):
|
|
23 |
# Semantic search using FAISS
|
24 |
def search(query, k=3):
|
25 |
query_vec = embed_query(query)
|
26 |
-
|
27 |
-
|
|
|
28 |
|
29 |
# Chat modes
|
30 |
def chat_no_rag(question):
|
|
|
23 |
# Semantic search using FAISS
|
24 |
def search(query, k=3):
|
25 |
query_vec = embed_query(query)
|
26 |
+
query_vec = query_vec.astype(np.float32) # Required by FAISS
|
27 |
+
D, I = index.search(query_vec, k)
|
28 |
+
return [chunks[i] for i in I[0]]
|
29 |
|
30 |
# Chat modes
|
31 |
def chat_no_rag(question):
|