Update app.py
Browse files
app.py
CHANGED
@@ -122,6 +122,9 @@ def get_titan_embedding(bedrock, doc_name, text, attempt=0, cutoff=10000):
|
|
122 |
def ask_ds(message, history):
|
123 |
|
124 |
|
|
|
|
|
|
|
125 |
question = message
|
126 |
|
127 |
# RAG
|
@@ -129,7 +132,7 @@ def ask_ds(message, history):
|
|
129 |
|
130 |
similar_documents = []
|
131 |
for file, data in extractions.items():
|
132 |
-
similarity = cosine_similarity(question_embedding, data['embedding'])
|
133 |
similar_documents.append((file, similarity))
|
134 |
|
135 |
similar_documents.sort(key=lambda x: x[1], reverse=False)
|
|
|
122 |
def ask_ds(message, history):
|
123 |
|
124 |
|
125 |
+
if len(message) == 0:
|
126 |
+
return
|
127 |
+
|
128 |
question = message
|
129 |
|
130 |
# RAG
|
|
|
132 |
|
133 |
similar_documents = []
|
134 |
for file, data in extractions.items():
|
135 |
+
similarity = cosine_similarity(question_embedding, np.array(data['embedding']))
|
136 |
similar_documents.append((file, similarity))
|
137 |
|
138 |
similar_documents.sort(key=lambda x: x[1], reverse=False)
|